Update app.py

app.py

@@ -1,14 +1,18 @@
 import json
-from paddleocr import PaddleOCR
+import os
+import gradio as gr
+from utils import extract_kyc_fields
 from simple_salesforce import Salesforce
 
 # ------------------------
 # CONFIGURATION
 # ------------------------
-SF_USERNAME = "licproject@2025.com"
-SF_PASSWORD = "Lic@2025"
-SF_TOKEN = "AmmfRcd6IiYaRtSGntBnzNMQU"
-IMAGE_PATH = "sample.jpg"  # Change this to your actual image file path
+SF_USERNAME = os.getenv("SF_USERNAME", "licproject@2025.com")
+SF_PASSWORD = os.getenv("SF_PASSWORD", "Lic@2025")
+SF_TOKEN = os.getenv("SF_TOKEN", "AmmfRcd6IiYaRtSGntBnzNMQU")
+
+# Limit Paddle threads to avoid container crashes
+os.environ["OMP_NUM_THREADS"] = "1"
 
 # ------------------------
 # CONNECT TO SALESFORCE
@@ -22,7 +26,7 @@ def connect_salesforce():
             domain="login"  # Production
         )
         print("✅ Connected to Salesforce (Production)")
-        return sf
+        return {"status": "success", "env": "Production"}
     except Exception as e:
         print("❌ Production login failed:", e)
         try:
@@ -33,49 +37,44 @@ def connect_salesforce():
                 domain="test"  # Sandbox
             )
             print("✅ Connected to Salesforce (Sandbox)")
-            return sf
+            return {"status": "success", "env": "Sandbox"}
         except Exception as e2:
             print("❌ Sandbox login failed:", e2)
-            return None
+            return {"status": "error", "message": "Salesforce login failed"}
+
+sf_status = connect_salesforce()
 
 # ------------------------
-# RUN OCR
+# PROCESS IMAGE FUNCTION
 # ------------------------
-def extract_text_from_image(image_path):
+def process_image(image_file):
+    if sf_status["status"] != "success":
+        return {"salesforce_status": "error", "message": sf_status.get("message", "Login failed")}
+
+    if image_file is None:
+        return {"ocr_status": "error", "message": "No image uploaded"}
+
     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}
+        # Run OCR extraction from utils.py
+        ocr_result = extract_kyc_fields(image_file.name)
+        return {
+            "salesforce_status": "connected",
+            "ocr_result": ocr_result
+        }
     except Exception as e:
-        return {"status": "error", "message": str(e)}
+        return {"ocr_status": "error", "message": str(e)}
 
 # ------------------------
-# MAIN EXECUTION
+# GRADIO INTERFACE
 # ------------------------
-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()
-
-    ocr_output = extract_text_from_image(IMAGE_PATH)
-
-    final_output = {
-        "salesforce_result": {
-            "status": "connected"
-        },
-        "ocr_result": ocr_output
-    }
+iface = gr.Interface(
+    fn=process_image,
+    inputs=gr.Image(type="file"),
+    outputs=gr.JSON(),
+    title="AI KYC OCR & Salesforce Integration",
+    description="Upload PAN or Aadhaar card images to extract text and verify Salesforce connection."
+)
 
-    print(json.dumps(final_output, indent=4))
-
-iface.launch()
-    
+if __name__ == "__main__":
+    # Launch server to keep container alive
+    iface.launch(server_name="0.0.0.0", server_port=7860)