Update gpt_analyzer.py

gpt_analyzer.py

@@ -11,41 +11,55 @@ class GPTAnalyzer:
         prompt = f"""
         As a hospital data analyst, analyze this data request:
         "{request_text}"
-
+    
         Consider these available data sources in hospital Web Data system:
         {json.dumps(available_categories, indent=2, ensure_ascii=False)}
-
-        Return JSON with this structure:
+    
+        Return JSON that precisely matches this structure. Fields must exactly match the available report fields:
         {{
             "required_reports": [
                 {{
-                    "category": "Which category (OPD/IPD/PCT/etc)",
-                    "report_type": "Specific report name needed",
-                    "fields_needed": ["List of required fields"],
+                    "category": "OPD",
+                    "report_type": "diagnosis",
+                    "fields_needed": ["HN", "Patient_Name", "VisitDate", "Doctor_Name", "ICD_Code", "ICD_Name"],
+                    "filters": {{
+                        "date_range": "January 2024",
+                        "other_filters": []
+                    }}
+                }},
+                {{
+                    "category": "OPD",
+                    "report_type": "waiting_times",
+                    "fields_needed": ["HN", "Wait_Time_Statistics"],
                     "filters": {{
-                        "date_range": "Required date range if specified",
-                        "other_filters": ["Other filters needed"]
+                        "date_range": "January 2024",
+                        "other_filters": []
                     }}
                 }}
             ],
             "interpretation": "Brief explanation of what data is needed",
             "confidence_score": "HIGH/MEDIUM/LOW"
         }}
+    
+        IMPORTANT:
+        - report_type must match exactly one of the available report names
+        - fields_needed must match exactly the field names in the reports
+        - The response must maintain this exact structure
         """
-
+    
         try:
             response = self.client.chat.completions.create(
                 messages=[
                     {
                         "role": "system",
-                        "content": "You are a healthcare data analyst expert who understands hospital information systems."
+                        "content": "You are a healthcare data analyst expert who understands hospital information systems. Always return exact matching report names and field names."
                     },
                     {
                         "role": "user",
                         "content": prompt
                     }
                 ],
-                model="gpt-4o-mini",
+                model="gpt-4",
                 response_format={ "type": "json_object" }
             )
             return json.loads(response.choices[0].message.content)