Update app/core/analyzer.py

app/core/analyzer.py

@@ -1,53 +1,53 @@
-from typing import Dict, List
-import json
-import google.generativeai as genai
-from fastapi import HTTPException
-from ..utils.prompt_templates import generate_traditional_prompt, generate_body_based_prompt
-
-class SymptomAnalyzer:
-    def __init__(self):
-        self.model = genai.GenerativeModel('gemini-1.5-flash')
-        
-    async def analyze_traditional(self, symptoms: List[str], age: int, gender: str, duration: str) -> Dict:
-        try:
-            prompt = generate_traditional_prompt(symptoms, age, gender, duration)
-            response = self.model.generate_content(prompt)
-            
-            try:
-                parsed_response = json.loads(response.text)
-                return parsed_response
-            except json.JSONDecodeError as e:
-                cleaned_response = response.text.strip().strip('`').strip('json')
-                try:
-                    parsed_response = json.loads(cleaned_response)
-                    return parsed_response
-                except json.JSONDecodeError:
-                    raise HTTPException(
-                        status_code=500,
-                        detail="Failed to parse response into valid JSON"
-                    )
-                
-        except Exception as e:
-            raise HTTPException(status_code=500, detail=f"Analysis error: {str(e)}")
-
-    async def analyze_body_based(self, data: Dict) -> Dict:
-        try:
-            prompt = generate_body_based_prompt(data)
-            response = self.model.generate_content(prompt)
-            
-            try:
-                parsed_response = json.loads(response.text)
-                return parsed_response
-            except json.JSONDecodeError as e:
-                cleaned_response = response.text.strip().strip('`').strip('json')
-                try:
-                    parsed_response = json.loads(cleaned_response)
-                    return parsed_response
-                except json.JSONDecodeError:
-                    raise HTTPException(
-                        status_code=500,
-                        detail="Failed to parse response into valid JSON"
-                    )
-                
-        except Exception as e:
+from typing import Dict, List
+import json
+import google.generativeai as genai
+from fastapi import HTTPException
+from ..utils.prompt_templates import generate_traditional_prompt, generate_body_based_prompt
+
+class SymptomAnalyzer:
+    def __init__(self):
+        self.model = genai.GenerativeModel('gemini-2.5-flash')
+        
+    async def analyze_traditional(self, symptoms: List[str], age: int, gender: str, duration: str) -> Dict:
+        try:
+            prompt = generate_traditional_prompt(symptoms, age, gender, duration)
+            response = self.model.generate_content(prompt)
+            
+            try:
+                parsed_response = json.loads(response.text)
+                return parsed_response
+            except json.JSONDecodeError as e:
+                cleaned_response = response.text.strip().strip('`').strip('json')
+                try:
+                    parsed_response = json.loads(cleaned_response)
+                    return parsed_response
+                except json.JSONDecodeError:
+                    raise HTTPException(
+                        status_code=500,
+                        detail="Failed to parse response into valid JSON"
+                    )
+                
+        except Exception as e:
+            raise HTTPException(status_code=500, detail=f"Analysis error: {str(e)}")
+
+    async def analyze_body_based(self, data: Dict) -> Dict:
+        try:
+            prompt = generate_body_based_prompt(data)
+            response = self.model.generate_content(prompt)
+            
+            try:
+                parsed_response = json.loads(response.text)
+                return parsed_response
+            except json.JSONDecodeError as e:
+                cleaned_response = response.text.strip().strip('`').strip('json')
+                try:
+                    parsed_response = json.loads(cleaned_response)
+                    return parsed_response
+                except json.JSONDecodeError:
+                    raise HTTPException(
+                        status_code=500,
+                        detail="Failed to parse response into valid JSON"
+                    )
+                
+        except Exception as e:
             raise HTTPException(status_code=500, detail=f"Analysis error: {str(e)}")