Spaces:
Runtime error
Runtime error
File size: 562 Bytes
c8f4eef |
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 |
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from model import generate_response
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=False,
allow_methods=["*"],
allow_headers=["*"],
)
class ChatRequest(BaseModel):
message: str
history: list[str] = []
@app.get("/")
def health():
return {"status": "ok"}
@app.post("/chat")
def chat(req: ChatRequest):
reply = generate_response(req.message)
return {"reply": reply}
|