app.py

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

FIELDS = [
    "Name", "DOB", "Gender", "FatherName", "MotherName",
    "Address", "City", "State", "Pincode", "Mobile", "Email",
    "DocumentType", "DocumentNumber", "IssueAuthority",
    "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)})

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