Update app.py

app.py

@@ -21,7 +21,7 @@ class MediClearBackend:
 
     @staticmethod
     def analyze_medical_image(image_path):
-        """Step 1: Extract technical info"""
+        """Step 1: Extract technical info from a single image"""
         try:
             base64_image = MediClearBackend.encode_image(image_path)
             
@@ -55,8 +55,29 @@ class MediClearBackend:
             raise Exception(f"Image Analysis Error: {str(e)}")
 
     @staticmethod
-    def summarize_for_patient(technical_text):
-        """Step 2: Convert technical info to patient-friendly summary"""
+    def extract_text_from_multiple_images(image_files):
+        """Extract and combine text from multiple medical images"""
+        combined_text = "COMBINED MEDICAL REPORTS ANALYSIS:\n\n"
+        
+        for index, image_file in enumerate(image_files, 1):
+            try:
+                filename = Path(image_file.name).name
+                combined_text += f"=== REPORT {index}: {filename} ===\n"
+                
+                # Extract text from this image
+                image_text = MediClearBackend.analyze_medical_image(image_file.name)
+                combined_text += f"{image_text}\n\n"
+                combined_text += "=" * 50 + "\n\n"
+                
+            except Exception as e:
+                combined_text += f"❌ Failed to process {filename}: {str(e)}\n\n"
+                combined_text += "=" * 50 + "\n\n"
+        
+        return combined_text
+
+    @staticmethod
+    def summarize_combined_reports(combined_text):
+        """Step 2: Convert combined technical info to patient-friendly summary"""
         try:
             if not HYPERBOLIC_KEY:
                 raise Exception("Hyperbolic API key not configured")
@@ -68,19 +89,24 @@ 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.
+            Your role is to analyze multiple medical reports and provide a comprehensive, unified analysis to patients in simple, reassuring, everyday language.
             Never mention the doctors name whatsoever and never ask any counter questions.
+            
+            You will receive multiple medical reports combined together. Analyze them as a complete patient case and provide:
+            
             Structure:
-            1. Warm greeting.
-            2. Overall assessment.
-            3. Key findings (compared to normal).
-            4. Reassuring closing.
-            Speak directly to the patient."""
+            1. Warm greeting acknowledging this is a comprehensive analysis of multiple reports.
+            2. Overall unified assessment of the patient's condition across all reports.
+            3. Key findings from all reports, highlighting patterns, consistencies, or important observations.
+            4. Comprehensive recommendations based on the complete picture.
+            5. Reassuring closing.
+            
+            Speak directly to the patient and provide a holistic view of their medical situation."""
 
             data = {
                 "messages": [
                     {"role": "system", "content": system_prompt},
-                    {"role": "user", "content": technical_text}
+                    {"role": "user", "content": combined_text}
                 ],
                 "model": "moonshotai/Kimi-K2-Instruct",  # This is the Kimi K2 model
                 "max_tokens": 4096,
@@ -106,23 +132,18 @@ def process_single_image(image_file):
         technical_data = backend.analyze_medical_image(image_file.name)
         
         # Step 2: Patient-friendly summary with Kimi K2
-        final_report = backend.summarize_for_patient(technical_data)
+        final_report = backend.summarize_combined_reports(technical_data)
         
         # Format the final report as Markdown
         filename = Path(image_file.name).name
         formatted_report = f"""
 # 🩺 MEDICAL ANALYSIS REPORT: {filename}
-
 ---
-
 ## 🔬 Technical Analysis
 *Medical image processing complete*
-
 ## 📝 Patient Report Summary
 {final_report}
-
 ---
-
 > ✅ **Analysis Complete** - This tool provides AI-powered insights and is not a substitute for professional medical diagnosis.
 """
         return formatted_report
@@ -131,46 +152,49 @@ def process_single_image(image_file):
         return f"## ❌ Error\n**Error processing image:** {str(e)}"
 
 def process_multiple_images(image_files):
-    """Process multiple medical images and return combined analysis"""
+    """Process multiple medical images and return a SINGLE combined analysis"""
     backend = MediClearBackend()
     
     if not image_files:
         return "❌ Please upload at least one medical image."
     
-    full_report = f"""# 🩺 SEHATSCAN - COMPREHENSIVE MEDICAL ANALYSIS
+    try:
+        # Step 1: Extract and combine text from all images
+        combined_text = backend.extract_text_from_multiple_images(image_files)
+        
+        # Step 2: Generate a single comprehensive summary from combined text
+        comprehensive_summary = backend.summarize_combined_reports(combined_text)
+        
+        # Format as a single comprehensive report
+        full_report = f"""# 🩺 COMPREHENSIVE MEDICAL ANALYSIS
 
-**🔬 Step 1: Groq (Llama 4 Scout) | 📝 Step 2: Hyperbolic (Kimi K2)**
+## 📊 Analysis of {len(image_files)} Medical Reports
+
+**Combined insights from all uploaded medical images:**
 
 ---
 
-"""
-    total = len(image_files)
-    
-    for index, image_file in enumerate(image_files, 1):
-        try:
-            filename = Path(image_file.name).name
-            full_report += f"""
-## 📄 IMAGE {index}/{total}: {filename}
+## 📝 Comprehensive Patient Summary
+
+{comprehensive_summary}
 
 ---
+
+## 📋 Reports Analyzed:
 """
-            # 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 += "---\n"
-            full_report += f"**✅ Analysis complete for {filename}**\n\n"
-            
-        except Exception as e:
-            full_report += f"## ❌ Processing Failed\n**Failed to process {filename}:** {str(e)}\n\n---\n"
-    
-    full_report += """
+        # List all analyzed files
+        for index, image_file in enumerate(image_files, 1):
+            filename = Path(image_file.name).name
+            full_report += f"- **Report {index}:** {filename}\n"
+        
+        full_report += """
 ---
-
-> 🏁 **All analyses complete.** Remember: This tool provides AI-powered insights and is not a substitute for professional medical diagnosis.
+> 🏁 **Comprehensive Analysis Complete** - This unified analysis provides insights across all your medical reports. Remember: This tool provides AI-powered insights and is not a substitute for professional medical diagnosis.
 """
-    return full_report
+        return full_report
+        
+    except Exception as e:
+        return f"## ❌ Processing Error\n**Failed to process multiple images:** {str(e)}"
 
 def create_gradio_interface():
     """Create the Gradio interface"""
@@ -209,7 +233,7 @@ def create_gradio_interface():
         border: 1px solid #4CAF50 !important;
         border-radius: 8px !important;
         padding: 15px !important;
-        margin: 10px 0 !important;
+        margin: 10x 0 !important;
     }
     .processing {
         background: #FFD60A20 !important;
@@ -298,7 +322,7 @@ def create_gradio_interface():
                     
                     # Multiple image upload
                     multiple_images = gr.File(
-                        label="Batch Image Analysis (Multiple)",
+                        label="Batch Image Analysis (Multiple) - Combined Report",
                         file_types=[".jpg", ".jpeg", ".png", ".bmp"],
                         file_count="multiple",
                         type="filepath"
@@ -320,9 +344,15 @@ def create_gradio_interface():
                     # Clear button
                     clear_btn = gr.Button("🗑️ Clear All", variant="stop")
                 
-                # Model Info
+                # Processing Info
                 with gr.Group(elem_classes="processing"):
-                    pass
+                    gr.Markdown(
+                        """
+                        ### 🔄 Processing Information
+                        **Single Image:** Individual analysis
+                        **Multiple Images:** Combined comprehensive analysis from all reports
+                        """
+                    )
                 
                 # Disclaimer
                 with gr.Group(elem_classes="warning"):
@@ -405,7 +435,7 @@ def create_gradio_interface():
         gr.Markdown("""
         1. **Upload** a medical report image 
         2. **Click Analyze** to get AI-powered medical insights  
-        3. **Review** the patient-friendly explanation 
+        3. **For multiple images:** Get a single combined analysis of all reports
         4. **Always consult** with healthcare professionals for medical decisions
         """)