|
|
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))) |
|
|
|
