Update app.py

app.py

@@ -1,27 +1,41 @@
 import os
-os.environ["OMP_NUM_THREADS"] = "1"  # Avoid multithread warning
+os.environ["OMP_NUM_THREADS"] = "1"  # Avoid multithread warning from some BLAS libs
 
 import gradio as gr
 from utils import extract_kyc_fields
 
 def process_documents(files):
     """
-    Accepts multiple files, OCRs each, and returns a list of JSON results.
+    Accepts multiple files, OCRs each, and returns a single JSON payload:
+    {
+      "results": [
+        { ...result for file 1..., "source_file": "name1.png" },
+        { ...result for file 2..., "source_file": "name2.jpg" },
+        ...
+      ]
+    }
     """
     if not files:
         return {"error": "No file(s) uploaded."}
 
+    # gr.File may return a single object or a list depending on Gradio version.
+    if not isinstance(files, list):
+        files = [files]
+
     results = []
     for f in files:
         try:
-            # gr.File returns a tempfile.NamedTemporaryFile-like object
-            res = extract_kyc_fields(f.name)
-            # Attach source file name for clarity
-            res["source_file"] = getattr(f, "name", "uploaded_file")
+            fp = getattr(f, "name", None) or getattr(f, "path", None)
+            if not fp:
+                results.append({"error": "Could not read uploaded file path."})
+                continue
+
+            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)}"})
-    # Return a dict so it’s easy to read in JSON component
+
     return {"results": results}
 
 with gr.Blocks(title="Smart KYC OCR") as demo:
@@ -29,7 +43,7 @@ with gr.Blocks(title="Smart KYC OCR") as demo:
         """
         # 🧾 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 layout-based field detection.
+        Powered by **PaddleOCR** and smart field detection.
 
         ---
         """
@@ -40,7 +54,7 @@ with gr.Blocks(title="Smart KYC OCR") as demo:
             uploader = gr.File(
                 label="📤 Upload Aadhaar / PAN Image(s)",
                 file_types=[".jpg", ".jpeg", ".png"],
-                multiple=True
+                file_count="multiple",   # works on older Gradio; for newer, you can use gr.Files
             )
             submit_btn = gr.Button("🔍 Extract KYC Info", variant="primary")
 
@@ -58,4 +72,5 @@ with gr.Blocks(title="Smart KYC OCR") as demo:
     )
 
 if __name__ == "__main__":
+    # Change host/port if you need to expose it elsewhere
     demo.launch()