V7

src/report_generator.py

@@ -16,7 +16,7 @@ import requests
 
 from model import orchestrator_chat
 from utils import format_conversation_history
-from prompts import REPORT_GENERATION_PROMPT
+from session_state import get_full_history
 
 
 # Function to fetch current date and time from an API
@@ -109,9 +109,24 @@ def extract_medical_json(conversation_text: str) -> dict:
     Extract medical report data from conversation text into structured JSON.
     Uses the existing orchestrator_chat model.
     """
-    # Use imported prompt
-    system_prompt = REPORT_GENERATION_PROMPT
-    
+    system_prompt = """
+You are an expert medical report generator. Extract ALL info from the patient-assistant conversation into this JSON schema:
+{
+  "patient": {"name":"","age":"","gender":""},
+  "visit_date":"YYYY-MM-DD",
+  "chief_complaint":"",
+  "history_of_present_illness":"",
+  "past_medical_history":"",
+  "medications":"",
+  "allergies":"",
+  "examination":"",
+  "diagnosis":"",
+  "recommendations":"",
+  "reasoning":"",
+  "sources":""
+}
+Do NOT invent any data. Return ONLY the JSON object.
+"""
     # Create a fake history with the system prompt as a system message
     history = [
         {"role": "system", "content": system_prompt}