Squash commits

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,21 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+
+# Gurobi files
+gurobi.lic
+
+# LDraw files
+ldraw
+complete.zip
+
+.idea
+.gradio
+out

.gitmodules

@@ -0,0 +1,4 @@
+[submodule "ImportLDraw"]
+	path = ImportLDraw
+	url = https://github.com/sgebbie/ImportLDraw.git
+	branch = bugfix/material-fabric-warning

.python-version

@@ -0,0 +1 @@
+3.11

ImportLDraw

@@ -0,0 +1 @@
+Subproject commit 7fe64deef48c8f726e44a30341d04b5fa396ff35

README.md

@@ -1,13 +1,76 @@
 ---
-title: LegoGPT Demo
-emoji: 📉
-colorFrom: gray
-colorTo: red
+title: LegoGPT-Demo
+emoji: 🧱
+short_description: Generate stable LEGO structures from text prompts.
+app_file: app.py
 sdk: gradio
 sdk_version: 5.29.0
-app_file: app.py
-pinned: false
-short_description: Official demo for LegoGPT.
+python_version: 3.11
+models:
+  - AvaLovelace/LegoGPT
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# LegoGPT Demo
+
+Gradio demo for LegoGPT.
+
+## Prerequisites
+
+- **Llama-3.2-1B-Instruct:** LegoGPT is fine-tuned from meta-llama/Llama-3.2-1B-Instruct, a gated model. Request access
+  to the model [here](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct), then generate
+  a [Hugging Face user access token](https://huggingface.co/docs/hub/en/security-tokens).
+    - *If running locally,* set your access token as an environment variable: `export HF_TOKEN=<your_token>`
+    - *If running on Hugging Face Spaces,* set your access token as the `HF_TOKEN` secret in the Settings tab of your
+      Space.
+
+  The model will be
+  automatically downloaded upon running the code.
+- **Gurobi:** Running stability analysis requires a [Gurobi licence](https://www.gurobi.com/downloads/) to use Gurobi.
+  Academics may request a free licence from the Gurobi
+  website [here](https://www.gurobi.com/academia/academic-program-and-licenses/).
+    - *If running locally,* place the Gurobi licence file in your *home directory* or
+      another [recommended location](https://support.gurobi.com/hc/en-us/articles/360013417211-Where-do-I-place-the-Gurobi-license-file-gurobi-lic).
+    - *If running on Hugging Face Spaces,* the licence type must be **Web License Service (WLS)**. Set the
+      `WLSACCESSID`, `WLSSECRET`, and `LICENSEID` secrets in the Settings tab of your Space to their values in your
+      Gurobi licence file.
+- **ImportLDraw:** Rendering LEGO visualizations requires ImportLDraw, provided as a Git submodule.
+    - *If running locally,* follow these instructions to install ImportLDraw:
+        - Download [Git LFS](https://git-lfs.com), then run `git lfs install`.
+        - Install Git submodules with `git submodule update --init`.
+        - Download the [LDraw parts library](https://library.ldraw.org/library/updates/complete.zip) and
+          extract it in your *home directory*:
+          `(cd ~ && wget https://library.ldraw.org/library/updates/complete.zip && unzip complete.zip)`.
+            - If you wish to put the LDraw parts library in a different directory, set the environment variable
+              `LDRAW_LIBRARY_PATH` to the path of the `ldraw` directory: `export LDRAW_LIBRARY_PATH=path/to/ldraw`.
+    - *If running on Hugging Face Spaces,* ImportLDraw and the LDraw parts library will automatically be downloaded and
+      installed by the `app.py` script.
+
+## Running locally
+
+Install the Python project manager [uv](https://docs.astral.sh/uv). Then run the demo with:
+
+```zsh
+uv run app.py
+```
+
+## Running on Hugging Face Spaces
+
+Make sure the origin of this repo is set to your Space:
+
+```zsh
+git remote set-url origin [email protected]:spaces/<your_username>/<your_space_name>
+```
+
+Then commit and push your changes to your Space with
+
+```
+git add -A && git commit -m "Update" && git push
+```
+
+> [!NOTE]
+> If you've changed the dependencies in `pyproject.toml`, update the `requirements.txt` file accordingly:
+> ```zsh
+> uv export --format requirements-txt --no-hashes > requirements.txt
+> ```
+> Then, `git commit` and `git push` the changes to your Space. You may have to restart the Space for the changes to take
+> effect.

app.py

@@ -0,0 +1,150 @@
+import json
+import os
+import subprocess
+import time
+import uuid
+import zipfile
+from dataclasses import fields
+from urllib.request import urlretrieve
+
+import gradio as gr
+import transformers
+from legogpt.models import LegoGPT, LegoGPTConfig
+
+
+def setup():
+    # Set up Gurobi licence
+    licence_filename = 'gurobi.lic'
+    licence_lines = []
+    for secret_name in ['WLSACCESSID', 'WLSSECRET', 'LICENSEID']:
+        secret = os.environ.get(secret_name)
+        if not secret:
+            raise ValueError(f'Env variable {secret_name} not found. Please set it in the Hugging Face Space settings.')
+        licence_lines.append(f'{secret_name}={secret}\n')
+    with open(licence_filename, 'w') as f:
+        f.writelines(licence_lines)
+    os.environ['GRB_LICENSE_FILE'] = os.path.abspath(licence_filename)
+
+    # Download LDraw part library and set LDraw library path
+    ldraw_zip_url = 'https://library.ldraw.org/library/updates/complete.zip'
+    ldraw_zip_filename = 'complete.zip'
+    urlretrieve(ldraw_zip_url, ldraw_zip_filename)
+    with zipfile.ZipFile(ldraw_zip_filename) as zip_ref:
+        zip_ref.extractall()
+    os.environ['LDRAW_LIBRARY_PATH'] = os.path.abspath('ldraw')
+
+
+def main():
+    if os.environ.get('IS_HF_SPACE') == '1':
+        print('Running in Hugging Face Space, setting up environment...')
+        setup()
+
+    model_cfg = LegoGPTConfig(max_regenerations=10)
+    model = LegoGPT(model_cfg)
+
+    def generate_lego(
+            prompt: str,
+            temperature: float | None,
+            seed: int | None,
+            max_bricks: int | None,
+            max_brick_rejections: int | None,
+            max_regenerations: int | None,
+    ):
+        # Set model parameters
+        if temperature is not None: model.temperature = temperature
+        if max_bricks is not None: model.max_bricks = max_bricks
+        if max_brick_rejections is not None: model.max_brick_rejections = max_brick_rejections
+        if max_regenerations is not None: model.max_regenerations = max_regenerations
+        if seed is not None: transformers.set_seed(seed)
+
+        # Generate LEGO
+        print(f'Generating LEGO for prompt: "{prompt}"')
+        start_time = time.time()
+        output = model(prompt)
+
+        # Write output LDR to file
+        output_dir = os.path.abspath('out')
+        output_uuid = str(uuid.uuid4())
+        os.makedirs(output_dir, exist_ok=True)
+        ldr_filename = os.path.join(output_dir, f'{output_uuid}.ldr')
+        with open(ldr_filename, 'w') as f:
+            f.write(output['lego'].to_ldr())
+        print(f'Finished generation in {time.time() - start_time:.1f}s!')
+
+        # Render LEGO model to image
+        print('Rendering image...')
+        start_time = time.time()
+        img_filename = os.path.join(output_dir, f'{output_uuid}.png')
+        subprocess.run(['python', 'render_lego.py', '--in_file', ldr_filename, '--out_file', img_filename],
+                       check=True)  # Run render as a subprocess to prevent issues with Blender
+        print(f'Finished rendering in {time.time() - start_time:.1f}s!')
+
+        return img_filename, output['lego']
+
+    # Define inputs and outputs
+    in_prompt = gr.Textbox(label='Prompt', placeholder='Enter a prompt to generate a LEGO model.')
+    in_temperature = gr.Slider(0.01, 2.0, value=model_cfg.temperature, step=0.01,
+                               label='Temperature', info=get_help_string('temperature'))
+    in_seed = gr.Number(value=42, label='Seed', info='Random seed for generation.', precision=0, step=1)
+    in_bricks = gr.Number(value=model_cfg.max_bricks, label='Max bricks', info=get_help_string('max_bricks'),
+                          precision=0, minimum=1, step=1)
+    in_rejections = gr.Number(value=model_cfg.max_brick_rejections, label='Max brick rejections',
+                              info=get_help_string('max_brick_rejections'), precision=0, minimum=0, step=1)
+    in_regenerations = gr.Number(value=model_cfg.max_regenerations, label='Max regenerations',
+                                 info=get_help_string('max_regenerations'), precision=0, minimum=0, step=1)
+    out_img = gr.Image(label='Output image', format='png')
+    out_txt = gr.Textbox(label='Output LEGO bricks', lines=5, max_lines=5, show_copy_button=True,
+                         info='The LEGO structure in text format. Each line of the form "hxw (x,y,z)" represents a '
+                              '1-unit-tall rectangular brick with dimensions hxw placed at coordinates (x,y,z).')
+
+    # Define Gradio interface
+    demo = gr.Interface(
+        fn=generate_lego,
+        title='LegoGPT Demo',
+        description='Official demo for [LegoGPT](https://avalovelace1.github.io/LegoGPT/), the first approach for generating physically stable LEGO brick models from text prompts.\n\n'
+                    'The model is restricted to creating structures made of 1-unit-tall cuboid bricks on a 20x20x20 grid. It was trained on a dataset of 21 object categories: '
+                    '*basket, bed, bench, birdhouse, bookshelf, bottle, bowl, bus, camera, car, chair, guitar, jar, mug, piano, pot, sofa, table, tower, train, vessel.* '
+                    'Performance on prompts from outside these categories may be limited. This demo does not include texturing or coloring.',
+        inputs=[in_prompt],
+        additional_inputs=[in_temperature, in_seed, in_bricks, in_rejections, in_regenerations],
+        outputs=[out_img, out_txt],
+        flagging_mode='never',
+    )
+    with demo:
+        with gr.Row():
+            examples = get_examples()
+            dummy_name = gr.Textbox(visible=False, label='Name')
+            dummy_out_img = gr.Image(visible=False, label='Result')
+            gr.Examples(
+                examples=[[name, example['prompt'], example['temperature'], example['seed'], example['output_img']]
+                          for name, example in examples.items()],
+                inputs=[dummy_name, in_prompt, in_temperature, in_seed, dummy_out_img],
+                outputs=[out_img, out_txt],
+                fn=lambda *args: (args[-1], examples[args[0]]['output_txt']),
+                run_on_click=True,
+            )
+    demo.launch(share=True)
+
+
+def get_help_string(field_name: str) -> str:
+    """
+    :param field_name: Name of a field in LegoGPTConfig.
+    :return: Help string for the field.
+    """
+    data_fields = fields(LegoGPTConfig)
+    name_field = next(f for f in data_fields if f.name == field_name)
+    return name_field.metadata['help']
+
+
+def get_examples(example_dir: str = os.path.abspath('examples')) -> dict[str, dict[str, str]]:
+    examples_file = os.path.join(example_dir, 'examples.json')
+    with open(examples_file) as f:
+        examples = json.load(f)
+
+    for example in examples.values():
+        example['output_img'] = os.path.join(example_dir, example['output_img'])
+    return examples
+
+
+if __name__ == '__main__':
+    main()

examples/examples.json

@@ -0,0 +1,44 @@
+{
+  "table": {
+    "prompt": "Table featuring a flat rectangular surface over four evenly spaced legs.",
+    "temperature": 0.6,
+    "seed": 42,
+    "output_img": "9c1b0058dfe027cbf519adc9991b5f11.png",
+    "output_txt": "1x2 (16,18,0)\n1x2 (16,13,0)\n2x2 (0,18,0)\n2x2 (0,13,0)\n1x2 (16,18,1)\n1x2 (16,13,1)\n2x2 (0,18,1)\n2x2 (0,13,1)\n1x1 (16,18,2)\n1x2 (16,13,2)\n2x1 (0,18,2)\n2x2 (0,13,2)\n1x2 (16,17,3)\n1x1 (16,14,3)\n2x2 (0,17,3)\n2x1 (0,14,3)\n1x2 (16,17,4)\n1x2 (16,14,4)\n2x2 (0,17,4)\n2x2 (0,14,4)\n1x2 (16,17,5)\n1x2 (16,14,5)\n2x2 (0,17,5)\n2x2 (0,14,5)\n1x1 (16,18,6)\n1x2 (16,16,6)\n1x1 (16,14,6)\n8x1 (8,17,6)\n2x1 (0,18,6)\n8x1 (0,17,6)\n2x1 (0,16,6)\n2x1 (0,14,6)\n2x6 (16,13,7)\n1x6 (15,13,7)\n2x6 (13,13,7)\n2x6 (11,13,7)\n8x1 (10,19,7)\n2x6 (9,13,7)\n6x2 (3,17,7)\n6x2 (3,15,7)\n6x2 (3,13,7)\n8x1 (2,19,7)\n1x6 (2,13,7)\n2x6 (0,14,7)\n2x1 (0,13,7)\n1x6 (17,14,8)\n2x6 (15,14,8)\n1x6 (14,14,8)\n2x6 (12,14,8)\n8x1 (10,13,8)\n4x2 (8,14,8)\n2x1 (8,13,8)\n6x2 (6,18,8)\n6x2 (6,16,8)\n2x2 (6,14,8)\n2x6 (4,14,8)\n2x6 (2,14,8)\n2x6 (0,14,8)\n8x1 (0,13,8)\n"
+  },
+  "sofa": {
+    "prompt": "Compact sofa with a geometric design.",
+    "temperature": 0.6,
+    "seed": 42,
+    "output_img": "8981c0ffae1af00a50fc88ed745bdb67.png",
+    "output_txt": "1x2 (16,17,0)\n1x1 (16,13,0)\n2x2 (0,17,0)\n2x1 (0,13,0)\n1x1 (17,18,1)\n1x6 (17,12,1)\n2x6 (15,12,1)\n1x6 (14,12,1)\n8x1 (9,18,1)\n1x1 (8,18,1)\n6x2 (8,16,1)\n6x2 (8,14,1)\n6x2 (8,12,1)\n6x2 (2,16,1)\n6x2 (2,14,1)\n6x2 (2,12,1)\n8x1 (0,18,1)\n2x6 (0,12,1)\n2x6 (16,13,2)\n2x1 (16,12,2)\n1x1 (15,18,2)\n2x6 (14,12,2)\n1x6 (13,12,2)\n2x6 (11,12,2)\n6x1 (9,18,2)\n2x6 (9,12,2)\n2x6 (7,13,2)\n2x6 (5,13,2)\n4x1 (5,12,2)\n1x1 (4,18,2)\n2x6 (3,12,2)\n2x1 (2,18,2)\n2x6 (1,12,2)\n2x1 (0,18,2)\n1x6 (0,12,2)\n1x6 (17,12,3)\n2x1 (16,18,3)\n1x1 (16,17,3)\n1x2 (15,17,3)\n2x4 (15,13,3)\n2x1 (15,12,3)\n2x1 (13,18,3)\n8x1 (5,18,3)\n1x1 (4,18,3)\n2x1 (2,18,3)\n2x6 (0,13,3)\n2x1 (0,12,3)\n1x1 (17,18,4)\n2x6 (16,12,4)\n1x2 (15,17,4)\n6x2 (9,18,4)\n1x2 (8,18,4)\n6x2 (2,18,4)\n2x1 (0,18,4)\n2x6 (0,12,4)\n2x1 (16,16,5)\n2x4 (16,12,5)\n6x2 (12,17,5)\n8x1 (9,19,5)\n1x1 (8,19,5)\n6x2 (6,17,5)\n8x1 (0,19,5)\n6x2 (0,17,5)\n2x1 (0,16,5)\n2x4 (0,12,5)\n1x2 (17,18,6)\n2x1 (16,17,6)\n4x2 (13,18,6)\n6x2 (7,18,6)\n1x2 (6,18,6)\n6x2 (0,18,6)\n2x1 (0,17,6)\n1x2 (16,18,7)\n2x2 (14,18,7)\n6x2 (8,18,7)\n6x2 (2,18,7)\n2x2 (0,18,7)\n"
+  },
+  "car": {
+    "prompt": "Small car featuring a rectangular body, flat top, and stepped edges.",
+    "temperature": 0.6,
+    "seed": 42,
+    "output_img": "60066a5369f1354e631a23fef2ba638b.png",
+    "output_txt": "1x2 (7,15,0)\n1x1 (7,13,0)\n1x2 (7,5,0)\n2x2 (6,7,0)\n1x1 (6,5,0)\n2x6 (5,11,0)\n2x2 (5,3,0)\n2x6 (3,11,0)\n2x6 (3,3,0)\n2x6 (1,11,0)\n6x2 (1,9,0)\n2x2 (1,3,0)\n1x2 (0,15,0)\n1x1 (0,13,0)\n2x2 (0,7,0)\n1x2 (0,5,0)\n1x1 (7,18,1)\n1x4 (7,8,1)\n2x6 (6,12,1)\n2x6 (6,2,1)\n1x1 (5,19,1)\n2x6 (4,12,1)\n2x2 (4,2,1)\n2x1 (3,19,1)\n2x6 (2,12,1)\n2x2 (2,2,1)\n6x2 (1,10,1)\n6x2 (1,8,1)\n2x6 (0,14,1)\n2x2 (0,12,1)\n1x4 (0,8,1)\n6x2 (0,6,1)\n6x2 (0,4,1)\n2x1 (0,3,1)\n1x1 (7,18,2)\n1x4 (7,13,2)\n1x1 (7,2,2)\n2x6 (6,7,2)\n2x2 (6,5,2)\n2x1 (6,3,2)\n2x6 (5,14,2)\n1x6 (5,7,2)\n1x2 (5,3,2)\n2x6 (3,14,2)\n2x6 (3,7,2)\n1x2 (3,5,2)\n2x2 (3,3,2)\n4x1 (3,2,2)\n1x1 (2,2,2)\n2x6 (1,14,2)\n6x1 (1,13,2)\n2x4 (1,9,2)\n2x6 (1,3,2)\n1x1 (0,18,2)\n1x4 (0,13,2)\n1x1 (0,12,2)\n1x1 (7,13,3)\n1x2 (7,11,3)\n1x8 (7,3,3)\n2x2 (6,18,3)\n2x1 (6,16,3)\n2x2 (6,14,3)\n1x1 (5,19,3)\n1x2 (5,15,3)\n1x1 (4,16,3)\n2x1 (3,19,3)\n1x2 (3,15,3)\n6x1 (1,17,3)\n2x2 (1,15,3)\n2x1 (1,14,3)\n1x1 (1,13,3)\n6x2 (1,11,3)\n6x2 (1,9,3)\n6x2 (1,7,3)\n6x2 (1,5,3)\n6x2 (1,3,3)\n2x1 (0,18,3)\n1x1 (0,15,3)\n1x1 (0,14,3)\n1x1 (0,12,3)\n1x1 (7,18,4)\n2x6 (6,12,4)\n2x2 (6,10,4)\n1x1 (4,9,4)\n4x1 (3,18,4)\n1x2 (3,16,4)\n4x2 (2,10,4)\n1x1 (1,18,4)\n2x6 (1,12,4)\n1x1 (1,11,4)\n1x1 (1,10,4)\n1x1 (0,12,4)\n"
+  },
+  "train": {
+    "prompt": "Train with rectangular body and geometric components.",
+    "temperature": 0.6,
+    "seed": 42,
+    "output_img": "e5c8853f28328d0c2511f68da65f4c4.png",
+    "output_txt": "1x2 (3,17,0)\n1x2 (3,13,0)\n2x2 (2,2,0)\n2x2 (1,17,0)\n2x2 (1,13,0)\n1x2 (0,17,0)\n4x1 (0,6,0)\n2x2 (0,2,0)\n1x1 (3,18,1)\n1x2 (3,16,1)\n1x2 (3,12,1)\n1x2 (3,2,1)\n2x2 (2,14,1)\n2x6 (2,6,1)\n1x1 (2,3,1)\n2x1 (1,18,1)\n2x2 (1,16,1)\n1x2 (1,14,1)\n1x6 (1,6,1)\n2x1 (1,2,1)\n1x1 (0,18,1)\n1x2 (0,16,1)\n2x2 (0,12,1)\n1x2 (0,10,1)\n1x8 (0,2,1)\n1x2 (3,9,2)\n1x1 (3,2,2)\n2x6 (2,13,2)\n2x4 (2,5,2)\n2x1 (1,2,2)\n2x6 (0,13,2)\n4x2 (0,11,2)\n1x2 (0,9,2)\n2x4 (0,5,2)\n4x2 (0,3,2)\n1x1 (0,2,2)\n1x8 (3,10,3)\n1x6 (3,4,3)\n2x6 (1,12,3)\n2x6 (1,6,3)\n2x2 (1,4,3)\n4x1 (0,18,3)\n1x8 (0,10,3)\n1x6 (0,4,3)\n4x2 (0,2,3)\n2x6 (2,10,4)\n2x6 (2,3,4)\n4x1 (0,18,4)\n4x2 (0,16,4)\n2x6 (0,10,4)\n4x1 (0,9,4)\n2x6 (0,3,4)\n1x2 (2,17,5)\n1x4 (2,11,5)\n2x2 (1,9,5)\n2x2 (0,17,5)\n2x2 (0,13,5)\n2x2 (0,11,5)\n1x2 (0,9,5)\n"
+  },
+  "chair-1": {
+    "prompt": "Square-seated chair featuring an upright, rectangular backrest and straight legs.",
+    "temperature": 0.6,
+    "seed": 42,
+    "output_img": "e488826128fe3854b300c4ca2f51c01b.png",
+    "output_txt": "1x2 (7,18,0)\n1x2 (7,11,0)\n2x2 (0,18,0)\n2x2 (0,11,0)\n1x2 (7,18,1)\n1x2 (7,11,1)\n2x2 (0,18,1)\n2x2 (0,11,1)\n1x1 (7,18,2)\n1x2 (7,11,2)\n2x1 (0,18,2)\n2x2 (0,11,2)\n1x2 (7,17,3)\n1x2 (7,11,3)\n2x2 (0,17,3)\n2x2 (0,11,3)\n1x2 (7,17,4)\n1x2 (7,11,4)\n2x2 (0,17,4)\n2x2 (0,11,4)\n1x2 (7,17,5)\n1x2 (7,11,5)\n2x2 (0,17,5)\n2x2 (0,11,5)\n1x1 (7,18,6)\n1x2 (7,11,6)\n2x1 (6,17,6)\n1x1 (6,12,6)\n1x1 (1,12,6)\n6x1 (0,17,6)\n2x1 (0,18,6)\n1x1 (0,12,6)\n2x1 (0,11,6)\n1x6 (7,13,7)\n6x2 (2,11,7)\n6x1 (1,18,7)\n1x1 (1,17,7)\n1x4 (1,13,7)\n1x6 (0,13,7)\n2x2 (0,11,7)\n1x2 (7,17,8)\n1x1 (7,16,8)\n2x2 (7,14,8)\n1x2 (7,12,8)\n1x1 (6,18,8)\n4x1 (2,18,8)\n6x2 (1,16,8)\n6x2 (1,14,8)\n6x2 (1,12,8)\n2x1 (0,18,8)\n1x6 (0,12,8)\n2x6 (6,13,9)\n2x1 (6,12,9)\n2x6 (4,13,9)\n2x6 (2,13,9)\n4x1 (2,12,9)\n1x1 (1,18,9)\n1x6 (1,12,9)\n1x2 (7,17,10)\n6x2 (1,17,10)\n6x2 (2,17,11)\n1x2 (1,17,11)\n1x2 (7,17,12)\n6x2 (1,17,12)\n6x2 (2,17,13)\n1x2 (1,17,13)\n1x2 (7,17,14)\n6x2 (1,17,14)\n6x2 (2,17,15)\n1x2 (1,17,15)\n1x2 (7,18,16)\n6x2 (1,18,16)\n6x2 (2,18,17)\n1x2 (1,18,17)\n"
+  },
+  "chair-2": {
+    "prompt": "Compact chair with a tall backrest and serrated seat.",
+    "temperature": 0.6,
+    "seed": 42,
+    "output_img": "60790035c8126a677645b4fdaedbc34.png",
+    "output_txt": "1x2 (7,17,0)\n1x2 (7,13,0)\n2x2 (0,17,0)\n2x2 (0,13,0)\n1x2 (7,17,1)\n1x2 (7,13,1)\n2x2 (0,17,1)\n2x2 (0,13,1)\n1x1 (7,18,2)\n1x6 (7,12,2)\n2x2 (0,17,2)\n2x2 (0,13,2)\n1x2 (7,17,3)\n1x2 (7,12,3)\n2x2 (0,17,3)\n2x2 (0,12,3)\n1x2 (7,17,4)\n1x2 (7,12,4)\n2x2 (0,17,4)\n2x2 (0,12,4)\n1x2 (7,17,5)\n1x2 (7,12,5)\n2x2 (0,17,5)\n2x2 (0,12,5)\n1x1 (7,18,6)\n1x6 (7,12,6)\n4x1 (3,17,6)\n4x1 (3,13,6)\n1x1 (2,17,6)\n1x1 (2,13,6)\n2x1 (0,18,6)\n2x6 (0,12,6)\n1x1 (7,18,7)\n1x1 (7,12,7)\n6x1 (2,17,7)\n6x1 (2,13,7)\n2x1 (0,18,7)\n2x6 (0,12,7)\n1x1 (7,18,8)\n1x1 (7,12,8)\n2x1 (0,18,8)\n2x1 (0,12,8)\n1x8 (7,11,9)\n1x1 (6,18,9)\n1x1 (6,17,9)\n1x6 (6,11,9)\n4x1 (2,18,9)\n4x1 (2,11,9)\n2x6 (0,13,9)\n2x2 (0,11,9)\n1x2 (7,17,10)\n2x6 (6,11,10)\n1x6 (5,11,10)\n2x6 (3,11,10)\n6x2 (1,17,10)\n2x6 (1,11,10)\n1x1 (0,18,10)\n1x6 (0,12,10)\n1x1 (7,18,11)\n6x1 (1,18,11)\n1x2 (7,18,12)\n1x2 (6,18,12)\n4x2 (2,18,12)\n1x1 (1,18,12)\n1x1 (7,18,13)\n6x1 (2,19,13)\n1x1 (1,19,13)\n6x1 (1,18,13)\n1x2 (7,18,14)\n6x2 (1,18,14)\n6x2 (2,18,15)\n1x2 (1,18,15)\n1x2 (7,18,16)\n6x2 (1,18,16)\n1x2 (5,18,17)\n2x2 (3,18,17)\n"
+  }
+}

pyproject.toml

@@ -0,0 +1,14 @@
+[project]
+name = "legogpt-demo"
+version = "0.1.0"
+description = "LegoGPT demo webapp"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "bpy<=4.3.0", # lower version of bpy needed to run on Hugging Face A100
+    "gradio>=5.29.0",
+    "legogpt",
+]
+
+[tool.uv.sources]
+legogpt = { git = "https://github.com/AvaLovelace1/LegoGPT.git" }

render_lego.py

@@ -0,0 +1,4 @@
+from legogpt.render_lego import main as render_lego_main
+
+if __name__ == '__main__':
+    render_lego_main()

requirements.txt

@@ -0,0 +1,91 @@
+# This file was autogenerated by uv via the following command:
+#    uv export --format requirements-txt --no-hashes
+accelerate==1.6.0
+aiofiles==24.1.0
+annotated-types==0.7.0
+anyio==4.9.0
+audioop-lts==0.2.1 ; python_full_version >= '3.13'
+bpy==4.3.0
+certifi==2025.4.26
+cffi==1.17.1 ; platform_python_implementation == 'PyPy'
+charset-normalizer==3.4.2
+click==8.1.8 ; sys_platform != 'emscripten'
+colorama==0.4.6 ; sys_platform == 'win32'
+cython==3.1.0
+fastapi==0.115.12
+ffmpy==0.5.0
+filelock==3.18.0
+fsspec==2025.3.0
+gradio==5.29.0
+gradio-client==1.10.0
+groovy==0.1.2
+gurobipy==12.0.2
+h11==0.16.0
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.30.2
+idna==3.10
+jinja2==3.1.6
+legogpt @ git+https://github.com/AvaLovelace1/LegoGPT.git@c95d35119d98ee4833d2c74d8d6046e5835c2c1c
+markdown-it-py==3.0.0 ; sys_platform != 'emscripten'
+markupsafe==3.0.2
+mdurl==0.1.2 ; sys_platform != 'emscripten'
+mpmath==1.3.0
+networkx==3.4.2
+numpy==1.26.4
+nvidia-cublas-cu12==12.6.4.1 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cuda-cupti-cu12==12.6.80 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cuda-nvrtc-cu12==12.6.77 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cuda-runtime-cu12==12.6.77 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cudnn-cu12==9.5.1.17 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cufft-cu12==11.3.0.4 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cufile-cu12==1.11.1.6 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-curand-cu12==10.3.7.77 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cusolver-cu12==11.7.1.2 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cusparse-cu12==12.5.4.2 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-cusparselt-cu12==0.6.3 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-nccl-cu12==2.26.2 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-nvjitlink-cu12==12.6.85 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+nvidia-nvtx-cu12==12.6.77 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+orjson==3.10.18
+packaging==25.0
+pandas==2.2.3
+peft==0.15.2
+pillow==11.2.1
+psutil==7.0.0
+pycparser==2.22 ; platform_python_implementation == 'PyPy'
+pydantic==2.11.4
+pydantic-core==2.33.2
+pydub==0.25.1
+pygments==2.19.1 ; sys_platform != 'emscripten'
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.2
+pyyaml==6.0.2
+regex==2024.11.6
+requests==2.32.3
+rich==14.0.0 ; sys_platform != 'emscripten'
+ruff==0.11.8 ; sys_platform != 'emscripten'
+safehttpx==0.1.6
+safetensors==0.5.3
+semantic-version==2.10.0
+setuptools==80.4.0 ; (python_full_version >= '3.12' and platform_machine != 'x86_64') or (python_full_version >= '3.12' and sys_platform != 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')
+shellingham==1.5.4 ; sys_platform != 'emscripten'
+six==1.17.0
+sniffio==1.3.1
+starlette==0.46.2
+sympy==1.14.0
+tokenizers==0.21.1
+tomlkit==0.13.2
+torch==2.7.0
+tqdm==4.67.1
+transformers==4.51.3
+triton==3.3.0 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+typer==0.15.3 ; sys_platform != 'emscripten'
+typing-extensions==4.13.2
+typing-inspection==0.4.0
+tzdata==2025.2
+urllib3==2.4.0
+uvicorn==0.34.2 ; sys_platform != 'emscripten'
+websockets==15.0.1
+zstandard==0.23.0