Update app.py

app.py

@@ -1,10 +1,6 @@
-from fastapi import FastAPI, UploadFile, Form
-from fastapi.responses import JSONResponse
-import uvicorn, os, tempfile
+import gradio as gr
 from pipeline import extract_text_from_pdf, extract_key_values_with_gemini
 
-app = FastAPI(title="AutoFill Form Extractor API")
-
 FIELDS = [
     "Name", "DOB", "Gender", "FatherName", "MotherName",
     "Address", "City", "State", "Pincode", "Mobile", "Email",
@@ -12,24 +8,16 @@ FIELDS = [
     "IssueDate", "ExpiryDate"
 ]
 
-@app.post("/extract")
-async def extract_api(file: UploadFile):
-    try:
-        # Save uploaded file temporarily
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
-            tmp.write(await file.read())
-            tmp_path = tmp.name
-
-        text = extract_text_from_pdf(tmp_path)
-        result = extract_key_values_with_gemini(text, fields=FIELDS)
-
-        return JSONResponse(content={"success": True, "data": result})
-    except Exception as e:
-        return JSONResponse(content={"success": False, "error": str(e)})
+def process_pdf(file):
+    if not file:
+        return {"error": "No file uploaded."}
+    text = extract_text_from_pdf(file.name)
+    return extract_key_values_with_gemini(text, FIELDS)
 
-@app.get("/")
-def home():
-    return {"message": "AutoFill Form API is running! Use POST /extract with a PDF."}
+with gr.Blocks() as demo:
+    gr.Markdown("## AutoForm Extractor")
+    file_input = gr.File(label="Upload PDF")
+    output = gr.JSON()
+    file_input.change(process_pdf, file_input, output)
 
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+demo.launch()