from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
from model import generate_structure
import uvicorn

app = FastAPI()

@app.get("/")
def index():
    return {"message": "Minecraft AI Builder Backend is running."}

@app.post("/prompt")
async def prompt_handler(req: Request):
    try:
        data = await req.json()
        prompt = data.get("prompt", "")
        response = generate_structure(prompt)
        return {"response": response}
    except Exception as e:
        return JSONResponse(status_code=500, content={"error": str(e)})

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