Update app.py

app.py

@@ -1,87 +1,78 @@
-import os
-os.environ["OMP_NUM_THREADS"] = "1"
-
-import gradio as gr
-from utils import extract_kyc_fields
+import json
+from paddleocr import PaddleOCR
 from simple_salesforce import Salesforce
 
-# ------------------ Salesforce Login ------------------
-def salesforce_connect():
-    username = "licproject@2025.com"
-    password = "Lic@2025"
-    token = "AmmfRcd6IiYaRtSGntBnzNMQU"
-
+# ------------------------
+# 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
+
+# ------------------------
+# CONNECT TO SALESFORCE
+# ------------------------
+def connect_salesforce():
     try:
-        # Append token to password for login
         sf = Salesforce(
-            username=username,
-            password=password + token,
-            security_token="",
-            domain="login"  # change to "test" for sandbox
+            username=SF_USERNAME,
+            password=SF_PASSWORD,
+            security_token=SF_TOKEN,
+            domain="login"  # Production
         )
+        print("✅ Connected to Salesforce (Production)")
         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):
-    if isinstance(sf, str):
-        return {"status": "error", "message": "Salesforce not connected"}
-
+        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
+
+# ------------------------
+# RUN OCR
+# ------------------------
+def extract_text_from_image(image_path):
     try:
-        record = {
-            "Aadhaar_Name__c": aadhaar_data.get("name", ""),
-            "Aadhaar_Number__c": aadhaar_data.get("aadhaar_number", ""),
-            "Aadhaar_DOB__c": aadhaar_data.get("dob", ""),
-            "Aadhaar_Gender__c": aadhaar_data.get("gender", ""),
-            "PAN_Name__c": pan_data.get("name", ""),
-            "PAN_Number__c": pan_data.get("pan_number", ""),
-            "PAN_DOB__c": pan_data.get("dob", "")
-        }
-        result = sf.KYC_Record__c.create(record)
-        return {"status": "success", "salesforce_id": result.get("id")}
+        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)}
 
-# ------------------ 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, force_type="AADHAAR")
-    pan_data = extract_kyc_fields(pan_file.name, force_type="PAN")
+# ------------------------
+# 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()
 
-    sf_result = send_to_salesforce(aadhaar_data, pan_data)
+    ocr_output = extract_text_from_image(IMAGE_PATH)
 
-    return {
-        "aadhaar_data": aadhaar_data,
-        "pan_data": pan_data,
-        "salesforce_result": sf_result
+    final_output = {
+        "salesforce_result": {
+            "status": "connected"
+        },
+        "ocr_result": ocr_output
     }
 
-# ------------------ Gradio UI ------------------
-with gr.Blocks(title="Smart KYC OCR") as demo:
-    gr.Markdown("# 🧾 Smart KYC OCR Tool — Aadhaar & PAN Scanner")
-
-    with gr.Row():
-        with gr.Column():
-            aadhaar_uploader = gr.File(label="📤 Upload Aadhaar Card", file_types=[".jpg", ".jpeg", ".png"])
-            pan_uploader = gr.File(label="📤 Upload PAN Card", file_types=[".jpg", ".jpeg", ".png"])
-            submit_btn = gr.Button("🔍 Extract & Send to Salesforce", variant="primary")
-
-        with gr.Column():
-            output_json = gr.JSON(label="📋 Extracted KYC Data")
-
-    submit_btn.click(fn=process_documents, inputs=[aadhaar_uploader, pan_uploader], outputs=output_json)
-
-    gr.Markdown("---")
-    gr.Markdown("🔒 **Privacy Note:** This app processes your document locally in the cloud. No data is stored or shared.")
-
-if __name__ == "__main__":
-    demo.launch()
+    print(json.dumps(final_output, indent=4))