Update gpt_analyzer.py

gpt_analyzer.py

@@ -11,11 +11,12 @@ 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 that precisely matches this structure. Fields must exactly match the available report fields:
+
+        You must return a response that exactly matches this JSON structure. Fields must exactly match the available report fields. DO NOT include any other text in your response except the JSON:
+
         {{
             "required_reports": [
                 {{
@@ -40,28 +41,32 @@ class GPTAnalyzer:
             "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
+        - Return only the JSON, no other text
         """
-    
+
         try:
             response = self.client.chat.completions.create(
                 messages=[
                     {
                         "role": "system",
-                        "content": "You are a healthcare data analyst expert who understands hospital information systems. Always return exact matching report names and field names."
+                        "content": "You are a healthcare data analyst expert who understands hospital information systems. Always return exact matching report names and field names. Return only JSON, no other text."
                     },
                     {
                         "role": "user",
                         "content": prompt
                     }
                 ],
-                model="gpt-4",
-                response_format={ "type": "json_object" }
+                model="gpt-4o-mini",
             )
-            return json.loads(response.choices[0].message.content)
+            # Extract the JSON string from the response
+            response_text = response.choices[0].message.content.strip()
+            # Parse the JSON string
+            return json.loads(response_text)
+        except json.JSONDecodeError as e:
+            return {"error": f"Failed to parse GPT response as JSON: {str(e)}"}
         except Exception as e:
             return {"error": str(e)}