app.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from model import generate_structure
import uvicorn

app = FastAPI()

# Allow all CORS (for testing or frontend use)
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

class PromptRequest(BaseModel):
    prompt: str

@app.post("/prompt")
async def prompt_route(data: PromptRequest):
    try:
        response = generate_structure(data.prompt)
        return {"response": response}
    except Exception as e:
        return {"error": str(e)}

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