Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,7 +15,6 @@ from sendgrid_service import SendGridService
 from report_generator import (
     extract_medical_json,
     build_medical_report,
-    format_conversation_history,
     generate_and_download_report,
     show_email_form
 )
@@ -646,4 +645,51 @@ st.markdown("""
 <div class="footer-text">
     For informational purposes only. Not a substitute for professional medical advice.
 </div>
-""", unsafe_allow_html=True)
+""", unsafe_allow_html=True)
+
+# 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
+
+# This ensures app.py can be used as a direct entry point for deployment platforms like Hugging Face
+if __name__ == "__main__":
+    # Add the current directory to the Python path if needed
+    import os
+    import sys
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    if current_dir not in sys.path:
+        sys.path.insert(0, current_dir)
+    
+    # The app is already running due to the Streamlit script structure