Update app.py

app.py

@@ -2,40 +2,81 @@ import os
 import gradio as gr
 from utils import extract_kyc_fields, connect_salesforce, create_kyc_record
 
-# Connect Salesforce on startup
-SF = connect_salesforce()
+def process(
+    aadhaar_image, 
+    pan_image, 
+    agent_id,
+    sf_username, 
+    sf_password, 
+    sf_token, 
+    sf_domain
+):
+    # Validate at least one image
+    if not aadhaar_image and not pan_image:
+        return {"status": "error", "message": "Upload at least one image (Aadhaar or PAN)."}
 
-def process_image(image_file, agent_id_optional):
-    if not image_file:
-        return {"status": "error", "message": "No image uploaded"}
-    if not SF:
-        return {"status": "error", "message": "Salesforce connection failed"}
+    # Connect Salesforce (show *real* error cause back to UI)
+    try:
+        sf = connect_salesforce(
+            username=(sf_username or None),
+            password=(sf_password or None),
+            token=(sf_token or None),
+            domain=(sf_domain or None),
+        )
+    except Exception as e:
+        return {"status": "error", "message": f"Salesforce connection failed: {str(e)}"}
 
-    kyc_data = extract_kyc_fields(image_file)
+    # OCR each card separately so we can fill both sides of the record
+    aadhaar_out = extract_kyc_fields(aadhaar_image, force_type="AADHAAR") if aadhaar_image else None
+    pan_out     = extract_kyc_fields(pan_image, force_type="PAN") if pan_image else None
 
-    agent_id = agent_id_optional.strip() if agent_id_optional else None
+    # Create the record
     sf_result = create_kyc_record(
-        SF,
-        kyc_data,
-        file_name=os.path.basename(image_file),
-        agent_id=agent_id or None,  # only send if provided
+        sf=sf,
+        aadhaar=aadhaar_out,
+        pan=pan_out,
+        agent_id=(agent_id.strip() if agent_id else None)
     )
 
-    return {"ocr_result": kyc_data, "salesforce_result": sf_result}
-
-iface = gr.Interface(
-    fn=process_image,
-    inputs=[
-        gr.Image(type="filepath", label="Upload PAN / Aadhaar"),
-        gr.Textbox(label="Agent Salesforce ID (optional)", placeholder="Enter Agent__c Id (optional)")
-    ],
-    outputs=gr.JSON(),
-    title="AI KYC OCR → Salesforce (KYC_Record__c)",
-    description=(
-        "Uploads a PAN/Aadhaar image, extracts fields, and creates a KYC_Record__c with:\n"
-        "Aadhaar_Name__c, Aadhaar_DOB__c, Aadhaar_Number__c, Pan_Name__c, Pan_DOB__c, PAN_Number__c."
+    return {
+        "aadhaar_ocr": aadhaar_out,
+        "pan_ocr": pan_out,
+        "salesforce_result": sf_result
+    }
+
+with gr.Blocks(title="AI KYC OCR → Salesforce (KYC_Record__c)") as demo:
+    gr.Markdown(
+        "## AI KYC OCR → Salesforce (KYC_Record__c)\n"
+        "Upload **Aadhaar** and/or **PAN** images. We’ll extract fields and create a **KYC_Record__c** "
+        "with the following fields: `Aadhaar_Name__c`, `Aadhaar_DOB__c`, `Aadhaar_Number__c`, "
+        "`Pan_Name__c`, `Pan_DOB__c`, `PAN_Number__c`."
+    )
+
+    with gr.Row():
+        aadhaar_img = gr.Image(type="filepath", label="Upload Aadhaar (Front)")
+        pan_img = gr.Image(type="filepath", label="Upload PAN")
+
+    agent_id = gr.Textbox(label="Agent Salesforce ID (optional)", placeholder="Agent__c (optional)")
+
+    with gr.Accordion("Salesforce connection (optional — leave blank to use env vars)", open=False):
+        sf_username = gr.Textbox(label="SF Username", placeholder="e.g. name@example.com", lines=1)
+        sf_password = gr.Textbox(label="SF Password", type="password")
+        sf_token    = gr.Textbox(label="SF Security Token", type="password")
+        sf_domain   = gr.Dropdown(
+            label="SF Domain", 
+            choices=["login", "test"], 
+            value="login",
+            info="Use 'login' for Production, 'test' for Sandbox"
+        )
+
+    out = gr.JSON(label="output")
+
+    run_btn = gr.Button("Extract & Create KYC_Record__c", variant="primary")
+    run_btn.click(
+        fn=process,
+        inputs=[aadhaar_img, pan_img, agent_id, sf_username, sf_password, sf_token, sf_domain],
+        outputs=[out]
     )
-)
 
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))