app.py

from fastapi import FastAPI, File, UploadFile
from fastapi.responses import JSONResponse
import os
from py.process_pdfs_grok import extract_text_from_pdf
from extract_entities import extract_entities_spacy, extract_entities_bert

app = FastAPI()

@app.post("/process_pdf")
async def process_pdf(file: UploadFile = File(...)):
    try:
        # Save uploaded PDF temporarily
        upload_dir = "uploads"
        os.makedirs(upload_dir, exist_ok=True)
        file_path = os.path.join(upload_dir, file.filename)
        with open(file_path, "wb") as f:
            f.write(await file.read())
        
        # Extract text
        text = extract_text_from_pdf(file_path)
        
        # Extract entities (use spaCy for PoC; BERT for comparison)
        spacy_entities = extract_entities_spacy(text)
        bert_entities = extract_entities_bert(text)
        
        # Clean up
        os.remove(file_path)
        
        return JSONResponse({
            "status": "success",
            "text": text[:500],  # Limit for brevity
            "spacy_entities": spacy_entities,
            "bert_entities": bert_entities
        })
    except Exception as e:
        return JSONResponse({
            "status": "error",
            "message": str(e)
        }, status_code=500)

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000)