Update app.py

app.py

@@ -69,7 +69,7 @@ class MediClearBackend:
             
             system_prompt = """You are a senior medical doctor with 50 years of experience. 
             Your role is to explain medical reports to patients in simple, reassuring, everyday language.
-            Never mention the doctors name whatsoever.
+            Never mention the doctors name whatsoever and never ask any counter questions.
             Structure:
             1. Warm greeting.
             2. Overall assessment.
@@ -108,23 +108,27 @@ def process_single_image(image_file):
         # Step 2: Patient-friendly summary with Kimi K2
         final_report = backend.summarize_for_patient(technical_data)
         
-        # Format the final report
+        # Format the final report as Markdown
         filename = Path(image_file.name).name
         formatted_report = f"""
-{'='*60}
-🩺 MEDICAL ANALYSIS REPORT: {filename}
-{'='*60}
-🔬 Technical Analysis: Medical image processed
-📝 Patient Report Summary: 
+# 🩺 MEDICAL ANALYSIS REPORT: {filename}
+
+---
+
+## 🔬 Technical Analysis
+*Medical image processing complete*
+
+## 📝 Patient Report Summary
 {final_report}
-{'='*60}
-✅ Analysis Complete - This tool provides AI-powered insights and is not a substitute for professional medical diagnosis.
-{'='*60}
+
+---
+
+> ✅ **Analysis Complete** - This tool provides AI-powered insights and is not a substitute for professional medical diagnosis.
 """
         return formatted_report
         
     except Exception as e:
-        return f"❌ Error processing image: {str(e)}"
+        return f"## ❌ Error\n**Error processing image:** {str(e)}"
 
 def process_multiple_images(image_files):
     """Process multiple medical images and return combined analysis"""
@@ -133,31 +137,39 @@ def process_multiple_images(image_files):
     if not image_files:
         return "❌ Please upload at least one medical image."
     
-    full_report = f"🩺 SEHATSCAN - COMPREHENSIVE MEDICAL ANALYSIS\n"
-    full_report += f"🔬 Step 1: Groq (Llama 4 Scout) | 📝 Step 2: Hyperbolic (Kimi K2)\n"
-    full_report += f"{'='*70}\n\n"
-    
+    full_report = f"""# 🩺 SEHATSCAN - COMPREHENSIVE MEDICAL ANALYSIS
+
+**🔬 Step 1: Groq (Llama 4 Scout) | 📝 Step 2: Hyperbolic (Kimi K2)**
+
+---
+
+"""
     total = len(image_files)
     
     for index, image_file in enumerate(image_files, 1):
         try:
             filename = Path(image_file.name).name
-            full_report += f"\n📄 IMAGE {index}/{total}: {filename}\n{'-'*50}\n"
-            
+            full_report += f"""
+## 📄 IMAGE {index}/{total}: {filename}
+
+---
+"""
             # Process the image
             technical_data = backend.analyze_medical_image(image_file.name)
             patient_summary = backend.summarize_for_patient(technical_data)
             
             full_report += f"{patient_summary}\n\n"
-            full_report += "✅ Analysis complete for this image.\n\n"
+            full_report += "---\n"
+            full_report += f"**✅ Analysis complete for {filename}**\n\n"
             
         except Exception as e:
-            full_report += f"❌ Failed to process {filename}: {str(e)}\n\n"
-    
-    full_report += f"{'='*70}\n"
-    full_report += "🏁 All analyses complete. Remember: This tool provides AI-powered insights and is not a substitute for professional medical diagnosis.\n"
-    full_report += f"{'='*70}"
+            full_report += f"## ❌ Processing Failed\n**Failed to process {filename}:** {str(e)}\n\n---\n"
     
+    full_report += """
+---
+
+> 🏁 **All analyses complete.** Remember: This tool provides AI-powered insights and is not a substitute for professional medical diagnosis.
+"""
     return full_report
 
 def create_gradio_interface():
@@ -167,6 +179,7 @@ def create_gradio_interface():
     custom_css = """
     .gradio-container {
         background: linear-gradient(135deg, #1A1A1A 0%, #2D2D2D 100%);
+        font-family: 'Segoe UI', system-ui, sans-serif;
     }
     .dark {
         background: transparent !important;
@@ -205,6 +218,48 @@ def create_gradio_interface():
         padding: 15px !important;
         margin: 10px 0 !important;
     }
+    .markdown-output {
+        background: #1a1a1a !important;
+        border-radius: 8px !important;
+        padding: 20px !important;
+    }
+    .markdown-output h1, .markdown-output h2, .markdown-output h3 {
+        color: #FFD60A !important;
+        margin-top: 1em !important;
+        margin-bottom: 0.5em !important;
+    }
+    .markdown-output p {
+        color: #E8E8E8 !important;
+        line-height: 1.6 !important;
+    }
+    .markdown-output ul, .markdown-output ol {
+        color: #E8E8E8 !important;
+        margin-left: 20px !important;
+    }
+    .markdown-output blockquote {
+        border-left: 4px solid #FFD60A !important;
+        margin: 1em 0 !important;
+        padding-left: 1em !important;
+        color: #B0B0B0 !important;
+        font-style: italic !important;
+    }
+    .markdown-output code {
+        background: #404040 !important;
+        padding: 2px 6px !important;
+        border-radius: 4px !important;
+        color: #E8E8E8 !important;
+    }
+    .markdown-output pre {
+        background: #404040 !important;
+        padding: 15px !important;
+        border-radius: 8px !important;
+        overflow-x: auto !important;
+    }
+    .markdown-output hr {
+        border: none !important;
+        border-top: 2px solid #404040 !important;
+        margin: 2em 0 !important;
+    }
     """
     
     with gr.Blocks(
@@ -269,7 +324,7 @@ def create_gradio_interface():
                 with gr.Group(elem_classes="processing"):
                     gr.Markdown(
                         """
-                        🤖 **AI Models Used:**
+                        ### 🤖 AI Models Used:
                         - **Step 1**: Groq + Llama 4 Scout (Medical Image Analysis)
                         - **Step 2**: Hyperbolic + Kimi K2 (Patient Communication)
                         """
@@ -279,7 +334,7 @@ def create_gradio_interface():
                 with gr.Group(elem_classes="warning"):
                     gr.Markdown(
                         """
-                        ⚠️ **Medical Disclaimer**
+                        ### ⚠️ Medical Disclaimer
                         This tool provides AI-powered insights and is not a substitute for professional medical diagnosis, 
                         treatment, or medical advice. Always consult with qualified healthcare providers for medical decisions.
                         """
@@ -290,12 +345,11 @@ def create_gradio_interface():
                 with gr.Group(elem_classes="panel"):
                     gr.Markdown("### 📋 Doctor's Analysis Report")
                     
-                    output_text = gr.Textbox(
+                    # Changed from Textbox to Markdown for pure markdown rendering
+                    output_markdown = gr.Markdown(
                         label="Analysis Results",
-                        lines=25,
-                        max_lines=50,
-                        show_copy_button=True,
-                        container=False
+                        value="*Upload a medical image and click analyze to see the results here...*",
+                        elem_classes="markdown-output"
                     )
                     
                     # Output actions
@@ -313,59 +367,50 @@ def create_gradio_interface():
                     """
                 )
         
-        # Event handlers
+        # Event handlers - updated to use Markdown output
         analyze_single_btn.click(
             fn=process_single_image,
             inputs=[single_image],
-            outputs=[output_text],
+            outputs=[output_markdown],
             api_name="analyze_single"
         )
         
         analyze_batch_btn.click(
             fn=process_multiple_images,
             inputs=[multiple_images],
-            outputs=[output_text],
+            outputs=[output_markdown],
             api_name="analyze_batch"
         )
         
         # Clear functionality
         def clear_all():
-            return None, None, ""
+            return None, None, "*Upload a medical image and click analyze to see the results here...*"
         
         clear_btn.click(
             fn=clear_all,
-            outputs=[single_image, multiple_images, output_text]
+            outputs=[single_image, multiple_images, output_markdown]
         )
         
-        # Copy functionality
-        def copy_to_clipboard(text):
-            return gr.update(value=text)
-        
+        # Copy functionality - for Markdown we'll use a different approach
         copy_btn.click(
-            fn=copy_to_clipboard,
-            inputs=[output_text],
-            outputs=[output_text]
+            fn=lambda: gr.Info("Use the copy button in the Markdown output or select and copy the text directly"),
+            outputs=None
         )
         
         # Export functionality
-        def export_report(text):
-            if text:
-                timestamp = time.strftime("%Y%m%d-%H%M%S")
-                filename = f"sehatscan_report_{timestamp}.txt"
-                return f"✅ Report ready! You can copy the text above and save it as '{filename}'"
-            return "❌ No analysis to export."
+        def export_report():
+            return gr.Info("To export, please select and copy the Markdown content above, then paste it into a text file.")
         
         export_btn.click(
             fn=export_report,
-            inputs=[output_text],
-            outputs=[output_text]
+            outputs=None
         )
         
         # Instructions
         gr.Markdown("### 🎯 How to use:")
         gr.Markdown("""
         1. **Upload** a medical image (X-ray, MRI, CT scan, ultrasound, etc.)
-        2. **Click Analyze** to get AI-powered medical insights
+        2. **Click Analyze** to get AI-powered medical insights  
         3. **Review** the patient-friendly explanation (powered by Kimi K2)
         4. **Always consult** with healthcare professionals for medical decisions
         """)