Create app.py

app.py

@@ -0,0 +1,97 @@
+def _generate_medical_report(self, has_tumor: bool, tumor_size: str, metadata: PatientMetadata) -> str:
+    """Generate a standardized medical report using Mistral."""
+    prompt = f"""<s>[INST] Generate a structured medical report for a breast imaging scan using the following format exactly.
+Keep sections consistent and use proper medical terminology. Be concise yet thorough.
+
+EXAMINATION PERFORMED:
+- Diagnostic Digital Mammogram
+- Date: {datetime.now().strftime('%B %d, %Y')}
+
+CLINICAL FINDINGS:
+Primary Finding: {'Abnormal area detected' if has_tumor else 'No abnormalities detected'}
+{f'Lesion Size: {tumor_size} cm' if has_tumor else ''}
+
+PATIENT HISTORY:
+- Age: {metadata.age} years
+- Menopausal Status: {metadata.menopause_status}
+- Previous Mammogram: {'Yes' if metadata.previous_mammogram else 'No'}
+- Breast Tissue Density: {metadata.breast_density}
+
+RISK FACTORS:
+{f'• Family History: {"Present" if metadata.family_history else "None"}'}
+• Smoking Status: {metadata.smoking_status}
+• Hormone Therapy: {'Yes' if metadata.hormone_therapy else 'No'}
+
+Please generate a report with these exact sections:
+
+1. DETAILED FINDINGS
+[Describe the mammographic findings in detail, including location, characteristics, and comparison with any previous studies if available]
+
+2. IMPRESSION
+[Provide a clear assessment using BI-RADS classification if applicable]
+
+3. RECOMMENDATIONS
+[List specific follow-up actions and timeline]
+
+4. ADDITIONAL NOTES
+[Include any relevant screening considerations or risk-management suggestions]
+
+Format each section consistently and maintain professional medical terminology throughout. [/INST]</s>"""
+
+    # Generate response using Mistral
+    response = self.report_generator.text_generation(
+        prompt,
+        max_new_tokens=800,  # Increased for comprehensive report
+        temperature=0.3,     # Low temperature for consistency
+        top_p=0.9,
+        repetition_penalty=1.1,
+        do_sample=True,
+        seed=42
+    )
+        
+    # Post-process the response to ensure consistent formatting
+    formatted_response = f"""MAMMOGRAM REPORT
+Date: {datetime.now().strftime('%B %d, %Y')}
+----------------------------------------
+
+{response.strip()}
+
+----------------------------------------
+NOTE: This report was generated using AI assistance and should be reviewed by a qualified healthcare professional.
+All findings and recommendations should be clinically verified."""
+
+    return formatted_response
+
+# Update the analyze method to ensure consistent output formatting
+def analyze(self, image: Image.Image) -> str:
+    """Main analysis pipeline with standardized output."""
+    try:
+        processed_image = self._process_image(image)
+        metadata = self._generate_synthetic_metadata()
+
+        # Detect tumor
+        tumor_result = self.tumor_classifier(processed_image)
+        has_tumor = tumor_result[0]['label'] == 'tumor'
+        tumor_confidence = tumor_result[0]['score']
+
+        # Measure size if tumor detected
+        size_result = self.size_classifier(processed_image)
+        tumor_size = size_result[0]['label'].replace('tumor-', '')
+
+        # Generate report
+        report = self._generate_medical_report(has_tumor, tumor_size, metadata)
+
+        return f"""BREAST IMAGING ANALYSIS REPORT
+========================================
+
+INITIAL SCAN ASSESSMENT:
+{'⚠️ ABNORMAL FINDING DETECTED' if has_tumor else '✓ NO ABNORMALITIES DETECTED'}
+Detection Confidence: {tumor_confidence:.2%}
+{f'Estimated Size: {tumor_size} cm' if has_tumor else ''}
+
+----------------------------------------
+{report}"""
+
+    except Exception as e:
+        import traceback
+        return f"ERROR IN ANALYSIS:\n{str(e)}\n\nDebug Information:\n{traceback.format_exc()}"