Spaces:

build-small-hackathon
/

ThreeGen

Running on Zero

App Files Files Community

bolajiev commited on 21 days ago

Commit

f0a5709

1 Parent(s): 5f02e75

Add 5 new shapes: capsule, ring, circle, tube, roundedBox (+ FEWSHOT examples)

Browse files

Files changed (3) hide show

compiler.py +20 -0
llm.py +41 -1
scene.py +11 -0

compiler.py CHANGED Viewed

@@ -45,6 +45,25 @@ def geometry_js(o: Obj) -> str:
     }
     if s in poly:
         return f"new THREE.{poly[s]}({g('radius', 0.6)}, 0)"
     return "new THREE.BoxGeometry(1, 1, 1)"
@@ -207,6 +226,7 @@ TEMPLATE = """<!DOCTYPE html>
   import { EffectComposer } from 'three/addons/postprocessing/EffectComposer.js';
   import { RenderPass } from 'three/addons/postprocessing/RenderPass.js';
   import { UnrealBloomPass } from 'three/addons/postprocessing/UnrealBloomPass.js';
   const scene = new THREE.Scene();
   scene.background = new THREE.Color('__BG__');

     }
     if s in poly:
         return f"new THREE.{poly[s]}({g('radius', 0.6)}, 0)"
+    if s == "capsule":
+        return f"new THREE.CapsuleGeometry({g('radius', 0.4)}, {g('length', 1.0)}, 8, 16)"
+    if s == "ring":
+        return f"new THREE.RingGeometry({g('innerRadius', 0.3)}, {g('outerRadius', 0.6)}, 32)"
+    if s == "circle":
+        return f"new THREE.CircleGeometry({g('radius', 0.6)}, 32)"
+    if s == "tube":
+        r = g('radius', 0.1)
+        return (
+            f"(()=>{{ const _c=new THREE.CatmullRomCurve3(["
+            f"new THREE.Vector3(-0.4,-0.6,0),"
+            f"new THREE.Vector3(0.4,-0.2,0),"
+            f"new THREE.Vector3(-0.4,0.2,0),"
+            f"new THREE.Vector3(0.4,0.6,0)"
+            f"]); return new THREE.TubeGeometry(_c,64,{r},12,false); }})()"
+        )
+    if s == "roundedBox":
+        return (f"new RoundedBoxGeometry({g('width', 1.0)}, {g('height', 1.0)}, "
+                f"{g('depth', 1.0)}, 4, {g('radius', 0.1)})")
     return "new THREE.BoxGeometry(1, 1, 1)"
   import { EffectComposer } from 'three/addons/postprocessing/EffectComposer.js';
   import { RenderPass } from 'three/addons/postprocessing/RenderPass.js';
   import { UnrealBloomPass } from 'three/addons/postprocessing/UnrealBloomPass.js';
+  import { RoundedBoxGeometry } from 'three/addons/geometries/RoundedBoxGeometry.js';
   const scene = new THREE.Scene();
   scene.background = new THREE.Color('__BG__');

llm.py CHANGED Viewed

@@ -29,7 +29,7 @@ Schema:
   "background": "#RRGGBB",
   "objects": [
     {
-      "shape": "box|sphere|cylinder|cone|torus|torusKnot|plane|tetrahedron|icosahedron|dodecahedron|octahedron",
       "position": [x, y, z],
       "rotation": [x, y, z],          // radians, optional
       "scale": [x, y, z],             // optional
@@ -127,6 +127,46 @@ FEWSHOT = [
         ],
         "animation": {"type": "rotate", "speed": 0.5, "axis": "y"},
     })},
 ]
 _tok = None

   "background": "#RRGGBB",
   "objects": [
     {
+      "shape": "box|sphere|cylinder|cone|torus|torusKnot|plane|tetrahedron|icosahedron|dodecahedron|octahedron|capsule|ring|circle|tube|roundedBox",
       "position": [x, y, z],
       "rotation": [x, y, z],          // radians, optional
       "scale": [x, y, z],             // optional
         ],
         "animation": {"type": "rotate", "speed": 0.5, "axis": "y"},
     })},
+    {"role": "user", "content": "a cute capsule character with a rounded body"},
+    {"role": "assistant", "content": json.dumps({
+        "background": "#0d1b2a",
+        "objects": [
+            {"shape": "capsule", "color": "#ff6b6b", "emissive": "#330000", "material": "standard",
+             "metalness": 0.1, "roughness": 0.5, "position": [0, 0, 0],
+             "params": {"radius": 0.35, "length": 0.6}},
+            {"shape": "sphere", "color": "#ffe8cc", "material": "standard",
+             "metalness": 0.0, "roughness": 0.8, "position": [0, 0.72, 0],
+             "params": {"radius": 0.28}},
+        ],
+        "lights": [
+            {"type": "ambient", "intensity": 0.5},
+            {"type": "directional", "intensity": 1.3, "position": [5, 8, 6]},
+        ],
+        "animation": {"type": "float", "speed": 1.0, "axis": "y"},
+    })},
+    {"role": "user", "content": "a stack of soft rounded boxes"},
+    {"role": "assistant", "content": json.dumps({
+        "background": "#0b0e14",
+        "objects": [{
+            "type": "stack", "axis": "y", "gap": 0.08,
+            "children": [
+                {"shape": "roundedBox", "color": "#ff9f43", "material": "standard",
+                 "metalness": 0.1, "roughness": 0.4,
+                 "params": {"width": 1.2, "height": 0.4, "depth": 1.2, "radius": 0.12}},
+                {"shape": "roundedBox", "color": "#ee5a24", "material": "standard",
+                 "metalness": 0.1, "roughness": 0.4,
+                 "params": {"width": 1.0, "height": 0.4, "depth": 1.0, "radius": 0.12}},
+                {"shape": "roundedBox", "color": "#c0392b", "material": "standard",
+                 "metalness": 0.1, "roughness": 0.4,
+                 "params": {"width": 0.8, "height": 0.4, "depth": 0.8, "radius": 0.12}},
+            ],
+        }],
+        "lights": [
+            {"type": "ambient", "intensity": 0.5},
+            {"type": "directional", "intensity": 1.3, "position": [5, 8, 6]},
+        ],
+        "animation": {"type": "rotate", "speed": 0.8, "axis": "y"},
+    })},
 ]
 _tok = None

scene.py CHANGED Viewed

@@ -19,6 +19,7 @@ log = logging.getLogger(__name__)
 SHAPES = {
     "box", "sphere", "cylinder", "cone", "torus", "torusKnot", "plane",
     "tetrahedron", "icosahedron", "dodecahedron", "octahedron",
 }
 MATERIALS = {"standard", "basic", "phong", "wireframe"}
 LIGHT_TYPES = {"ambient", "directional", "point"}
@@ -51,6 +52,16 @@ def _shape_extent(shape: str, params: Dict[str, float]) -> tuple:
         return (d, d, d)
     if shape == "plane":
         return (p("width", 5.0), 0.01, p("height", 5.0))
     return (1.0, 1.0, 1.0)

 SHAPES = {
     "box", "sphere", "cylinder", "cone", "torus", "torusKnot", "plane",
     "tetrahedron", "icosahedron", "dodecahedron", "octahedron",
+    "capsule", "ring", "circle", "tube", "roundedBox",
 }
 MATERIALS = {"standard", "basic", "phong", "wireframe"}
 LIGHT_TYPES = {"ambient", "directional", "point"}
         return (d, d, d)
     if shape == "plane":
         return (p("width", 5.0), 0.01, p("height", 5.0))
+    if shape == "capsule":
+        d = p("radius", 0.4) * 2
+        return (d, p("length", 1.0) + d, d)
+    if shape in ("ring", "circle"):
+        r = p("outerRadius", p("radius", 0.6)) * 2
+        return (r, 0.01, r)
+    if shape == "tube":
+        return (1.0, 1.5, 1.0)
+    if shape == "roundedBox":
+        return (p("width", 1.0), p("height", 1.0), p("depth", 1.0))
     return (1.0, 1.0, 1.0)