Update app.py

app.py

@@ -1,20 +1,28 @@
-import gradio as gr
-from pipeline import extract_text_from_pdf, extract_key_values_with_gemini, FORMS
+from fastapi import FastAPI, UploadFile, Form
+from fastapi.responses import JSONResponse
+import uvicorn, os, tempfile
+from pipeline import extract_text_from_pdf, extract_key_values_with_gemini
 
-def process(pdf_file, form_type):
-    text = extract_text_from_pdf(pdf_file.name)
-    result = extract_key_values_with_gemini(text, form_type=form_type)
-    return result
+app = FastAPI(title="AutoFill Form Extractor API")
 
-iface = gr.Interface(
-    fn=process,
-    inputs=[
-        gr.File(label="Upload Document (PDF only)"),
-        gr.Dropdown(choices=list(FORMS.keys()), value="pancard_form", label="Form Type")
-    ],
-    outputs="json",
-    title="🧾 AutoFill Form Extractor",
-    description="Upload a document and extract structured fields using OCR + Gemini LLM"
-)
+@app.post("/extract")
+async def extract_api(file: UploadFile, form_type: str = Form("pancard_form")):
+    try:
+        # Save uploaded file temporarily
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+            tmp.write(await file.read())
+            tmp_path = tmp.name
 
-iface.launch()
+        text = extract_text_from_pdf(tmp_path)
+        result = extract_key_values_with_gemini(text, form_type=form_type)
+
+        return JSONResponse(content={"success": True, "data": result})
+    except Exception as e:
+        return JSONResponse(content={"success": False, "error": str(e)})
+
+@app.get("/")
+def home():
+    return {"message": "AutoFill Form API is running! Use POST /extract with a PDF."}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))