Update app.py

app.py

@@ -1,28 +1,42 @@
 import os
-os.environ["OMP_NUM_THREADS"] = "1"  # Avoid threading conflict in Hugging Face
+os.environ["OMP_NUM_THREADS"] = "1"  # Avoid multithread warning
 
 import gradio as gr
 from utils import extract_kyc_fields
-import traceback
 
 def process_document(file):
-    try:
-        result = extract_kyc_fields(file.name)
-        return result
-    except Exception as e:
-        return {
-            "error": f"Failed to process image.\n{str(e)}",
-            "traceback": traceback.format_exc()
-        }
-
-iface = gr.Interface(
-    fn=process_document,
-    inputs=gr.File(label="Upload Aadhaar/PAN image", file_types=[".jpg", ".jpeg", ".png"]),
-    outputs=gr.JSON(label="Extracted KYC Fields"),
-    title="🔍 Smart KYC OCR (PaddleOCR)",
-    description="Upload Aadhaar or PAN card image to extract structured KYC fields using PaddleOCR.",
-    theme="default"
-)
+    if file is None:
+        return {"error": "No file uploaded."}
+    return extract_kyc_fields(file.name)
+
+with gr.Blocks(title="Smart KYC OCR") as demo:
+    gr.Markdown(
+        """
+        # 🧾 Smart KYC OCR Tool
+        Upload an image of an **Aadhaar** or **PAN** card and get structured KYC data instantly.  
+        Powered by **PaddleOCR** and smart layout-based field detection.
+
+        ---
+        """
+    )
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            uploader = gr.File(label="📤 Upload Aadhaar / PAN Image", file_types=[".jpg", ".jpeg", ".png"])
+            submit_btn = gr.Button("🔍 Extract KYC Info", variant="primary")
+
+        with gr.Column(scale=1):
+            output_json = gr.JSON(label="📋 Extracted KYC Fields")
+
+    submit_btn.click(fn=process_document, inputs=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__":
-    iface.launch()
+    demo.launch()