Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 os.environ["OMP_NUM_THREADS"] = "1"
 
 import re
+import json
 import datetime as dt
 import gradio as gr
 from utils import extract_kyc_fields
@@ -101,18 +102,49 @@ def sf_connect():
     if not SF_AVAILABLE:
         raise RuntimeError("simple-salesforce is not installed. Add `simple-salesforce` to requirements.txt.")
 
+    sf = Salesforce(
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN,
+        domain=SF_DOMAIN,
+    )
+    # quick auth sanity check (will raise on bad auth)
+    sf.query("SELECT Id FROM User LIMIT 1")
+    return sf
+
+
+def _raw_create_with_fallback(sf, object_api_name: str, payload: dict):
+    """
+    Fallback path that bypasses simple-salesforce's error wrappers and calls REST directly.
+    Always returns a dict:
+      - on success: {"success": True, "id": "...", "url": "...", "status_code": 201}
+      - on error:   {"success": False, "status_code": <int>, "url": "...", "response_json": <json or None>, "response_text": <str>}
+    """
+    url = f"{sf.base_url}sobjects/{object_api_name}"
     try:
-        sf = Salesforce(
-            username=SF_USERNAME,
-            password=SF_PASSWORD,
-            security_token=SF_SECURITY_TOKEN,
-            domain=SF_DOMAIN,
-        )
-        # quick auth sanity check
-        sf.query("SELECT Id FROM User LIMIT 1")
-        return sf
+        resp = sf.session.post(url, json=payload, headers=sf.headers, timeout=30)
+        status = resp.status_code
+        try:
+            body = resp.json()
+        except Exception:
+            body = None
+
+        if 200 <= status < 300:
+            # Salesforce returns {"id": "...", "success": true, "errors": []}
+            rec_id = (body or {}).get("id") if isinstance(body, dict) else None
+            return {"success": True, "id": rec_id, "status_code": status, "url": url, "response_json": body}
+        else:
+            # Return raw details so you see the exact field/object error
+            return {
+                "success": False,
+                "status_code": status,
+                "url": url,
+                "response_json": body,
+                "response_text": resp.text,
+            }
     except Exception as e:
-        raise e
+        # Network or session issues
+        return {"success": False, "url": url, "exception": _fmt_sf_error(e)}
 
 
 def sf_push_kyc_record(sf, ocr_results):
@@ -136,25 +168,32 @@ def sf_push_kyc_record(sf, ocr_results):
     payload = {
         "Aadhaar_Number__c": aadhaar_number,
         "Aadhaar_Name__c":   aadhaar_name,
-        "Aadhaar_DOB__c":    aadhaar_dob,
+        "Aadhaar_DOB__c":    aadhaar_dob,   # Date field in SF
         "PAN_Number__c":     pan_number,
         "Pan_Name__c":       pan_name,
-        "Pan_DOB__c":        pan_dob,
+        "Pan_DOB__c":        pan_dob,       # Date field in SF
     }
+    # Remove None keys to avoid nulling non-nullable fields
     payload = {k: v for k, v in payload.items() if v is not None}
 
+    # First try the nice SDK method
     try:
         result = sf.KYC_Record__c.create(payload)
-        return {"success": True, "id": result.get("id"), "payload": payload}
+        return {"success": True, "id": result.get("id"), "payload": payload, "via": "sdk"}
     except Exception as e:
-        return {"success": False, "error": _fmt_sf_error(e), "payload": payload}
+        # If simple-salesforce throws its unhelpful TypeError, fall back to a raw REST POST
+        raw = _raw_create_with_fallback(sf, "KYC_Record__c", payload)
+        if raw.get("success"):
+            return {"success": True, "id": raw.get("id"), "payload": payload, "via": "raw", "raw": raw}
+        else:
+            return {"success": False, "error": _fmt_sf_error(e), "payload": payload, "raw": raw}
 
 
 # ---------- gradio callback ----------
 def process_documents(aadhaar_file, pan_file, push_to_sf):
     """
     - Runs OCR on Aadhaar and PAN separately.
-    - Optionally pushes a single KYC_Record__c to Salesforce.
+    - Optionally pushes a single KYC_Record__c to Salesforce with robust fallback.
     """
     results = {"aadhaar": None, "pan": None}
 
@@ -187,6 +226,7 @@ def process_documents(aadhaar_file, pan_file, push_to_sf):
             created = sf_push_kyc_record(sf, results)
             output["salesforce"] = {"pushed": created.get("success", False), **created}
         except Exception as e:
+            # Even connection/auth errors will be formatted
             output["salesforce"] = {"pushed": False, "error": _fmt_sf_error(e)}
 
     return output
@@ -227,9 +267,10 @@ with gr.Blocks(title="Smart KYC OCR → Salesforce (KYC_Record__c)") as demo:
     gr.Markdown("---")
     gr.Markdown(
         """
-        ⚠️ Credentials are embedded in this file. Keep this code private and rotate the token if shared.
+        If an error occurs, you'll now see **status_code**, **url**, and Salesforce’s **response_json** for fast debugging.
         """
     )
 
+# Important for Spaces: keep `demo` at module level
 if __name__ == "__main__":
     demo.launch()