Update app.py

app.py

@@ -1,19 +1,30 @@
 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": str(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"),
+    inputs=gr.File(
+        label="Upload Aadhaar/PAN image",
+        file_types=[".jpg", ".jpeg", ".png", ".webp", ".tiff", ".bmp"]
+    ),
     outputs=gr.JSON(label="Extracted KYC Fields"),
-    title="Smart KYC OCR",
-    description="Upload an Aadhaar or PAN card image to extract structured KYC fields using Tesseract OCR"
+    title="🔍 Smart KYC OCR",
+    description=(
+        "Upload an image of an Aadhaar or PAN card, and this tool will extract key KYC fields like "
+        "**Name**, **Aadhaar Number**, and **Date of Birth** using Tesseract OCR."
+    ),
+    theme="dark"
 )
 
 if __name__ == "__main__":