from fastapi import FastAPI
from pydantic import BaseModel
from app import OpenRouterLLM   # or the function that actually runs your model

app = FastAPI()

class GaiaRequest(BaseModel):
    prompt: str

class GaiaResponse(BaseModel):
    output: str

@app.post("/predict", response_model=GaiaResponse)
def predict(req: GaiaRequest):
    result = run_agent(req.prompt)
    return {"output": result}