Update app.py

app.py

@@ -1,11 +1,7 @@
-import json
 import os
 import gradio as gr
 from utils import extract_kyc_fields
 from simple_salesforce import Salesforce
-from paddleocr import PaddleOCR
-
-ocr = PaddleOCR(use_angle_cls= True, lang ='en')
 
 # ------------------------
 # CONFIGURATION
@@ -21,6 +17,7 @@ os.environ["OMP_NUM_THREADS"] = "1"
 # CONNECT TO SALESFORCE
 # ------------------------
 def connect_salesforce():
+    """Connects to Salesforce (Production first, then Sandbox fallback)"""
     try:
         sf = Salesforce(
             username=SF_USERNAME,
@@ -29,7 +26,7 @@ def connect_salesforce():
             domain="login"  # Production
         )
         print("✅ Connected to Salesforce (Production)")
-        return {"status": "success", "env": "Production"}
+        return {"status": "success", "env": "Production", "sf": sf}
     except Exception as e:
         print("❌ Production login failed:", e)
         try:
@@ -40,7 +37,7 @@ def connect_salesforce():
                 domain="test"  # Sandbox
             )
             print("✅ Connected to Salesforce (Sandbox)")
-            return {"status": "success", "env": "Sandbox"}
+            return {"status": "success", "env": "Sandbox", "sf": sf}
         except Exception as e2:
             print("❌ Sandbox login failed:", e2)
             return {"status": "error", "message": "Salesforce login failed"}
@@ -51,6 +48,7 @@ sf_status = connect_salesforce()
 # PROCESS IMAGE FUNCTION
 # ------------------------
 def process_image(image_file):
+    """Process uploaded KYC image and extract PAN/Aadhaar details"""
     if sf_status["status"] != "success":
         return {"salesforce_status": "error", "message": sf_status.get("message", "Login failed")}
 
@@ -62,23 +60,26 @@ def process_image(image_file):
         ocr_result = extract_kyc_fields(image_file.name)
         return {
             "salesforce_status": "connected",
+            "sf_env": sf_status.get("env"),
             "ocr_result": ocr_result
         }
     except Exception as e:
-        return {"ocr_status": "error", "message": str(e)}
+        return {"ocr_status": "error", "message": f"OCR extraction failed: {str(e)}"}
 
 # ------------------------
 # GRADIO INTERFACE
 # ------------------------
 iface = gr.Interface(
     fn=process_image,
-    inputs=gr.Image(type="filepath"),  # ✅ fixed
+    inputs=gr.Image(type="filepath"),  # Use filepath for PaddleOCR
     outputs=gr.JSON(),
     title="AI KYC OCR & Salesforce Integration",
     description="Upload PAN or Aadhaar card images to extract text and verify Salesforce connection."
 )
 
-
+# ------------------------
+# LAUNCH APP
+# ------------------------
 if __name__ == "__main__":
-    # Launch server to keep container alive
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    # Launch Gradio server
+    iface.launch(server_name="0.0.0.0", server_port=7860, debug=True)