Update app.py

app.py

@@ -5,15 +5,35 @@ import gradio as gr
 from utils import extract_kyc_fields
 from simple_salesforce import Salesforce
 
-# Salesforce login
-sf = Salesforce(
-    username="[email protected]",
-    password="Lic@2025",
-    security_token="AmmfRcd6IiYaRtSGntBnzNMQU"
-)
+# ------------------ Salesforce Login ------------------
+def salesforce_connect():
+    username = "[email protected]"
+    password = "Lic@2025"
+    token = "AmmfRcd6IiYaRtSGntBnzNMQU"
 
+    try:
+        # Append token to password for login
+        sf = Salesforce(
+            username=username,
+            password=password + token,
+            security_token="",
+            domain="login"  # change to "test" for sandbox
+        )
+        return sf
+    except Exception as e:
+        return str(e)
+
+sf = salesforce_connect()
+if isinstance(sf, str):
+    print("Salesforce login failed:", sf)
+else:
+    print("Salesforce connected successfully!")
+
+# ------------------ Send Data to Salesforce ------------------
 def send_to_salesforce(aadhaar_data, pan_data):
-    """Send both Aadhaar & PAN extracted data to Salesforce."""
+    if isinstance(sf, str):
+        return {"status": "error", "message": "Salesforce not connected"}
+
     try:
         record = {
             "Aadhaar_Name__c": aadhaar_data.get("name", ""),
@@ -29,14 +49,14 @@ def send_to_salesforce(aadhaar_data, pan_data):
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
+# ------------------ OCR Process ------------------
 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)
+    aadhaar_data = extract_kyc_fields(aadhaar_file.name, force_type="AADHAAR")
+    pan_data = extract_kyc_fields(pan_file.name, force_type="PAN")
 
-    # Send to Salesforce
     sf_result = send_to_salesforce(aadhaar_data, pan_data)
 
     return {
@@ -45,6 +65,7 @@ def process_documents(aadhaar_file, pan_file):
         "salesforce_result": sf_result
     }
 
+# ------------------ Gradio UI ------------------
 with gr.Blocks(title="Smart KYC OCR") as demo:
     gr.Markdown("# 🧾 Smart KYC OCR Tool — Aadhaar & PAN Scanner")