Update app.py

app.py

@@ -68,7 +68,7 @@ def extract_medical_data(df: pd.DataFrame) -> Dict[str, Any]:
     
     return medical_data
 
-def identify_red_flags(records: List[Dict[str, Any]]) -> Dict[str, List[str]]:
+def identify_red_flags(records: List[Dict[str, Any]]) -> Dict[str, Any]:
     """Identify potential red flags in medical records"""
     red_flags = {
         'symptoms': defaultdict(list),
@@ -116,7 +116,7 @@ def identify_red_flags(records: List[Dict[str, Any]]) -> Dict[str, List[str]]:
     
     return red_flags
 
-def generate_analysis_prompt(booking: str, records: List[Dict[str, Any]], red_flags: Dict[str, Any]]) -> str:
+def generate_analysis_prompt(booking: str, records: List[Dict[str, Any]], red_flags: Dict[str, Any]) -> str:
     """Generate structured prompt for analysis"""
     records_text = "\n".join(
         f"- {r['form_name']}: {r['form_item']} = {r['response']} ({r['date']} by {r['interviewer']})\n  {r['description']}"