Update gpt_analyzer.py

gpt_analyzer.py

@@ -5,9 +5,12 @@ from typing import Dict, List, Any
 
 class GPTAnalyzer:
     def __init__(self, api_key: str):
+        print("Initializing GPT Analyzer")
         self.client = OpenAI(api_key=api_key)
     
     def analyze_request(self, request_text: str, available_categories: List[str]) -> Dict[str, Any]:
+        print(f"Analyzing request with available categories: {available_categories}")
+        
         prompt = f"""
         As a hospital data analyst, analyze this data request:
         "{request_text}"
@@ -27,15 +30,6 @@ class GPTAnalyzer:
                         "date_range": "January 2024",
                         "other_filters": []
                     }}
-                }},
-                {{
-                    "category": "OPD",
-                    "report_type": "waiting_times",
-                    "fields_needed": ["HN", "Wait_Time_Statistics"],
-                    "filters": {{
-                        "date_range": "January 2024",
-                        "other_filters": []
-                    }}
                 }}
             ],
             "interpretation": "Brief explanation of what data is needed",
@@ -49,6 +43,7 @@ class GPTAnalyzer:
         """
 
         try:
+            print("Sending request to GPT")
             response = self.client.chat.completions.create(
                 messages=[
                     {
@@ -62,11 +57,15 @@ class GPTAnalyzer:
                 ],
                 model="gpt-4o-mini",
             )
-            # Extract the JSON string from the response
+            
             response_text = response.choices[0].message.content.strip()
-            # Parse the JSON string
+            print(f"GPT Response received: {response_text}")
+            
             return json.loads(response_text)
+            
         except json.JSONDecodeError as e:
+            print(f"JSON Parse Error: {str(e)}")
             return {"error": f"Failed to parse GPT response as JSON: {str(e)}"}
         except Exception as e:
+            print(f"General Error: {str(e)}")
             return {"error": str(e)}