Update app.py

app.py

@@ -4,77 +4,42 @@ os.environ["OMP_NUM_THREADS"] = "1"
 import gradio as gr
 from utils import extract_kyc_fields
 
-def process_documents(files):
+
+def process_documents(aadhaar_file, pan_file):
     """
-    Upload BOTH Aadhaar and PAN together.
-    Returns:
-    {
-      "summary": {
-        "aadhaar": [ {...}, ... ],
-        "pan": [ {...}, ... ],
-        "unknown": [ {...}, ... ]
-      },
-      "results": [
-        { ... , "source_file": "file1.png" },
-        { ... , "source_file": "file2.jpg" }
-      ]
-    }
+    Takes one Aadhaar file and one PAN file,
+    extracts KYC fields, and returns combined JSON.
     """
-    if not files:
-        return {"error": "Please upload both files (Aadhaar and PAN)."}
-
-    # Gradio may pass a single file or list depending on version
-    if not isinstance(files, list):
-        files = [files]
+    results = {"aadhaar": None, "pan": None}
 
-    if len(files) < 2:
-        # soft validation; still process but nudge user
-        warning = "Tip: You uploaded fewer than 2 files. Upload both Aadhaar and PAN for best results."
-    else:
-        warning = None
-
-    grouped = {"aadhaar": [], "pan": [], "unknown": []}
-    flat_results = []
+    if aadhaar_file:
+        try:
+            res = extract_kyc_fields(aadhaar_file.name)
+            res["source_file"] = os.path.basename(aadhaar_file.name)
+            results["aadhaar"] = res
+        except Exception as e:
+            results["aadhaar"] = {"error": f"Aadhaar OCR failed: {str(e)}"}
 
-    for f in files:
+    if pan_file:
         try:
-            fp = getattr(f, "name", None) or getattr(f, "path", None)
-            if not fp:
-                res = {"error": "Could not read uploaded file path."}
-            else:
-                res = extract_kyc_fields(fp)
+            res = extract_kyc_fields(pan_file.name)
+            res["source_file"] = os.path.basename(pan_file.name)
+            results["pan"] = res
+        except Exception as e:
+            results["pan"] = {"error": f"PAN OCR failed: {str(e)}"}
 
-            res["source_file"] = os.path.basename(fp) if fp else "uploaded_file"
-            flat_results.append(res)
+    if not aadhaar_file and not pan_file:
+        return {"error": "Please upload both Aadhaar and PAN files."}
 
-            ct = (res.get("card_type") or "UNKNOWN").upper()
-            if ct == "AADHAAR":
-                grouped["aadhaar"].append(res)
-            elif ct == "PAN":
-                grouped["pan"].append(res)
-            else:
-                grouped["unknown"].append(res)
+    return results
 
-        except Exception as e:
-            err = {"error": f"OCR failed for {getattr(f, 'name', 'file')}: {str(e)}",
-                   "card_type": "UNKNOWN",
-                   "source_file": getattr(f, "name", "file")}
-            flat_results.append(err)
-            grouped["unknown"].append(err)
-
-    payload = {"summary": grouped, "results": flat_results}
-    if warning:
-        payload["note"] = warning
-    return payload
 
 with gr.Blocks(title="Smart KYC OCR") as demo:
     gr.Markdown(
         """
         # 🧾 Smart KYC OCR Tool
-        Upload **both**: an **Aadhaar** image and a **PAN** image (you can also upload more).  
-        The app will group the output by card type and show all results.
-
-        **How to upload both:** hold **Ctrl/⌘** to select two files, or drag & drop both images together.
+        Upload an **Aadhaar card** and a **PAN card** separately.  
+        Click **Extract KYC Info** to get structured output.
 
         ---
         """
@@ -82,23 +47,29 @@ with gr.Blocks(title="Smart KYC OCR") as demo:
 
     with gr.Row():
         with gr.Column(scale=1):
-            # Works with older Gradio: file_count="multiple"
-            uploader = gr.File(
-                label="📤 Upload Aadhaar & PAN Image(s)",
-                file_types=[".jpg", ".jpeg", ".png"],
-                file_count="multiple"
+            aadhaar_uploader = gr.File(
+                label="📤 Aadhaar Upload",
+                file_types=[".jpg", ".jpeg", ".png"]
             )
-            submit_btn = gr.Button("🔍 Extract KYC Info", variant="primary")
-
         with gr.Column(scale=1):
-            output_json = gr.JSON(label="📋 Grouped Output (aadhaar / pan) + All Results")
+            pan_uploader = gr.File(
+                label="📤 PAN Upload",
+                file_types=[".jpg", ".jpeg", ".png"]
+            )
+
+    submit_btn = gr.Button("🔍 Extract KYC Info", variant="primary")
+    output_json = gr.JSON(label="📋 Extracted KYC Fields")
 
-    submit_btn.click(fn=process_documents, inputs=uploader, outputs=output_json)
+    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.  
+        🔒 **Privacy Note:** This app processes your documents locally in the cloud.  
         No data is stored or shared.
         """
     )