Update main.py

main.py

@@ -201,7 +201,7 @@ def extract_fields_with_sarvam(raw_text: str, doc_type: str) -> dict:
     }
 
     if not SARVAM_API_KEY:
-        return {**empty_fields, "ocr_error": "SARVAM_API_KEY not configured in Space secrets"}
+        return {**empty_fields, "ocr_error": "SARVAM_API_KEY not configured"}
 
     system_prompt = (
         "You are a document field extractor for Indian identity documents. "
@@ -210,9 +210,9 @@ def extract_fields_with_sarvam(raw_text: str, doc_type: str) -> dict:
         "Fields to extract:\n"
         "  - name: Full name of the document holder\n"
         "  - address: Full address (combine multi-line if needed)\n"
-        "  - date_of_birth: In DD/MM/YYYY or YYYY format; use year_of_birth key if only year is present\n"
+        "  - date_of_birth: In DD/MM/YYYY or YYYY format\n"
         "  - gender: Male / Female / Transgender\n"
-        "  - id_number: Aadhaar=12-digit number, PAN=10-char alphanumeric, Voter ID=alphanumeric card number\n"
+        "  - id_number: Aadhaar=12-digit, PAN=10-char alphanumeric, Voter ID=alphanumeric\n"
         "  - mobile_number: 10-digit mobile number if present\n\n"
         "Use null for any field not found. Return a flat JSON object only."
     )
@@ -242,36 +242,54 @@ def extract_fields_with_sarvam(raw_text: str, doc_type: str) -> dict:
             timeout=30
         )
 
+        # Log the raw response for debugging
+        raw_body = response.text[:500]  # truncate for safety
+
         if response.status_code != 200:
             return {
                 **empty_fields,
-                "sarvam_error": f"API returned {response.status_code}: {response.text[:300]}"
+                "sarvam_error": f"API returned {response.status_code}: {raw_body}"
+            }
+
+        resp_json = response.json()
+
+        # Defensive: check schema before indexing
+        choices = resp_json.get("choices")
+        if not choices or not isinstance(choices, list) or len(choices) == 0:
+            return {
+                **empty_fields,
+                "sarvam_error": f"Unexpected response schema: {raw_body}"
             }
 
-        content = response.json()["choices"][0]["message"]["content"].strip()
+        message = choices[0].get("message", {})
+        content = message.get("content", "").strip()
+
+        if not content:
+            return {
+                **empty_fields,
+                "sarvam_error": f"Empty content in response. Full body: {raw_body}"
+            }
 
         # Strip markdown code fences if model wraps output
         content = re.sub(r"^```(?:json)?\s*", "", content)
-        content = re.sub(r"\s*```$", "", content)
+        content = re.sub(r"\s*```$", "", content).strip()
 
         extracted = json.loads(content)
 
-        # Normalize keys — ensure all expected fields are present
         fields = ["name", "address", "date_of_birth", "gender", "id_number", "mobile_number"]
         normalized = {f: extracted.get(f) for f in fields}
 
-        # Handle year_of_birth fallback (if model returns it instead of date_of_birth)
         if normalized["date_of_birth"] is None and "year_of_birth" in extracted:
             normalized["date_of_birth"] = str(extracted["year_of_birth"])
 
         return normalized
 
     except json.JSONDecodeError as e:
-        return {**empty_fields, "sarvam_parse_error": f"JSON parse failed: {str(e)}"}
+        return {**empty_fields, "sarvam_parse_error": f"JSON parse failed: {str(e)} | content was: '{content[:200]}'"}
     except requests.exceptions.Timeout:
         return {**empty_fields, "sarvam_error": "Sarvam API request timed out"}
     except Exception as e:
-        return {**empty_fields, "sarvam_error": str(e)}
+        return {**empty_fields, "sarvam_error": f"{type(e).__name__}: {str(e)}"}
 
 # -----------------------------
 # FastAPI App