Update app.py

app.py

@@ -1,145 +1,66 @@
 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
 from simple_salesforce import Salesforce
 
-# ------------------ SALESFORCE CREDS ------------------
-SF_USERNAME = "[email protected]"
-SF_PASSWORD = "Lic@2025"
-SF_SECURITY_TOKEN = "AmmfRcd6IiYaRtSGntBnzNMQU"
-SF_DOMAIN = "login"  # change to "test" if sandbox
-# -------------------------------------------------------
-
-def _parse_birthdate(dob_text: str):
-    if not dob_text or dob_text == "Not found":
-        return None
-    s = dob_text.strip()
-
-    m = re.fullmatch(r"(\d{4})-(\d{2})-(\d{2})", s)
-    if m:
-        y, mo, d = map(int, m.groups())
-        try:
-            return dt.date(y, mo, d).isoformat()
-        except ValueError:
-            return None
-
-    m = re.fullmatch(r"(\d{2})[./-](\d{2})[./-](\d{4})", s)
-    if m:
-        d, mo, y = map(int, m.groups())
-        try:
-            return dt.date(y, mo, d).isoformat()
-        except ValueError:
-            return None
-
-    m = re.fullmatch(r"(19|20)\d{2}", s)
-    if m:
-        y = int(s)
-        try:
-            return dt.date(y, 6, 15).isoformat()
-        except ValueError:
-            return None
-
-    return None
-
-def sf_login():
-    sf = Salesforce(
-        username=SF_USERNAME,
-        password=SF_PASSWORD,
-        security_token=SF_SECURITY_TOKEN,
-        domain=SF_DOMAIN,
-    )
-    sf.query("SELECT Id FROM User LIMIT 1")
-    return sf
-
-def sf_create_kyc_via_session(sf, payload: dict):
-    url = f"{sf.base_url}sobjects/KYC_Record__c"
-    resp = sf.session.post(url, json=payload, headers=sf.headers, timeout=30)
+# Salesforce login
+sf = Salesforce(
+    username="licproject@2025.com",
+    password="Lic@2025",
+    security_token="AmmfRcd6IiYaRtSGntBnzNMQU"
+)
 
+def send_to_salesforce(aadhaar_data, pan_data):
+    """Send both Aadhaar & PAN extracted data to Salesforce."""
     try:
-        body = resp.json()
-    except Exception:
-        body = None
-
-    if 200 <= resp.status_code < 300:
-        rec_id = body.get("id") if isinstance(body, dict) else None
-        return {"success": True, "id": rec_id, "status_code": resp.status_code, "url": url, "response_json": body}
+        record = {
+            "Aadhaar_Name__c": aadhaar_data.get("name", ""),
+            "Aadhaar_Number__c": aadhaar_data.get("aadhaar_number", ""),
+            "Aadhaar_DOB__c": aadhaar_data.get("dob", ""),
+            "Aadhaar_Gender__c": aadhaar_data.get("gender", ""),
+            "PAN_Name__c": pan_data.get("name", ""),
+            "PAN_Number__c": pan_data.get("pan_number", ""),
+            "PAN_DOB__c": pan_data.get("dob", "")
+        }
+        result = sf.KYC_Record__c.create(record)
+        return {"status": "success", "salesforce_id": result.get("id")}
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
+
+def process_documents(aadhaar_file, pan_file):
+    if aadhaar_file is None or pan_file is None:
+        return {"error": "Please upload both Aadhaar and PAN images."}
+
+    aadhaar_data = extract_kyc_fields(aadhaar_file.name)
+    pan_data = extract_kyc_fields(pan_file.name)
+
+    # Send to Salesforce
+    sf_result = send_to_salesforce(aadhaar_data, pan_data)
 
     return {
-        "success": False,
-        "status_code": resp.status_code,
-        "url": url,
-        "response_json": body,
-        "response_text": resp.text,
+        "aadhaar_data": aadhaar_data,
+        "pan_data": pan_data,
+        "salesforce_result": sf_result
     }
 
-def build_payload(ocr_results: dict):
-    a = (ocr_results.get("aadhaar") or {})
-    p = (ocr_results.get("pan") or {})
-
-    payload = {
-        "Aadhaar_Number__c": a.get("aadhaar_number") if a.get("card_type") == "AADHAAR" else None,
-        "Aadhaar_Name__c":   a.get("name")           if a.get("card_type") == "AADHAAR" else None,
-        "Aadhaar_DOB__c":    _parse_birthdate(a.get("dob")) if a.get("card_type") == "AADHAAR" else None,
-        "PAN_Number__c":     p.get("pan_number")     if p.get("card_type") == "PAN" else None,
-        "Pan_Name__c":       p.get("name")           if p.get("card_type") == "PAN" else None,
-        "Pan_DOB__c":        _parse_birthdate(p.get("dob")) if p.get("card_type") == "PAN" else None,
-    }
-    return {k: v for k, v in payload.items() if v is not None}
-
-def process_documents(aadhaar_file, pan_file, push_to_sf):
-    results = {"aadhaar": None, "pan": None}
-
-    if not aadhaar_file and not pan_file:
-        return {"error": "Please upload at least one file (Aadhaar and/or PAN)."}
-
-    if aadhaar_file:
-        try:
-            res = extract_kyc_fields(aadhaar_file.name)
-            res["source_file"] = os.path.basename(aadhaar_file.name)
-            results["aadhaar"] = res
-        except Exception as e:
-            results["aadhaar"] = {"error": f"Aadhaar OCR failed: {str(e)}", "card_type": "UNKNOWN"}
-
-    if pan_file:
-        try:
-            res = extract_kyc_fields(pan_file.name)
-            res["source_file"] = os.path.basename(pan_file.name)
-            results["pan"] = res
-        except Exception as e:
-            results["pan"] = {"error": f"PAN OCR failed: {str(e)}", "card_type": "UNKNOWN"}
-
-    output = {"ocr": results}
-
-    if push_to_sf:
-        payload = build_payload(results)
-        try:
-            sf = sf_login()
-            created = sf_create_kyc_via_session(sf, payload)
-            output["salesforce"] = {"pushed": created.get("success", False), **created}
-        except Exception as e:
-            output["salesforce"] = {"pushed": False, "error": {"type": e.__class__.__name__, "message": str(e)}}
-
-    return output
-
-with gr.Blocks(title="Smart KYC OCR → Salesforce (KYC_Record__c)") as demo:
-    gr.Markdown("# 🧾 Smart KYC OCR → Salesforce\nUpload Aadhaar and PAN, then optionally push one record.")
+with gr.Blocks(title="Smart KYC OCR") as demo:
+    gr.Markdown("# 🧾 Smart KYC OCR Tool — Aadhaar & PAN Scanner")
 
     with gr.Row():
         with gr.Column():
-            aadhaar_uploader = gr.File(label="📤 Aadhaar Upload", file_types=[".jpg", ".jpeg", ".png"])
+            aadhaar_uploader = gr.File(label="📤 Upload Aadhaar Card", file_types=[".jpg", ".jpeg", ".png"])
+            pan_uploader = gr.File(label="📤 Upload PAN Card", file_types=[".jpg", ".jpeg", ".png"])
+            submit_btn = gr.Button("🔍 Extract & Send to Salesforce", variant="primary")
+
         with gr.Column():
-            pan_uploader = gr.File(label="📤 PAN Upload", file_types=[".jpg", ".jpeg", ".png"])
+            output_json = gr.JSON(label="📋 Extracted KYC Data")
 
-    push_to_sf = gr.Checkbox(label="Push to Salesforce", value=False)
-    submit_btn = gr.Button("🔍 Extract KYC Info", variant="primary")
-    output_json = gr.JSON(label="📋 Output")
+    submit_btn.click(fn=process_documents, inputs=[aadhaar_uploader, pan_uploader], outputs=output_json)
 
-    submit_btn.click(fn=process_documents, inputs=[aadhaar_uploader, pan_uploader, push_to_sf], outputs=output_json)
+    gr.Markdown("---")
+    gr.Markdown("🔒 **Privacy Note:** This app processes your document locally in the cloud. No data is stored or shared.")
 
 if __name__ == "__main__":
     demo.launch()