Update src/report_generator.py

src/report_generator.py

@@ -13,10 +13,46 @@ from reportlab.lib import colors
 from reportlab.lib.units import inch, cm
 from reportlab.platypus import PageBreak, KeepTogether
 
-from text_processors import format_conversation_history
 from model import orchestrator_chat
 
 
+# Function to format conversation history for report generation
+def format_conversation_history(history, patient_info=None):
+    """
+    Format the conversation history into a string suitable for LLM processing.
+    Optionally adds patient info at the beginning.
+    """
+    formatted_text = "# Medical Consultation\n\n"
+    
+    # Add patient info if provided
+    if patient_info:
+        formatted_text += "## Patient Information\n"
+        formatted_text += f"* Name: {patient_info.get('name', '')}\n"
+        formatted_text += f"* Age: {patient_info.get('age', '')}\n"
+        formatted_text += f"* Gender: {patient_info.get('gender', '')}\n\n"
+    
+    formatted_text += "## Conversation Transcript\n\n"
+    
+    for message in history:
+        role = message.get("role", "").strip()
+        content = message.get("content", "").strip()
+        
+        if not content:
+            continue  # Skip empty messages
+        
+        if role.lower() == "user":
+            formatted_text += f"PATIENT: {content}\n\n"
+        elif role.lower() == "assistant":
+            formatted_text += f"ASSISTANT: {content}\n\n"
+            # Include explanations which often contain diagnostic reasoning
+            if "explanation" in message and message["explanation"]:
+                explanation = message.get("explanation", "").strip()
+                if explanation:
+                    formatted_text += f"REASONING: {explanation}\n\n"
+    
+    return formatted_text
+
+
 # Function to encode images to base64 (similar to app.py)
 def get_image_base64(image_path):
     try:
@@ -459,7 +495,7 @@ def generate_and_download_report():
     # Step 4: Generate report
     elif st.session_state.report_step == 4:
         with st.spinner("Generating medical report..."):
-            # Format conversation with patient info
+            # Format conversation with patient info - using the local format_conversation_history
             conversation_text = format_conversation_history(
                 st.session_state.history, 
                 st.session_state.patient_info