swdeasdw

.gitignore

@@ -1,2 +1,3 @@
 tester.py
-__pycache__/
+__pycache__/
+.vscode/

app.py

@@ -1,12 +1,13 @@
+import os
+import json
+import random
 import curl_cffi
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
-import os
-
 import uvicorn
 
-# Load Gemini API key from secret
+# Load Gemini API key from environment
 GEM_API_KEY = os.getenv('GEMINI_API_KEY')
 
 # Init app
@@ -15,67 +16,90 @@ app = FastAPI()
 # CORS (allow frontend)
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Optional: restrict in prod
+    allow_origins=["*"],
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# Request model
+# Pydantic model for manual prompting
 class PromptRequest(BaseModel):
-    structure_name: str
-    structure_desc: str
-    blocks_allowed: list[str]  # Enforce lowercase block names client-side
+    prompt: str
 
-@app.post("/generate_structure")
-async def generate_with_gemini(req: PromptRequest):
-    try:
-        # 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}
+# Utility to pick a random structure and build a strict prompt
+def pick_random_structure():
+    with open("structure_list.json", "r") as f:
+        return random.choice(json.load(f))
 
-{req.structure_desc}
+def make_prompt(structure):
+    name = structure["name"]
+    desc = structure["description"]
+    blocks = " ".join(structure["blocks_allowed"])
+    return f"""You are a Minecraft-style structure planner. You have the curiosity to build almost anything you could think of. {name}
 
-Only output a JSON object describing a 3D structure using this format:
+Structure description: {desc}
+
+Only output a JSON object describing a 2D structure using this format:
 
 {{
   "width": <int>,
   "height": <int>,
-  "depth": <int>,
-  "layers": [
-    [
-      ["stone", "stone", "stone"],
-      ["stone", "air",   "stone"],
-      ["stone", "stone", "stone"]
-    ],
-    ...
+  "layout": [
+    ["stone", "air", "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)}
+You could only build this structure using {blocks}
 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(
+# Manual prompting route
+@app.post("/generate_structure")
+async def generate_with_gemini(req: PromptRequest):
+    try:
+        response = curl_cffi.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
-                    }]
-                }]
+                "contents": [
+                    {
+                        "parts": [{"text": req.prompt}]
+                    }
+                ]
             }
         )
-
         return {"response": response.text}
+    except Exception as e:
+        return {"error": str(e)}
+
+# Auto structure generation route
+@app.get("/generate_random_structure")
+async def generate_random_structure():
+    try:
+        structure = pick_random_structure()
+        prompt = make_prompt(structure)
 
+        response = curl_cffi.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.json()
     except Exception as e:
         return {"error": str(e)}
 
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+if __name__ == "__main__": uvicorn.run(app, host='localhost')

structure_list.json

@@ -0,0 +1,32 @@
+[
+  {
+    "name": "Tiny Watchtower",
+    "description": "A small elevated lookout tower made from stone and planks.",
+    "blocks_allowed": ["stone", "planks", "air"],
+    "layout": [
+      ["stone", "stone", "stone"],
+      ["planks", "air", "planks"],
+      ["stone", "stone", "stone"]
+    ]
+  },
+  {
+    "name": "Glass Hut",
+    "description": "A compact shelter made primarily from glass with a stone base.",
+    "blocks_allowed": ["glass", "stone", "air"],
+    "layout": [
+      ["glass", "glass", "glass"],
+      ["glass", "air", "glass"],
+      ["stone", "stone", "stone"]
+    ]
+  },
+  {
+    "name": "Lantern Post",
+    "description": "A single vertical pillar with a torch at the top.",
+    "blocks_allowed": ["stone", "torch"],
+    "layout": [
+      ["torch"],
+      ["stone"],
+      ["stone"]
+    ]
+  }
+]

utils.py

@@ -0,0 +1,32 @@
+import json
+import random
+
+def pick_random_structure():
+    with open("structure_list.json", "r") as f:
+        return random.choice(json.load(f))
+
+def make_prompt(structure):
+    name = structure["name"]
+    desc = structure["description"]
+    blocks = " ".join(structure["blocks_allowed"])
+
+    return f"""You are a Minecraft-style structure planner. You have the curiosity to build almost anything you could think of. {name}
+
+Structure description: {desc}
+
+Only output a JSON object describing a 2D structure using this format:
+
+{{
+  "width": <int>,
+  "height": <int>,
+  "layout": [
+    ["stone", "air", "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 {blocks}
+Do not include any natural language or explanation.
+Output strictly valid JSON only."""