|
|
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) |
|
|
|
