# app.py
# ==============================
# DocuMint Space entry point
# Exposes API for Telegram bot
# ==============================

from fastapi import FastAPI
from inference import generate_response

app = FastAPI()


@app.post("/predict")
def predict(payload: dict):
    """
    Expected JSON:
    {
        "prompt": "user message"
    }
    """

    prompt = payload.get("prompt", "").strip()

    if not prompt:
        return {"response": "Empty prompt received."}

    response = generate_response(prompt)
    return {"response": response}


@app.get("/")
def root():
    return {"status": "DocuMint Space is running"}

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