| 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 |
|
|
| app = FastAPI(title="AutoFill Form Extractor API") |
|
|
| @app.post("/extract") |
| async def extract_api(file: UploadFile, form_type: str = Form("pancard_form")): |
| try: |
| |
| 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, 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))) |
|
|
