d

Dockerfile

@@ -4,10 +4,10 @@ RUN apt-get update && apt-get install -y \
     build-essential cmake wget git \
     && rm -rf /var/lib/apt/lists/*
 
-WORKDIR /code
-COPY . /code
+WORKDIR /app
+COPY . /app
 
 RUN pip install --no-cache-dir -r requirements.txt
-RUN chmod -R 777 /code
+RUN chmod -R 777 /app
 
 CMD ["python", "app.py"]

app.py

@@ -1,30 +1,81 @@
+import curl_cffi
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
-from model import generate_structure
+import os
+
 import uvicorn
 
+# Load Gemini API key from secret
+GEM_API_KEY = os.getenv('GEMINI_API_KEY')
+
+# Init app
 app = FastAPI()
 
-# Allow all CORS (for testing or frontend use)
+# CORS (allow frontend)
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
+    allow_origins=["*"],  # Optional: restrict in prod
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
+# Request model
 class PromptRequest(BaseModel):
-    prompt: str
+    structure_name: str
+    structure_desc: str
+    blocks_allowed: list[str]  # Enforce lowercase block names client-side
 
-@app.post("/prompt")
-async def prompt_route(data: PromptRequest):
+@app.post("/generate_structure")
+async def generate_with_gemini(req: PromptRequest):
     try:
-        response = generate_structure(data.prompt)
-        return {"response": response}
+        # Compose the extremely strict prompt
+        prompt = f"""You are a Minecraft-style structure planner. You have the curiosity to build almost anything you could think of. {req.structure_name}
+
+{req.structure_desc}
+
+Only output a JSON object describing a 3D structure using this format:
+
+{{
+  "width": <int>,
+  "height": <int>,
+  "depth": <int>,
+  "layers": [
+    [
+      ["stone", "stone", "stone"],
+      ["stone", "air",   "stone"],
+      ["stone", "stone", "stone"]
+    ],
+    ...
+  ]
+}}
+
+Only use lowercase Minecraft block IDs (e.g. "stone", "air", "glass", "planks").
+You could only build this structure using {" ".join(req.blocks_allowed)}
+Do not include any natural language or explanation.
+Output strictly valid JSON only."""
+
+        # Call Gemini API using curl_cffi
+        session = curl_cffi.Session(impersonate="chrome110")  # Optional impersonation
+        response = session.post(
+            url="https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent",
+            headers={
+                "x-goog-api-key": GEM_API_KEY,
+                "Content-Type": "application/json"
+            },
+            json={
+                "contents": [{
+                    "parts": [{
+                        "text": prompt
+                    }]
+                }]
+            }
+        )
+
+        return {"response": response.text}
+
     except Exception as e:
         return {"error": str(e)}
 
-if __name__  == "__main__":
+if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)

data/example_output.json

@@ -1,17 +0,0 @@
-{
-  "width": 3,
-  "height": 2,
-  "depth": 3,
-  "layers": [
-    [
-      ["stone", "stone", "stone"],
-      ["stone", "air",   "stone"],
-      ["stone", "stone", "stone"]
-    ],
-    [
-      ["stone", "glass", "stone"],
-      ["glass", "air",   "glass"],
-      ["stone", "glass", "stone"]
-    ]
-  ]
-}

model.py

@@ -1,41 +0,0 @@
-import os
-import shutil
-import requests
-from llama_cpp import Llama
-
-MODEL_URL = "https://huggingface.co/MaziyarPanahi/Llama-3.2-1B-Instruct-GGUF/resolve/main/Llama-3.2-1B-Instruct.Q4_K_M.gguf"
-MODEL_FILENAME = "Llama-3.2-1B-Instruct.Q4_K_M.gguf"
-MODEL_PATH = f"./models/{MODEL_FILENAME}"
-
-def download_model():
-    os.makedirs("./models", exist_ok=True)
-    print("📦 Downloading GGUF model directly from URL...")
-    response = requests.get(MODEL_URL, stream=True)
-    if response.status_code == 200:
-        with open(MODEL_PATH, 'wb') as f:
-            shutil.copyfileobj(response.raw, f)
-        print(f"✅ Model downloaded to {MODEL_PATH}")
-    else:
-        raise RuntimeError(f"❌ Failed to download model. Status code: {response.status_code}")
-
-# Only download if not already present
-if not os.path.exists(MODEL_PATH):
-    download_model()
-
-# Load model with llama-cpp-python
-llm = Llama(
-    model_path=MODEL_PATH,
-    n_ctx=512,
-    n_batch=512,
-    n_threads=6,
-    chat_format="llama-3",
-    verbose=False
-)
-
-def generate_structure(prompt: str) -> str:
-    messages = [
-        {"role": "system", "content": "You are a Minecraft-style structure planner. You always respond with strictly valid JSON describing a 3D structure."},
-        {"role": "user", "content": prompt}
-    ]
-    output = llm.create_chat_completion(messages=messages)
-    return output["choices"][0]["message"]["content"]

prompt.py

@@ -1,24 +0,0 @@
-def get_strict_prompt() -> str:
-    return """
-You are a Minecraft-style structure planner.
-
-Only output a JSON object describing a 3D structure using this format:
-
-{
-  "width": <int>,
-  "height": <int>,
-  "depth": <int>,
-  "layers": [
-    [
-      ["stone", "stone", "stone"],
-      ["stone", "air",   "stone"],
-      ["stone", "stone", "stone"]
-    ],
-    ...
-  ]
-}
-
-Only use lowercase Minecraft block IDs (e.g. "stone", "air", "glass", "planks"). 
-Max size: 7×7×7. Do not include any natural language or explanation.
-Output strictly valid JSON only.
-"""

requirements.txt

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

structure_prompt_config.json

@@ -1,3 +0,0 @@
-{
-  "template": "You are a Minecraft-style structure planner. You have the curiosity to build almost anything you could think of. {structure_name}\n\nOnly output a JSON object describing a 3D structure using this format:\n\n{{\n  \"width\": <int>,\n  \"height\": <int>,\n  \"depth\": <int>,\n  \"layers\": [\n    [\n      [\"stone\", \"stone\", \"stone\"],\n      [\"stone\", \"air\",   \"stone\"],\n      [\"stone\", \"stone\", \"stone\"]\n    ],\n    ...\n  ]\n}}\n\nOnly use lowercase Minecraft block IDs (e.g. \"stone\", \"air\", \"glass\", \"planks\").\nYou could only build this structure using {blocks_allowed}\nDo not include any natural language or explanation.\nOutput strictly valid JSON only."
-}

structures.json

@@ -1,74 +0,0 @@
-[
-  {
-    "name": "castle_tower",
-    "description": "A medieval tower with battlements and defensive height.",
-    "required_blocks": ["stone_bricks", "cobblestone", "mossy_stone_bricks", "oak_planks", "torch"],
-    "tags": ["medieval", "defense"]
-  },
-  {
-    "name": "underground_bunker",
-    "description": "A secure underground base with reinforced walls and redstone lighting.",
-    "required_blocks": ["obsidian", "iron_block", "redstone_lamp", "lever", "smooth_stone"],
-    "tags": ["underground", "modern"]
-  },
-  {
-    "name": "treehouse",
-    "description": "A cozy elevated structure built into a tree canopy.",
-    "required_blocks": ["oak_logs", "oak_planks", "leaves", "ladder", "glass_pane"],
-    "tags": ["nature", "elevated"]
-  },
-  {
-    "name": "desert_hut",
-    "description": "A simple shelter for desert biomes.",
-    "required_blocks": ["sandstone", "smooth_sandstone", "cut_sandstone", "oak_door", "torch"],
-    "tags": ["desert", "village"]
-  },
-  {
-    "name": "ice_temple",
-    "description": "A mystical temple made from packed ice and glowing blocks.",
-    "required_blocks": ["packed_ice", "blue_ice", "snow_block", "sea_lantern", "light_blue_stained_glass"],
-    "tags": ["frozen", "temple"]
-  },
-  {
-    "name": "sky_platform",
-    "description": "A suspended platform in the sky, usable as base foundation.",
-    "required_blocks": ["stone_slab", "glass", "iron_block", "torch", "scaffolding"],
-    "tags": ["sky", "platform"]
-  },
-  {
-    "name": "nether_watchtower",
-    "description": "A watchtower adapted for survival in the Nether.",
-    "required_blocks": ["nether_bricks", "red_nether_bricks", "soul_torch", "basalt", "iron_bars"],
-    "tags": ["nether", "fortification"]
-  },
-  {
-    "name": "floating_island",
-    "description": "A small magical floating island with vegetation.",
-    "required_blocks": ["grass_block", "dirt", "leaves", "oak_logs", "glowstone"],
-    "tags": ["floating", "fantasy"]
-  },
-  {
-    "name": "library_room",
-    "description": "An interior room filled with books and study space.",
-    "required_blocks": ["bookshelf", "oak_planks", "lantern", "glass_pane", "carpet"],
-    "tags": ["indoor", "library"]
-  },
-  {
-    "name": "farm_stable",
-    "description": "A small stable and feeding zone for animals.",
-    "required_blocks": ["hay_block", "fence", "oak_planks", "coarse_dirt", "torch"],
-    "tags": ["farm", "animals"]
-  },
-  {
-    "name": "swamp_shack",
-    "description": "A mossy shack on stilts above swampy water.",
-    "required_blocks": ["mossy_cobblestone", "spruce_planks", "vines", "lily_pad", "oak_trapdoor"],
-    "tags": ["swamp", "shack"]
-  },
-  {
-    "name": "mine_entrance",
-    "description": "A small cave entrance with support beams and torches.",
-    "required_blocks": ["stone", "oak_logs", "rails", "torch", "gravel"],
-    "tags": ["mining", "cave"]
-  }
-]