from fastapi import FastAPI, Form
from fastapi.middleware.cors import CORSMiddleware
from .model_router import query_model

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

@app.post("/query")
async def get_response(prompt: str = Form(...), model: str = Form(None)):
    result = await query_model(prompt, model)
    return {"response": result}