Update src/ui_components_original.py

src/ui_components_original.py

@@ -2,12 +2,10 @@ import gradio as gr
 import os
 import re
 import logging
-import tempfile
-import shutil
 import base64
 from datetime import datetime
 from PIL import Image
-import html # Import the html module for escaping
+import html
 
 from .patient_history import PatientHistoryManager, ReportGenerator
 
@@ -45,7 +43,7 @@ class UIComponents:
         # Ensure uploads directory exists
         if not os.path.exists("uploads"):
             os.makedirs("uploads", exist_ok=True)
-    
+
     def image_to_base64(self, image_path):
         """Convert image to base64 data URL for embedding in HTML"""
         if not image_path or not os.path.exists(image_path):
@@ -70,7 +68,7 @@ class UIComponents:
         except Exception as e:
             logging.error(f"Error converting image to base64: {e}")
             return None
-    
+
     def markdown_to_html(self, markdown_text):
         """Convert markdown text to proper HTML format with enhanced support"""
         if not markdown_text:
@@ -104,7 +102,7 @@ class UIComponents:
         # Convert horizontal rules
         html_text = re.sub(r"^\s*[-*_]{3,}\s*$", r"<hr>", html_text, flags=re.MULTILINE)
 
-        # Convert bullet points and handle nested lists (simplified for example)
+        # Convert bullet points
         lines = html_text.split("\n")
         in_list = False
         result_lines = []
@@ -129,7 +127,7 @@ class UIComponents:
             result_lines.append("</ul>")
         
         return "\n".join(result_lines)
-    
+
     def get_organizations_dropdown(self):
         """Get list of organizations for dropdown"""
         try:
@@ -138,10 +136,10 @@ class UIComponents:
         except Exception as e:
             logging.error(f"Error getting organizations: {e}")
             return ["Default Hospital - Location"]
- 
+
     def get_custom_css(self):
         return """
-/* =================== ORIGINAL SMARTHEAL CSS =================== */
+/* =================== SMARTHEAL CSS =================== */
 /* Global Styling */
 body, html {
     margin: 0 !important;
@@ -168,23 +166,6 @@ body, html {
     overflow: hidden !important;
 }
 
-.medical-header::before {
-    content: '' !important;
-    position: absolute !important;
-    top: 0 !important;
-    left: 0 !important;
-    right: 0 !important;
-    bottom: 0 !important;
-    background: url('data:image/svg+xml,<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100"><defs><pattern id="grid" width="10" height="10" patternUnits="userSpaceOnUse"><path d="M 10 0 L 0 0 0 10" fill="none" stroke="rgba(255,255,255,0.1)" stroke-width="0.5"/></pattern></defs><rect width="100" height="100" fill="url(%23grid)" /></svg>') !important;
-    opacity: 0.1 !important;
-    z-index: 1 !important;
-}
-
-.medical-header > * {
-    position: relative !important;
-    z-index: 2 !important;
-}
-
 .logo {
     width: 80px !important;
     height: 80px !important;
@@ -229,17 +210,6 @@ body, html {
     overflow: hidden !important;
 }
 
-.gr-form::before {
-    content: '' !important;
-    position: absolute !important;
-    top: 0 !important;
-    left: 0 !important;
-    right: 0 !important;
-    height: 4px !important;
-    background: linear-gradient(90deg, #e53e3e 0%, #f56565 50%, #e53e3e 100%) !important;
-    z-index: 1 !important;
-}
-
 /* Professional Input Fields */
 .gr-textbox, .gr-number {
     border-radius: 12px !important;
@@ -252,10 +222,7 @@ body, html {
     padding: 16px 20px !important;
 }
 
-.gr-textbox:focus,
-.gr-number:focus,
-.gr-textbox input:focus,
-.gr-number input:focus {
+.gr-textbox:focus, .gr-number:focus, .gr-textbox input:focus, .gr-number input:focus {
     border-color: #E53E3E !important;
     box-shadow: 0 0 0 4px rgba(229, 62, 62, 0.1) !important;
     background: #FFFFFF !important;
@@ -263,32 +230,8 @@ body, html {
     transform: translateY(-1px) !important;
 }
 
-.gr-textbox input,
-.gr-number input {
-    background: transparent !important;
-    border: none !important;
-    outline: none !important;
-    color: #1A202C !important;
-    font-size: 1rem !important;
-    width: 100% !important;
-    padding: 0 !important;
-}
-
-.gr-textbox label,
-.gr-number label,
-.gr-dropdown label,
-.gr-radio label,
-.gr-checkbox label {
-    font-weight: 600 !important;
-    color: #2D3748 !important;
-    font-size: 1rem !important;
-    margin-bottom: 8px !important;
-    display: block !important;
-}
-
 /* Enhanced Button Styling */
-button.gr-button,
-button.gr-button-primary {
+button.gr-button, button.gr-button-primary {
     background: linear-gradient(135deg, #E53E3E 0%, #C53030 100%) !important;
     color: #FFFFFF !important;
     border: none !important;
@@ -306,42 +249,12 @@ button.gr-button-primary {
     cursor: pointer !important;
 }
 
-button.gr-button::before {
-    content: '' !important;
-    position: absolute !important;
-    top: 0 !important;
-    left: -100% !important;
-    width: 100% !important;
-    height: 100% !important;
-    background: linear-gradient(90deg, transparent, rgba(255,255,255,0.4), transparent) !important;
-    transition: left 0.5s !important;
-}
-
-button.gr-button:hover::before {
-    left: 100% !important;
-}
-
-button.gr-button:hover,
-button.gr-button-primary:hover {
+button.gr-button:hover, button.gr-button-primary:hover {
     background: linear-gradient(135deg, #C53030 0%, #9C2A2A 100%) !important;
     box-shadow: 0 8px 32px rgba(229, 62, 62, 0.4) !important;
     transform: translateY(-3px) !important;
 }
 
-button.gr-button:active,
-button.gr-button-primary:active {
-    transform: translateY(-1px) !important;
-    box-shadow: 0 4px 16px rgba(229, 62, 62, 0.5) !important;
-}
-
-button.gr-button:disabled {
-    background: #A0AEC0 !important;
-    color: #718096 !important;
-    cursor: not-allowed !important;
-    box-shadow: none !important;
-    transform: none !important;
-}
-
 /* Professional Status Messages */
 .status-success {
     background: linear-gradient(135deg, #F0FFF4 0%, #E6FFFA 100%) !important;
@@ -379,150 +292,6 @@ button.gr-button:disabled {
     backdrop-filter: blur(10px) !important;
 }
 
-/* Professional Card Layout */
-.medical-card {
-    background: linear-gradient(145deg, #FFFFFF 0%, #F7FAFC 100%) !important;
-    border-radius: 20px !important;
-    padding: 32px !important;
-    margin: 24px 0 !important;
-    box-shadow: 0 16px 48px rgba(0, 0, 0, 0.08) !important;
-    border: 1px solid rgba(229, 62, 62, 0.1) !important;
-    backdrop-filter: blur(10px) !important;
-    position: relative !important;
-    overflow: hidden !important;
-}
-
-.medical-card::before {
-    content: '' !important;
-    position: absolute !important;
-    top: 0 !important;
-    left: 0 !important;
-    right: 0 !important;
-    height: 4px !important;
-    background: linear-gradient(90deg, #E53E3E 0%, #F56565 50%, #E53E3E 100%) !important;
-}
-
-.medical-card-title {
-    font-size: 1.75rem !important;
-    font-weight: 700 !important;
-    color: #1A202C !important;
-    margin-bottom: 24px !important;
-    padding-bottom: 16px !important;
-    border-bottom: 2px solid #E53E3E !important;
-    text-align: center !important;
-    position: relative !important;
-}
-
-.medical-card-title::after {
-    content: '' !important;
-    position: absolute !important;
-    bottom: -2px !important;
-    left: 50% !important;
-    transform: translateX(-50%) !important;
-    width: 60px !important;
-    height: 4px !important;
-    background: linear-gradient(90deg, transparent, #E53E3E, transparent) !important;
-    border-radius: 2px !important;
-}
-
-/* Professional Dropdown Styling */
-.gr-dropdown {
-    border-radius: 12px !important;
-    border: 2px solid #E2E8F0 !important;
-    background: #FFFFFF !important;
-    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.05) !important;
-}
-
-.gr-dropdown:focus,
-.gr-dropdown select:focus {
-    border-color: #E53E3E !important;
-    box-shadow: 0 0 0 4px rgba(229, 62, 62, 0.1) !important;
-    outline: none !important;
-}
-
-.gr-dropdown select {
-    background: transparent !important;
-    border: none !important;
-    color: #1A202C !important;
-    font-size: 1rem !important;
-    padding: 16px 20px !important;
-    border-radius: 12px !important;
-}
-
-/* Radio button styling */
-.gr-radio input[type="radio"] {
-    margin-right: 8px !important;
-    transform: scale(1.2) !important;
-}
-
-.gr-radio label {
-    display: flex !important;
-    align-items: center !important;
-    padding: 8px 0 !important;
-    font-size: 1rem !important;
-    line-height: 1.5 !important;
-    cursor: pointer !important;
-    color: #1A202C !important;
-}
-
-/* Tab styling */
-.gr-tab {
-    color: #1A202C !important;
-    font-weight: 500 !important;
-    font-size: 1rem !important;
-    padding: 12px 20px !important;
-    background-color: #F7FAFC !important;
-}
-
-.gr-tab.selected {
-    color: #E53E3E !important;
-    font-weight: 600 !important;
-    border-bottom: 2px solid #E53E3E !important;
-    background-color: #FFFFFF !important;
-}
-
-/* Image upload styling */
-.gr-image {
-    border: 3px dashed #CBD5E0 !important;
-    border-radius: 16px !important;
-    background-color: #F7FAFC !important;
-    transition: all 0.2s ease !important;
-}
-
-.gr-image:hover {
-    border-color: #E53E3E !important;
-    background-color: #FFF5F5 !important;
-}
-
-/* Analyze button special styling */
-#analyze-btn {
-    background: linear-gradient(135deg, #1B5CF3 0%, #1E3A8A 100%) !important;
-    color: #FFFFFF !important;
-    border: none !important;
-    border-radius: 8px !important;
-    font-weight: 700 !important;
-    padding: 14px 28px !important;
-    font-size: 1.1rem !important;
-    letter-spacing: 0.5px !important;
-    text-align: center !important;
-    transition: all 0.2s ease-in-out !important;
-}
-
-#analyze-btn:hover {
-    background: linear-gradient(135deg, #174ea6 0%, #123b82 100%) !important;
-    box-shadow: 0 4px 14px rgba(27, 95, 193, 0.4) !important;
-    transform: translateY(-2px) !important;
-}
-
-#analyze-btn:disabled {
-    background: #A0AEC0 !important;
-    color: #1A202C !important;
-    cursor: not-allowed !important;
-    box-shadow: none !important;
-    transform: none !important;
-}
-
 /* Image gallery styling for better visualization */
 .image-gallery {
     display: grid;
@@ -558,6 +327,26 @@ button.gr-button:disabled {
     font-size: 0.9em;
 }
 
+/* Analyze button special styling */
+#analyze-btn {
+    background: linear-gradient(135deg, #1B5CF3 0%, #1E3A8A 100%) !important;
+    color: #FFFFFF !important;
+    border: none !important;
+    border-radius: 8px !important;
+    font-weight: 700 !important;
+    padding: 14px 28px !important;
+    font-size: 1.1rem !important;
+    letter-spacing: 0.5px !important;
+    text-align: center !important;
+    transition: all 0.2s ease-in-out !important;
+}
+
+#analyze-btn:hover {
+    background: linear-gradient(135deg, #174ea6 0%, #123b82 100%) !important;
+    box-shadow: 0 4px 14px rgba(27, 95, 193, 0.4) !important;
+    transform: translateY(-2px) !important;
+}
+
 /* Responsive design */
 @media (max-width: 768px) {
     .medical-header {
@@ -580,12 +369,6 @@ button.gr-button:disabled {
         margin: 8px 0 !important;
     }
     
-    button.gr-button,
-    button.gr-button-primary {
-        padding: 14px 20px !important;
-        font-size: 14px !important;
-    }
-    
     .image-gallery {
         grid-template-columns: 1fr;
     }
@@ -593,10 +376,10 @@ button.gr-button:disabled {
 """
 
     def create_interface(self):
-        """Create the main Gradio interface with original styling and base64 image embedding"""
+        """Create the main Gradio interface with comprehensive analysis display"""
         with gr.Blocks(css=self.get_custom_css(), title="SmartHeal - AI Wound Care Assistant") as app:
             
-            # Header with SmartHeal logo (from original)
+            # Header with SmartHeal logo
             logo_url = "https://scontent.fccu31-2.fna.fbcdn.net/v/t39.30808-6/275933824_102121829111657_3325198727201325354_n.jpg?_nc_cat=104&ccb=1-7&_nc_sid=6ee11a&_nc_ohc=45krrEUpcSUQ7kNvwGVdiMW&_nc_oc=AdkTdxEC_TkYGiyDkEtTJZ_DFZELW17XKFmWpswmFqGB7JSdvTyWtnrQyLS0USngEiY&_nc_zt=23&_nc_ht=scontent.fccu31-2.fna&_nc_gid=ufAA4Hj5gTRwON5POYzz0Q&oh=00_AfW1-jLEN5RGeggqOvGgEaK_gdg0EDgxf_VhKbZwFLUO0Q&oe=6897A98B"
             
             gr.HTML(f"""
@@ -609,7 +392,7 @@ button.gr-button:disabled {
             </div>
             """)
             
-            # Professional disclaimer (from original)
+            # Professional disclaimer
             gr.HTML("""
         <div style="border: 2px solid #FF6B6B; background-color: #FFE5E5; padding: 15px; border-radius: 12px; margin: 10px 0;">
             <h3 style="color: #D63031; margin-top: 0;">⚠️ IMPORTANT DISCLAIMER</h3>
@@ -619,9 +402,9 @@ button.gr-button:disabled {
         </div>
         """)
             
-            # Main interface with conditional visibility (ORIGINAL STRUCTURE)
+            # Main interface with conditional visibility
             with gr.Row():
-                # Professional Authentication Panel (visible when not logged in)
+                # Authentication Panel (visible when not logged in)
                 with gr.Column(visible=True) as auth_panel:
                     gr.HTML("""
                     <div style="text-align: center; margin: 40px 0;">
@@ -728,7 +511,7 @@ button.gr-button:disabled {
                     gr.HTML('<div class="medical-card-title">👩‍⚕️ Practitioner Dashboard</div>')
                     
                     user_info = gr.HTML("")
-                    logout_btn_prac = gr.Button("🚪 Logout", variant="secondary", elem_classes=["logout-btn"])
+                    logout_btn_prac = gr.Button("🚪 Logout", variant="secondary")
                     
                     # Main tabs for different functions
                     with gr.Tabs():
@@ -774,8 +557,7 @@ button.gr-button:disabled {
                                     gr.HTML("<h3>📸 Wound Image Upload</h3>")
                                     wound_image = gr.Image(
                                         label="Upload Wound Image",
-                                        type="filepath",
-                                        elem_classes=["image-upload"]
+                                        type="filepath"
                                     )
                                     
                                     gr.HTML("<h3>📝 Medical History</h3>")
@@ -892,27 +674,53 @@ button.gr-button:disabled {
                     if not wound_image:
                         return "<div class='status-error'>❌ Please upload a wound image for analysis.</div>"
                     
-                    # Show loading state
+                    # Show loading state first
                     loading_html = """
-                    <div style="text-align:center; padding: 30px;">
-                        <div style="display:inline-block; border:4px solid #3182ce; border-radius:50%; border-top-color:transparent; width:40px; height:40px; animation:spin 1s linear infinite;"></div>
-                        <p style="margin-top:15px; color:#3182ce; font-weight:600;">Processing wound analysis...</p>
+                    <div style="text-align:center; padding: 40px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); border-radius: 16px; color: white; margin: 20px 0;">
+                        <div style="display:inline-block; border:4px solid rgba(255,255,255,0.3); border-radius:50%; border-top-color:white; width:50px; height:50px; animation:spin 1s linear infinite; margin-bottom: 20px;"></div>
+                        <h3 style="margin: 0; font-size: 1.5rem;">🔬 SmartHeal AI Processing</h3>
+                        <p style="margin: 10px 0 0 0; font-size: 1rem; opacity: 0.9;">Analyzing wound image with advanced computer vision...</p>
+                        <div style="background: rgba(255,255,255,0.1); border-radius: 8px; padding: 10px; margin-top: 15px;">
+                            <p style="margin: 0; font-size: 0.9rem;">⚡ Detection → 📏 Segmentation → 🤖 AI Report</p>
+                        </div>
                         <style>@keyframes spin {0% {transform:rotate(0deg)} 100% {transform:rotate(360deg)}}</style>
                     </div>
                     """
                     
-                    # 1. Construct questionnaire dictionary for AIProcessor
-                    questionnaire_data = {
-                        'user_id': self.current_user.get('id'),
+                    # 1. Save questionnaire FIRST to get valid ID
+                    questionnaire_data_for_db = {
+                        'user_id': self.current_user.get('id', 1),  # Default to 1 if no user logged in
+                        'patient_name': patient_name,
+                        'patient_age': patient_age,
+                        'patient_gender': patient_gender,
+                        'wound_location': wound_location,
+                        'wound_duration': wound_duration,
+                        'pain_level': pain_level,
+                        'moisture_level': moisture_level,
+                        'infection_signs': infection_signs,
+                        'diabetic_status': diabetic_status,
+                        'previous_treatment': previous_treatment,
+                        'medical_history': medical_history,
+                        'medications': medications,
+                        'allergies': allergies,
+                        'additional_notes': additional_notes
+                    }
+                    
+                    # Save questionnaire FIRST to get the ID
+                    questionnaire_response_id = self.database_manager.save_questionnaire(questionnaire_data_for_db)
+                    logging.info(f"✅ Questionnaire saved with response ID: {questionnaire_response_id}")
+                    
+                    # 2. Create AIProcessor questionnaire format  
+                    questionnaire_data_for_ai = {
                         'age': patient_age,
                         'diabetic': 'Yes' if diabetic_status != 'Non-diabetic' else 'No',
                         'allergies': allergies,
                         'date_of_injury': 'Unknown',  # Not collected in this form
                         'professional_care': 'Yes',  # Assumed since using professional interface
-                        'oozing_bleeding': 'None',  # Could be mapped from infection_signs
+                        'oozing_bleeding': 'Minor Oozing' if infection_signs != 'None' else 'None',
                         'infection': 'Yes' if infection_signs != 'None' else 'No',
                         'moisture': moisture_level,
-                        # Additional fields for comprehensive analysis
+                        # Additional comprehensive fields
                         'patient_name': patient_name,
                         'patient_gender': patient_gender,
                         'wound_location': wound_location,
@@ -923,44 +731,87 @@ button.gr-button:disabled {
                         'medications': medications,
                         'additional_notes': additional_notes
                     }
-                    
-                    # 2. Save questionnaire in DB first
-                    questionnaire_id = self.database_manager.save_questionnaire(questionnaire_data)
             
                     # 3. Run AI analysis using the AIProcessor class
                     try:
-                        logging.info(f"Starting AI analysis for image: {wound_image}")
+                        logging.info(f"🔬 Starting AI analysis for image: {wound_image}")
                         
                         # Use the AIProcessor analyze_wound method
-                        analysis_result = self.wound_analyzer.analyze_wound(wound_image, questionnaire_data)
+                        analysis_result = self.wound_analyzer.analyze_wound(wound_image, questionnaire_data_for_ai)
                         
                         if not analysis_result.get('success', False):
-                            raise Exception(analysis_result.get('error', 'Analysis failed for unknown reason'))
+                            error_msg = analysis_result.get('error', 'Analysis failed for unknown reason')
+                            logging.error(f"❌ AI Analysis failed: {error_msg}")
+                            
+                            return f"""
+                            <div class='status-error' style='padding: 30px; background: #fff5f5; border-left: 5px solid #e53e3e; margin: 20px 0; border-radius: 12px;'>
+                                <h3 style="color: #c53030; margin-top: 0;">❌ AI Analysis Error</h3>
+                                <p style="color: #742a2a; font-size: 1.1rem;"><strong>Error Details:</strong></p>
+                                <div style='background: #fed7d7; padding: 15px; border-radius: 8px; margin: 10px 0;'>
+                                    <code style='color: #742a2a; font-size: 0.9rem;'>{error_msg}</code>
+                                </div>
+                                <p style="color: #742a2a;">Please try again with a different image or contact technical support.</p>
+                                <div style="margin-top: 15px; padding: 15px; background: #fff; border-radius: 8px; border: 1px solid #feb2b2;">
+                                    <strong>Troubleshooting Tips:</strong>
+                                    <ul style="margin: 10px 0; color: #742a2a;">
+                                        <li>Ensure image is clear and well-lit</li>
+                                        <li>Wound should be clearly visible in the image</li>
+                                        <li>Try a different image format (PNG, JPG)</li>
+                                        <li>Check image file size (max 10MB recommended)</li>
+                                    </ul>
+                                </div>
+                            </div>
+                            """
                         
-                        # 4. Save AI analysis result to database
-                        self.database_manager.save_analysis_result(questionnaire_id, analysis_result)
+                        logging.info("✅ AI Analysis completed successfully")
+                        
+                        # 4. Save analysis result with valid questionnaire_response_id
+                        try:
+                            if questionnaire_response_id:
+                                self.database_manager.save_analysis_result(questionnaire_response_id, analysis_result)
+                                logging.info("✅ Analysis result saved to database")
+                        except Exception as db_error:
+                            logging.error(f"❌ Database save error: {db_error}")
+                            # Continue with display even if DB save fails
                         
                         # 5. Format comprehensive analysis results with all images
                         formatted_analysis = self._format_comprehensive_analysis_results(
-                            analysis_result, wound_image, questionnaire_data
+                            analysis_result, wound_image, questionnaire_data_for_ai
                         )
                         
                         return formatted_analysis
             
                     except Exception as analysis_error:
-                        logging.error(f"AI analysis error: {analysis_error}")
+                        logging.error(f"❌ AI analysis exception: {analysis_error}", exc_info=True)
                         return f"""
-                        <div class='status-error' style='padding: 20px; background: #ffeeee; border-left: 5px solid #ff5555; margin: 20px 0;'>
-                            <h3>❌ Analysis Error</h3>
-                            <p>There was an error analyzing the wound image:</p>
-                            <pre style='background: #f5f5f5; padding: 10px; border-radius: 5px;'>{str(analysis_error)}</pre>
-                            <p>Please try again with a different image or contact support.</p>
+                        <div class='status-error' style='padding: 30px; background: #fff5f5; border-left: 5px solid #e53e3e; margin: 20px 0; border-radius: 12px;'>
+                            <h3 style="color: #c53030; margin-top: 0;">❌ Analysis Processing Error</h3>
+                            <p style="color: #742a2a;">There was an unexpected error during wound analysis:</p>
+                            <div style='background: #fed7d7; padding: 15px; border-radius: 8px; margin: 15px 0;'>
+                                <code style='color: #742a2a; font-size: 0.9rem; word-break: break-word;'>{str(analysis_error)}</code>
+                            </div>
+                            <p style="color: #742a2a;"><strong>Next Steps:</strong></p>
+                            <ul style="color: #742a2a; margin: 10px 0;">
+                                <li>Refresh the page and try again</li>
+                                <li>Check your internet connection</li>
+                                <li>Try with a different wound image</li>
+                                <li>Contact system administrator if the problem persists</li>
+                            </ul>
                         </div>
                         """
             
                 except Exception as e:
-                    logging.error(f"Analysis handler error: {e}")
-                    return f"<div class='status-error'>❌ Analysis failed: {str(e)}</div>"
+                    logging.error(f"❌ Analysis handler error: {e}", exc_info=True)
+                    return f"""
+                    <div class='status-error' style='padding: 30px; background: #fff5f5; border-left: 5px solid #e53e3e; margin: 20px 0; border-radius: 12px;'>
+                        <h3 style="color: #c53030; margin-top: 0;">❌ System Error</h3>
+                        <p style="color: #742a2a;">A system error occurred while processing your request:</p>
+                        <div style='background: #fed7d7; padding: 15px; border-radius: 8px; margin: 15px 0;'>
+                            <code style='color: #742a2a; font-size: 0.9rem;'>{str(e)}</code>
+                        </div>
+                        <p style="color: #742a2a;">Please contact technical support if this issue continues.</p>
+                    </div>
+                    """
 
             def handle_logout():
                 self.current_user = {}
@@ -983,7 +834,7 @@ button.gr-button:disabled {
             
             def load_patient_history():
                 try:
-                    user_id = self.current_user.get('id')
+                    user_id = self.current_user.get('id', 1)
                     if not user_id:
                         return "<div class='status-error'>❌ Please login first.</div>"
                     
@@ -996,7 +847,7 @@ button.gr-button:disabled {
             
             def search_specific_patient(patient_name):
                 try:
-                    user_id = self.current_user.get('id')
+                    user_id = self.current_user.get('id', 1)
                     if not user_id:
                         return "<div class='status-error'>❌ Please login first.</div>"
                     
@@ -1116,7 +967,7 @@ button.gr-button:disabled {
             risk_class = "low"
             if risk_level.lower() == "moderate":
                 risk_class = "moderate"
-            elif risk_level.lower() == "high":
+            elif risk_level.lower() in ["high", "very high"]:
                 risk_class = "high"
             
             # Format risk factors