Update app.py

app.py

@@ -1,78 +1,41 @@
-import json
-from paddleocr import PaddleOCR
-from simple_salesforce import Salesforce
+import os
+import gradio as gr
+from utils import extract_kyc_fields, connect_salesforce, create_kyc_record
 
-# ------------------------
-# CONFIGURATION
-# ------------------------
-SF_USERNAME = "[email protected]"
-SF_PASSWORD = "Lic@2025"
-SF_TOKEN = "AmmfRcd6IiYaRtSGntBnzNMQU"
-IMAGE_PATH = "sample.jpg"  # Change this to your actual image file path
+# Connect Salesforce on startup
+SF = connect_salesforce()
 
-# ------------------------
-# CONNECT TO SALESFORCE
-# ------------------------
-def connect_salesforce():
-    try:
-        sf = Salesforce(
-            username=SF_USERNAME,
-            password=SF_PASSWORD,
-            security_token=SF_TOKEN,
-            domain="login"  # Production
-        )
-        print("✅ Connected to Salesforce (Production)")
-        return sf
-    except Exception as e:
-        print("❌ Production login failed:", e)
-        try:
-            sf = Salesforce(
-                username=SF_USERNAME,
-                password=SF_PASSWORD,
-                security_token=SF_TOKEN,
-                domain="test"  # Sandbox
-            )
-            print("✅ Connected to Salesforce (Sandbox)")
-            return sf
-        except Exception as e2:
-            print("❌ Sandbox login failed:", e2)
-            return None
+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"}
 
-# ------------------------
-# RUN OCR
-# ------------------------
-def extract_text_from_image(image_path):
-    try:
-        ocr = PaddleOCR(use_angle_cls=True, lang='en')
-        results = ocr.ocr(image_path, cls=True)
-        extracted_text = [line[1][0] for line in results[0]]
-        return {"status": "success", "text": extracted_text}
-    except Exception as e:
-        return {"status": "error", "message": str(e)}
+    kyc_data = extract_kyc_fields(image_file)
 
-# ------------------------
-# MAIN EXECUTION
-# ------------------------
-if __name__ == "__main__":
-    sf_conn = connect_salesforce()
-    
-    if not sf_conn:
-        output = {
-            "salesforce_result": {
-                "status": "error",
-                "message": "Salesforce not connected"
-            }
-        }
-        print(json.dumps(output, indent=4))
-        exit()
+    agent_id = agent_id_optional.strip() if agent_id_optional else None
+    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
+    )
 
-    ocr_output = extract_text_from_image(IMAGE_PATH)
+    return {"ocr_result": kyc_data, "salesforce_result": sf_result}
 
-    final_output = {
-        "salesforce_result": {
-            "status": "connected"
-        },
-        "ocr_result": ocr_output
-    }
+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."
+    )
+)
 
-    print(json.dumps(final_output, indent=4))
+if __name__ == "__main__":
+    iface.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))