wdsa

app.py

@@ -1,11 +1,34 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from pydantic import BaseModel
+from fastapi.middleware.cors import CORSMiddleware
+from llama_cpp import Llama
+import os
 
 app = FastAPI()
 
-class PromptRequest(BaseModel):
+# Load model
+MODEL_PATH = "./models/gemma-2b-it.gguf"
+llm = Llama(model_path=MODEL_PATH, n_ctx=512)
+
+# Allow CORS (so frontend or Swagger can work)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # change to frontend origin in production
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Input model
+class PromptInput(BaseModel):
     prompt: str
 
 @app.post("/prompt")
-async def handle_prompt(request: PromptRequest):
-    return {"message": f"Prompt received: {request.prompt}"}
+async def generate_response(data: PromptInput):
+    output = llm(data.prompt, max_tokens=512, stop=["</s>", "\n\n"], echo=False)
+    return {"response": output["choices"][0]["text"].strip()}
+
+# Healthcheck
+@app.get("/")
+def read_root():
+    return {"message": "AI Builder Backend running"}

requirements.txt

@@ -1,2 +1,5 @@
 fastapi
 uvicorn
+llama-cpp-python==0.2.72  # prebuilt versions now available from PyPI
+python-dotenv
+huggingface_hub

tester.py

@@ -1,25 +1,14 @@
 import requests
 
-prompt = """
-You are an AI builder assistant. Your task is to generate a 3D Minecraft structure layout for a "wizard tower" using only the blocks listed below. The layout must be provided in strict JSON format as:
-layout[y][z][x] = block_name (3D array of strings)
-
-The JSON must be minimal, symmetrical if needed, and represent a small but visually meaningful structure. Avoid empty arrays, and make sure block names match exactly.
-
-You could only use these blocks for building wizard tower, including:
-Animated blocks: lava, fire, end_portal, magma_block, soul_fire, bubble_column
-Non-animated blocks: cobblestone, mossy_cobblestone, stone_bricks, cracked_stone_bricks, deepslate, bookshelves, oak_planks, oak_log, torch, stone_slab, obsidian, glass_pane, ladder
-
-Only return a valid JSON structure named `layout`. No explanation or extra text.
-"""
-
-
-response = requests.post(
-    url="http://thongcoder-minecraft-ai-builder-backend.hf.space/prompt",
-    json={"prompt": prompt},
-    headers={"Content-Type": 'application/json',
-             'accept': 'application/json'}
-)
-
-print(response.status_code)
-print(response.json())
+url = "http://thongcoder-minecraft-ai-builder-backend.hf.space/prompt"
+headers = {
+    "Content-Type": "application/json"
+}
+data = {
+    "prompt": "Design a small medieval gate using stone bricks and oak wood."
+}
+
+response = requests.post(url, headers=headers, json=data)
+
+print("Status Code:", response.status_code)
+print("Response:", response.text)