Update app.py

app.py

@@ -2,11 +2,11 @@
 """
 Financial Health Score Service (FastAPI)
 
-Fully updated version:
-- Uses GPT-4o-mini with strict JSON schema output
+FINAL VERSION:
+- Uses GPT-4o-mini with JSON schema
+- Universal-safe JSON parsing (supports all SDK formats)
 - Correct OpenAI Chat Completions API
-- Fully valid `.choices[0].message.parsed`
-- Bulletproof Mongo fetch + normalization
+- Bulletproof error handling
 """
 
 import json
@@ -97,10 +97,7 @@ def normalize_transactions(txns):
     out = []
     for txn in txns:
         date_val = txn.get("date")
-        if isinstance(date_val, datetime):
-            date_str = date_val.date().isoformat()
-        else:
-            date_str = None
+        date_str = date_val.date().isoformat() if isinstance(date_val, datetime) else None
 
         # Resolve currency
         currency_code = None
@@ -140,10 +137,10 @@ You are a financial wellness expert. Use the user's budgets and last 30 days of
 to compute a financial health score from 0 to 100.
 
 Rules:
-- When mentioning amounts, ALWAYS prefix with currency code when available (e.g., "INR 10,000").
+- ALWAYS prefix currency amounts with currency code (e.g., "INR 10,000").
 - Keep explanation short (1–2 sentences).
-- Score must reflect spending habits, budget usage, and financial balance.
-- Respond ONLY using the JSON schema enforced below.
+- Score must reflect spending, budget control, and overall balance.
+- Respond ONLY in the JSON schema required below.
 
 Budgets:
 {json.dumps(budgets, indent=2)}
@@ -193,7 +190,7 @@ def financial_score(payload: ScoreRequest):
     prompt = build_prompt(budgets, transactions)
 
     # ===========================
-    # STRICT GPT-4o-mini JSON SCHEMA
+    # GPT-4o-mini WITH JSON SCHEMA
     # ===========================
     try:
         response = openai.chat.completions.create(
@@ -222,18 +219,30 @@ def financial_score(payload: ScoreRequest):
     except Exception as exc:
         raise HTTPException(status_code=502, detail=f"OpenAI request failed: {exc}")
 
-    # SAFE PARSE JSON
-    # SAFE PARSE JSON (latest OpenAI format)
+    # ===========================
+    # UNIVERSAL JSON PARSING (supports ALL SDK formats)
+    # ===========================
     try:
-        raw_json = response.choices[0].message.content[0].text
+        content = response.choices[0].message.content
+
+        # Case A — list of dicts (new SDK)
+        if isinstance(content, list):
+            raw_json = content[0].get("text", content[0])
+        else:
+            # Case B — plain string (older SDK)
+            raw_json = content
+
+        if not isinstance(raw_json, str):
+            raw_json = str(raw_json)
+
         parsed = json.loads(raw_json)
+
     except Exception as e:
         raise HTTPException(
             status_code=502,
             detail=f"Could not parse JSON output: {e}"
         )
 
-
     # Score normalization
     score_val = parsed.get("score", 0)
     try: