Update app.py

app.py

@@ -86,14 +86,34 @@ def analyze_ecg_image(image, vision_model="llama-3.2-90b-vision-preview"):
     8. Rhythm classification
     9. Specific patterns (if any)
     
-    Report exact numerical values where visible. Format your response as a structured report with clear sections for different measurements and observations. If certain measurements aren't visible in the image, indicate that they cannot be determined.
+    Report exact numerical values where visible. Format your response using HTML list elements for better readability.
+    
+    If certain measurements aren't visible in the image, indicate that they cannot be determined.
     
     If you notice any abnormalities or concerning patterns, highlight them clearly but avoid making definitive diagnoses.
     
+    Format your response like this:
+    <h3>ECG Report</h3>
+    <ul>
+      <li><strong>Date and Time:</strong> {timestamp}</li>
+      <li><strong>Heart Rate:</strong> [value] bpm</li>
+      <li><strong>PR Interval:</strong> [value] ms</li>
+      <!-- Other measurements -->
+    </ul>
+    
+    <h3>Additional Observations</h3>
+    <ul>
+      <li>Observation 1</li>
+      <li>Observation 2</li>
+    </ul>
+    
+    <h3>Conclusion</h3>
+    <p>[Your conclusion text]</p>
+    
     Important formatting instructions:
-    - Use proper HTML/Markdown formatting with <strong> tags for headings or important findings
+    - Use HTML elements for structure (<ul>, <li>, <strong>, <h3>, etc.)
     - Do not use asterisks (**) for emphasis - use proper HTML formatting instead
-    - Include this timestamp at the top of your report: {timestamp}
+    - For any urgent findings, use <span style="color:red"> to highlight them
     """
     
     try:
@@ -123,7 +143,25 @@ def analyze_ecg_image(image, vision_model="llama-3.2-90b-vision-preview"):
         ecg_analysis = re.sub(r'\*\*([^*]+)\*\*', r'<strong>\1</strong>', ecg_analysis)
         
         # Make sure all headers are properly formatted (with complete pattern)
-        ecg_analysis = re.sub(r'^(#+)\s+(.+)$', r'<strong>\2</strong>', ecg_analysis, flags=re.MULTILINE)
+        ecg_analysis = re.sub(r'^(#+)\s+(.+)$', r'<h3>\2</h3>', ecg_analysis, flags=re.MULTILINE)
+        
+        # If there's no HTML formatting at all, wrap in basic structure
+        if not re.search(r'<[^>]+>', ecg_analysis):
+            lines = ecg_analysis.split('\n')
+            formatted_html = "<h3>ECG Report</h3>\n<ul>\n"
+            
+            for line in lines:
+                if line.strip():
+                    # Try to identify key-value pairs
+                    match = re.match(r'^([^:]+):\s*(.+)$', line)
+                    if match:
+                        key, value = match.groups()
+                        formatted_html += f"  <li><strong>{key}:</strong> {value}</li>\n"
+                    else:
+                        formatted_html += f"  <li>{line}</li>\n"
+            
+            formatted_html += "</ul>"
+            ecg_analysis = formatted_html
         
         return ecg_analysis
     
@@ -153,26 +191,47 @@ PATIENT HISTORY:
 
 TIMESTAMP: {timestamp}
 
-Provide your assessment with proper formatting:
-<strong>Summary of Findings</strong>
-(Your summary here)
+Format your assessment using HTML elements for readability:
 
-<strong>Key Abnormalities</strong>
-(List any abnormalities here)
+<h3>Summary of Findings</h3>
+<ul>
+  <li>Finding 1</li>
+  <li>Finding 2</li>
+  <!-- etc. -->
+</ul>
 
-<strong>Potential Clinical Implications</strong>
-(Describe implications here)
+<h3>Key Abnormalities</h3>
+<ul>
+  <li>Abnormality 1</li>
+  <li>Abnormality 2 (if any)</li>
+  <!-- etc. -->
+</ul>
 
-<strong>Recommendation</strong>
-(Include urgency level)
+<h3>Potential Clinical Implications</h3>
+<ul>
+  <li>Implication 1</li>
+  <li>Implication 2</li>
+  <!-- etc. -->
+</ul>
 
-<strong>Differential Considerations</strong>
-(List differentials here)
+<h3>Recommendation</h3>
+<ul>
+  <li>Recommendation 1 (include urgency level)</li>
+  <li>Recommendation 2 (if any)</li>
+  <!-- etc. -->
+</ul>
+
+<h3>Differential Considerations</h3>
+<ul>
+  <li>Differential 1</li>
+  <li>Differential 2</li>
+  <!-- etc. -->
+</ul>
 
 Important formatting instructions:
-- Use proper HTML formatting with <strong> tags for headings
+- Use HTML elements for structure (<ul>, <li>, <strong>, <h3>, etc.)
 - Do not use asterisks (**) for emphasis - use proper HTML formatting instead
-- For any urgent findings, use <strong style="color:red"> to highlight them
+- For any urgent findings, use <span style="color:red"> to highlight them
 """
     else:
         prompt = f"""You are a highly trained cardiologist assistant. Based on the ECG analysis below, provide a comprehensive assessment of the patient's cardiac status. Indicate clearly if there are any concerning findings that require immediate medical attention.
@@ -182,26 +241,47 @@ ECG ANALYSIS:
 
 TIMESTAMP: {timestamp}
 
-Provide your assessment with proper formatting:
-<strong>Summary of Findings</strong>
-(Your summary here)
+Format your assessment using HTML elements for readability:
+
+<h3>Summary of Findings</h3>
+<ul>
+  <li>Finding 1</li>
+  <li>Finding 2</li>
+  <!-- etc. -->
+</ul>
 
-<strong>Key Abnormalities</strong>
-(List any abnormalities here)
+<h3>Key Abnormalities</h3>
+<ul>
+  <li>Abnormality 1</li>
+  <li>Abnormality 2 (if any)</li>
+  <!-- etc. -->
+</ul>
 
-<strong>Potential Clinical Implications</strong>
-(Describe implications here)
+<h3>Potential Clinical Implications</h3>
+<ul>
+  <li>Implication 1</li>
+  <li>Implication 2</li>
+  <!-- etc. -->
+</ul>
 
-<strong>Recommendation</strong>
-(Include urgency level)
+<h3>Recommendation</h3>
+<ul>
+  <li>Recommendation 1 (include urgency level)</li>
+  <li>Recommendation 2 (if any)</li>
+  <!-- etc. -->
+</ul>
 
-<strong>Differential Considerations</strong>
-(List differentials here)
+<h3>Differential Considerations</h3>
+<ul>
+  <li>Differential 1</li>
+  <li>Differential 2</li>
+  <!-- etc. -->
+</ul>
 
 Important formatting instructions:
-- Use proper HTML formatting with <strong> tags for headings
+- Use HTML elements for structure (<ul>, <li>, <strong>, <h3>, etc.)
 - Do not use asterisks (**) for emphasis - use proper HTML formatting instead
-- For any urgent findings, use <strong style="color:red"> to highlight them
+- For any urgent findings, use <span style="color:red"> to highlight them
 """
     
     try:
@@ -227,7 +307,53 @@ Important formatting instructions:
         assessment_text = re.sub(r'\*\*([^*]+)\*\*', r'<strong>\1</strong>', assessment_text)
         
         # Make sure all headers are properly formatted (with complete pattern)
-        assessment_text = re.sub(r'^(#+)\s+(.+)$', r'<strong>\2</strong>', assessment_text, flags=re.MULTILINE)
+        assessment_text = re.sub(r'^(#+)\s+(.+)$', r'<h3>\2</h3>', assessment_text, flags=re.MULTILINE)
+        
+        # If there's no HTML formatting at all, wrap in basic structure
+        if not re.search(r'<[^>]+>', assessment_text):
+            sections = [
+                "Summary of Findings", 
+                "Key Abnormalities", 
+                "Potential Clinical Implications", 
+                "Recommendation", 
+                "Differential Considerations"
+            ]
+            
+            # Split by sections and format
+            formatted_html = ""
+            current_section = "Summary of Findings"  # Default
+            content_lines = []
+            
+            lines = assessment_text.split('\n')
+            
+            for line in lines:
+                # Check if this line is a section header
+                is_section = False
+                for section in sections:
+                    if section.lower() in line.lower():
+                        # If we have collected content for a previous section, output it
+                        if content_lines:
+                            formatted_html += f"<h3>{current_section}</h3>\n<ul>\n"
+                            for content_line in content_lines:
+                                formatted_html += f"  <li>{content_line}</li>\n"
+                            formatted_html += "</ul>\n"
+                            content_lines = []
+                        
+                        current_section = section
+                        is_section = True
+                        break
+                
+                if not is_section and line.strip():
+                    content_lines.append(line.strip())
+            
+            # Don't forget the last section
+            if content_lines:
+                formatted_html += f"<h3>{current_section}</h3>\n<ul>\n"
+                for content_line in content_lines:
+                    formatted_html += f"  <li>{content_line}</li>\n"
+                formatted_html += "</ul>\n"
+            
+            assessment_text = formatted_html if formatted_html else assessment_text
         
         return assessment_text
     
@@ -256,13 +382,11 @@ MEDICAL ASSESSMENT:
 {assessment}
 
 TIMESTAMP: {timestamp}
-
 """
     
     if patient_history and patient_history.strip():
         context += f"""PATIENT HISTORY:
 {patient_history}
-
 """
     
     # Construct full chat history for context
@@ -315,14 +439,14 @@ with gr.Blocks(title="Cardiac ECG Analysis System", theme=gr.themes.Soft()) as a
             with gr.Row():
                 with gr.Column(scale=1):
                     # Input components
-                    with gr.Group():  # Replaced gr.Box()
+                    with gr.Group():  # Using Group instead of Box
                         gr.Markdown("### 📊 ECG Image")
                         ecg_image = gr.Image(type="pil", label="Upload ECG Image")
                         # Display fixed model info
                         gr.Markdown("**Vision Model:** llama-3.2-90b-vision-preview")
                         analyze_button = gr.Button("Analyze ECG Image", variant="primary")
                     
-                    with gr.Group():  # Replaced gr.Box()
+                    with gr.Group():  # Using Group instead of Box
                         gr.Markdown("### 📋 Patient Information")
                         patient_history_text = gr.Textbox(
                             lines=8, 
@@ -335,7 +459,7 @@ with gr.Blocks(title="Cardiac ECG Analysis System", theme=gr.themes.Soft()) as a
                         )
                         load_history_button = gr.Button("Load Patient History from File")
                     
-                    with gr.Group():  # Replaced gr.Box()
+                    with gr.Group():  # Using Group instead of Box
                         gr.Markdown("### 🧠 Assessment Settings")
                         # Display fixed model info
                         gr.Markdown("**Chat Model:** llama-3.3-70b-versatile")
@@ -343,18 +467,18 @@ with gr.Blocks(title="Cardiac ECG Analysis System", theme=gr.themes.Soft()) as a
                 
                 with gr.Column(scale=1):
                     # Output components
-                    with gr.Group():  # Replaced gr.Box()
+                    with gr.Group():  # Using Group instead of Box
                         gr.Markdown("### 📈 ECG Analysis Results")
                         ecg_analysis_output = gr.HTML(label="ECG Analysis", elem_id="ecg-analysis")
                     
-                    with gr.Group():  # Replaced gr.Box()
+                    with gr.Group():  # Using Group instead of Box
                         gr.Markdown("### 📝 Medical Assessment")
                         assessment_output = gr.HTML(label="Assessment", elem_id="assessment-output")
             
             gr.Markdown("## 👨‍⚕️ Doctor's Consultation")
             gr.Markdown("Ask follow-up questions about the patient's ECG results and medical condition.")
             
-            with gr.Group():  # Replaced gr.Box()
+            with gr.Group():  # Using Group instead of Box
                 chatbot = gr.Chatbot(label="Consultation", height=400)
                 with gr.Row():
                     message = gr.Textbox(