File size: 725 Bytes
52f3332 5ef42a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | # 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)
|