Update app.py

app.py

@@ -1,49 +1,80 @@
 import os
-os.environ["OMP_NUM_THREADS"] = "1"  # Avoid multithread warning from some BLAS libs
+os.environ["OMP_NUM_THREADS"] = "1"
 
 import gradio as gr
 from utils import extract_kyc_fields
 
 def process_documents(files):
     """
-    Accepts multiple files, OCRs each, and returns a single JSON payload:
+    Upload BOTH Aadhaar and PAN together.
+    Returns:
     {
+      "summary": {
+        "aadhaar": [ {...}, ... ],
+        "pan": [ {...}, ... ],
+        "unknown": [ {...}, ... ]
+      },
       "results": [
-        { ...result for file 1..., "source_file": "name1.png" },
-        { ...result for file 2..., "source_file": "name2.jpg" },
-        ...
+        { ... , "source_file": "file1.png" },
+        { ... , "source_file": "file2.jpg" }
       ]
     }
     """
     if not files:
-        return {"error": "No file(s) uploaded."}
+        return {"error": "Please upload both files (Aadhaar and PAN)."}
 
-    # gr.File may return a single object or a list depending on Gradio version.
+    # Gradio may pass a single file or list depending on version
     if not isinstance(files, list):
         files = [files]
 
-    results = []
+    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 = []
+
     for f in files:
         try:
             fp = getattr(f, "name", None) or getattr(f, "path", None)
             if not fp:
-                results.append({"error": "Could not read uploaded file path."})
-                continue
+                res = {"error": "Could not read uploaded file path."}
+            else:
+                res = extract_kyc_fields(fp)
+
+            res["source_file"] = os.path.basename(fp) if fp else "uploaded_file"
+            flat_results.append(res)
+
+            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)
 
-            res = extract_kyc_fields(fp)
-            res["source_file"] = os.path.basename(fp)
-            results.append(res)
         except Exception as e:
-            results.append({"error": f"OCR failed for {getattr(f, 'name', 'file')}: {str(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)
 
-    return {"results": results}
+    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 images of **Aadhaar** and/or **PAN** cards — multiple at once — and get structured KYC data instantly.  
-        Powered by **PaddleOCR** and smart field detection.
+        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.
 
         ---
         """
@@ -51,15 +82,16 @@ 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)",
+                label="📤 Upload Aadhaar & PAN Image(s)",
                 file_types=[".jpg", ".jpeg", ".png"],
-                file_count="multiple",   # works on older Gradio; for newer, you can use gr.Files
+                file_count="multiple"
             )
             submit_btn = gr.Button("🔍 Extract KYC Info", variant="primary")
 
         with gr.Column(scale=1):
-            output_json = gr.JSON(label="📋 Extracted KYC Fields (All Files)")
+            output_json = gr.JSON(label="📋 Grouped Output (aadhaar / pan) + All Results")
 
     submit_btn.click(fn=process_documents, inputs=uploader, outputs=output_json)
 
@@ -72,5 +104,4 @@ with gr.Blocks(title="Smart KYC OCR") as demo:
     )
 
 if __name__ == "__main__":
-    # Change host/port if you need to expose it elsewhere
     demo.launch()