Upload app.py

app.py

@@ -0,0 +1,451 @@
+"""
+Open3DForge — Image-to-game-ready 3D asset pipeline
+====================================================
+
+Milestone 1: Foundation
+-----------------------
+This is the scaffold. It validates that:
+  - The HF Space builds with the right Gradio + ZeroGPU configuration
+  - `@spaces.GPU` allocates and releases an H200 successfully
+  - `gr.Model3D` renders GLB files inline
+  - The 5-tab UI structure is sound
+  - Workspace folder management works
+  - Quota tracking persists across page reloads
+
+Subsequent milestones fill in the actual pipeline stages.
+
+Deployed exclusively on HF Spaces ZeroGPU — no local execution path.
+"""
+
+from __future__ import annotations
+
+import time
+from pathlib import Path
+
+import gradio as gr
+import spaces  # HF ZeroGPU
+
+from src import quota, ui_helpers, workspace
+
+# ---------------------------------------------------------------------------
+# ZeroGPU test function — proves that GPU allocation works.
+# Replaced in Milestone 2 with the actual TRELLIS.2 generator.
+# ---------------------------------------------------------------------------
+
+@spaces.GPU(duration=15)
+def zerogpu_smoke_test() -> str:
+    """Allocate a GPU, run a trivial torch op, release. Reports timing."""
+    start = time.time()
+    try:
+        import torch
+        if not torch.cuda.is_available():
+            return "❌ GPU not available inside @spaces.GPU. Something is wrong."
+
+        device = torch.device("cuda")
+        # Trivial GPU work
+        a = torch.randn(1024, 1024, device=device)
+        b = torch.randn(1024, 1024, device=device)
+        c = a @ b
+        torch.cuda.synchronize()
+        result_sum = float(c.sum().item())
+
+        gpu_name = torch.cuda.get_device_name(0)
+        vram_total = torch.cuda.get_device_properties(0).total_memory / 1e9
+        elapsed = time.time() - start
+
+        quota.record_usage("smoke_test", elapsed)
+
+        return (
+            f"✅ **GPU allocation successful**\n\n"
+            f"- Device: `{gpu_name}`\n"
+            f"- VRAM: `{vram_total:.1f} GB`\n"
+            f"- Test op (1024×1024 matmul): `{elapsed:.2f}s`\n"
+            f"- Result sum: `{result_sum:.2f}` (sanity check, non-zero = ✓)\n\n"
+            f"ZeroGPU integration is working. Ready for Milestone 2."
+        )
+    except Exception as e:
+        elapsed = time.time() - start
+        return f"❌ GPU test failed after {elapsed:.2f}s:\n```\n{type(e).__name__}: {e}\n```"
+
+
+# ---------------------------------------------------------------------------
+# Stage stubs — return placeholder messages until milestones implement them.
+# ---------------------------------------------------------------------------
+
+def _stub(stage: str) -> str:
+    return (
+        f"🚧 **{stage}** — not implemented yet.\n\n"
+        f"This stub will be replaced in a future milestone. "
+        f"See `PLAN.md` for the full pipeline spec."
+    )
+
+
+def stub_generate(*_args, **_kwargs):
+    return _stub("Stage 1: Generation")
+
+
+def stub_post_process(*_args, **_kwargs):
+    return _stub("Stage 2: Post-Processing")
+
+
+def stub_auto_rig(*_args, **_kwargs):
+    return _stub("Stage 3: Auto-Rigging")
+
+
+def stub_export(*_args, **_kwargs):
+    return _stub("Stage 4: Export"), None  # message, file
+
+
+# ---------------------------------------------------------------------------
+# Presets tab logic (this one is real even in Milestone 1)
+# ---------------------------------------------------------------------------
+
+def ui_refresh_presets() -> gr.Dropdown:
+    return gr.Dropdown(choices=workspace.list_presets(), label="Saved presets")
+
+
+def ui_save_preset(name: str) -> tuple[str, gr.Dropdown]:
+    if not name or not name.strip():
+        return "❌ Preset name cannot be empty.", ui_refresh_presets()
+    # Placeholder config — Milestone 9 wires in real settings from each tab
+    config = {
+        "name": name,
+        "version": 1,
+        "created_at": time.time(),
+        "note": "Placeholder — real preset data wired in Milestone 9",
+    }
+    workspace.save_preset(name, config)
+    return f"✅ Saved preset: `{name}`", ui_refresh_presets()
+
+
+def ui_delete_preset(name: str) -> tuple[str, gr.Dropdown]:
+    if not name:
+        return "❌ Select a preset to delete.", ui_refresh_presets()
+    if workspace.delete_preset(name):
+        return f"🗑️ Deleted preset: `{name}`", ui_refresh_presets()
+    return f"❌ Preset not found: `{name}`", ui_refresh_presets()
+
+
+# ---------------------------------------------------------------------------
+# Layout
+# ---------------------------------------------------------------------------
+
+CUSTOM_CSS = """
+.status-bar {
+    font-size: 0.85em;
+    color: #888;
+    padding: 8px 12px;
+    border-top: 1px solid #333;
+    margin-top: 12px;
+}
+.asset-summary {
+    font-size: 0.9em;
+    background: rgba(255,255,255,0.03);
+    padding: 12px;
+    border-radius: 6px;
+    border: 1px solid rgba(255,255,255,0.08);
+}
+"""
+
+
+def build_ui() -> gr.Blocks:
+    with gr.Blocks(
+        title="Open3DForge",
+    ) as demo:
+
+        # --- Header --------------------------------------------------------
+        gr.Markdown(
+            "# 🛠️ Open3DForge\n"
+            "*Personal image-to-game-ready 3D asset pipeline · UE5-first · "
+            "Built on HF ZeroGPU*"
+        )
+
+        # --- Tabs ----------------------------------------------------------
+        with gr.Tabs() as tabs:
+
+            # ============ Tab 1: Generate =================================
+            with gr.Tab("1. Generate", id=1):
+                gr.Markdown(
+                    "### Stage 1 — Image to 3D\n"
+                    "Upload 1–4 reference images. Multi-view dramatically "
+                    "improves quality for characters (front / 3-quarter / side / back)."
+                )
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        gen_images = gr.File(
+                            label="Reference images (1–4)",
+                            file_count="multiple",
+                            file_types=["image"],
+                        )
+                        gen_model = gr.Radio(
+                            choices=[
+                                "TRELLIS.2 (Hard Surface)",
+                                "Hunyuan3D-2 (Organic / Characters)",
+                            ],
+                            value="TRELLIS.2 (Hard Surface)",
+                            label="Generation model",
+                        )
+                        gen_quality = gr.Radio(
+                            choices=["Fast (~30s)", "Balanced (~60s)", "Hero (~90s)"],
+                            value="Balanced (~60s)",
+                            label="Quality preset",
+                        )
+                        with gr.Accordion("Advanced", open=False):
+                            gen_seed = gr.Number(value=42, label="Seed", precision=0)
+                            gen_steps = gr.Slider(
+                                20, 50, value=35, step=5,
+                                label="Inference steps",
+                            )
+                            gen_octree = gr.Dropdown(
+                                choices=[128, 192, 256], value=192,
+                                label="Octree resolution",
+                            )
+                            gen_tex_size = gr.Dropdown(
+                                choices=[1024, 2048, 4096], value=2048,
+                                label="Texture size",
+                            )
+                            gen_symmetry = gr.Radio(
+                                choices=["off", "bilateral", "radial"],
+                                value="off",
+                                label="Symmetry hint",
+                            )
+                        gen_btn = gr.Button("Generate", variant="primary")
+                    with gr.Column(scale=1):
+                        gen_status = gr.Markdown("*Awaiting input.*")
+                gen_btn.click(
+                    fn=stub_generate,
+                    inputs=[gen_images, gen_model, gen_quality, gen_seed,
+                            gen_steps, gen_octree, gen_tex_size, gen_symmetry],
+                    outputs=gen_status,
+                )
+
+            # ============ Tab 2: Post-Process =============================
+            with gr.Tab("2. Post-Process", id=2):
+                gr.Markdown(
+                    "### Stage 2 — Mesh cleanup, UV unwrap, texture baking\n"
+                    "Toggle steps on/off. Decimation has a live preview, the "
+                    "rest run on confirm."
+                )
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        pp_repair = gr.Checkbox(value=True, label="Mesh repair (pymeshfix)")
+                        pp_cleanup = gr.Checkbox(value=True, label="Geometry cleanup (PyMeshLab)")
+                        pp_decimate = gr.Checkbox(value=True, label="Decimation")
+                        pp_target_faces = gr.Slider(
+                            1000, 200000, value=25000, step=1000,
+                            label="Target faces",
+                        )
+                        pp_symmetry = gr.Checkbox(value=False, label="Enforce bilateral symmetry")
+                        pp_unwrap = gr.Checkbox(value=True, label="UV unwrap (xatlas)")
+                        pp_normal_bake = gr.Checkbox(value=True, label="Normal bake (nvdiffrast)")
+                        pp_normal_format = gr.Radio(
+                            choices=["DirectX (UE5)", "OpenGL (Unity/Godot)"],
+                            value="DirectX (UE5)",
+                            label="Normal format",
+                        )
+                        pp_albedo_bake = gr.Checkbox(value=True, label="Albedo bake")
+                        pp_material_bake = gr.Checkbox(value=True, label="Material bake (TRELLIS.2 attrs)")
+                        pp_ao = gr.Checkbox(value=True, label="AO bake")
+                        pp_ao_quality = gr.Radio(
+                            choices=["Fast", "Standard", "High"],
+                            value="Standard",
+                            label="AO quality",
+                        )
+                        pp_inpaint = gr.Checkbox(value=False, label="SDXL inpaint hidden UVs (~30s GPU)")
+                        pp_lods = gr.Checkbox(value=True, label="Generate LODs (LOD0/1/2)")
+                        pp_collision = gr.Checkbox(value=True, label="Collision mesh (CoACD)")
+                        pp_pivot = gr.Radio(
+                            choices=["bottom_center", "geometric_center", "custom"],
+                            value="bottom_center",
+                            label="Pivot point",
+                        )
+                        pp_scale_m = gr.Number(value=1.8, label="Real-world height (meters)")
+                        pp_btn = gr.Button("Run Post-Processing", variant="primary")
+                    with gr.Column(scale=1):
+                        pp_status = gr.Markdown("*No asset to process. Generate one first.*")
+                pp_btn.click(
+                    fn=stub_post_process,
+                    inputs=[pp_repair, pp_cleanup, pp_decimate, pp_target_faces,
+                            pp_symmetry, pp_unwrap, pp_normal_bake, pp_normal_format,
+                            pp_albedo_bake, pp_material_bake, pp_ao, pp_ao_quality,
+                            pp_inpaint, pp_lods, pp_collision, pp_pivot, pp_scale_m],
+                    outputs=pp_status,
+                )
+
+            # ============ Tab 3: Auto-Rig =================================
+            with gr.Tab("3. Auto-Rig", id=3):
+                gr.Markdown(
+                    "### Stage 3 — Auto-rigging (optional)\n"
+                    "Uses UniRig (VAST-AI). For characters and creatures. "
+                    "After rigging, drop the FBX into [Mixamo](https://mixamo.com) "
+                    "for free animation presets."
+                )
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        rig_type = gr.Dropdown(
+                            choices=["Humanoid", "Quadruped", "Bird", "Insect", "Custom"],
+                            value="Humanoid",
+                            label="Character type",
+                        )
+                        rig_seed = gr.Number(value=0, label="Skeleton seed", precision=0)
+                        rig_spring = gr.Checkbox(value=False, label="Spring bones (hair/cloth/tail)")
+                        rig_format = gr.Radio(
+                            choices=["FBX (UE5 recommended)", "GLB"],
+                            value="FBX (UE5 recommended)",
+                            label="Export format",
+                        )
+                        rig_btn = gr.Button("Auto-Rig", variant="primary")
+                    with gr.Column(scale=1):
+                        rig_status = gr.Markdown("*Process an asset in Stage 2 first.*")
+                rig_btn.click(
+                    fn=stub_auto_rig,
+                    inputs=[rig_type, rig_seed, rig_spring, rig_format],
+                    outputs=rig_status,
+                )
+
+            # ============ Tab 4: Export ===================================
+            with gr.Tab("4. Export", id=4):
+                gr.Markdown(
+                    "### Stage 4 — Engine-ready export\n"
+                    "UE5 default: FBX with DirectX normals + ORM-packed textures."
+                )
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        ex_engine = gr.Dropdown(
+                            choices=["UE5", "Unity (HDRP)", "Godot 4", "Blender", "Web (Three.js)"],
+                            value="UE5",
+                            label="Target engine",
+                        )
+                        ex_name = gr.Textbox(value="Asset_01", label="Asset name")
+                        ex_type = gr.Radio(
+                            choices=["Character (SK_)", "Prop (SM_)", "Environment (SM_)"],
+                            value="Prop (SM_)",
+                            label="Asset type",
+                        )
+                        ex_include_lods = gr.Checkbox(value=True, label="Include LODs")
+                        ex_include_collision = gr.Checkbox(value=True, label="Include collision mesh")
+                        ex_btn = gr.Button("Export", variant="primary")
+                    with gr.Column(scale=1):
+                        ex_status = gr.Markdown("*Nothing to export yet.*")
+                        ex_file = gr.File(label="Download", visible=True)
+                ex_btn.click(
+                    fn=stub_export,
+                    inputs=[ex_engine, ex_name, ex_type, ex_include_lods, ex_include_collision],
+                    outputs=[ex_status, ex_file],
+                )
+
+            # ============ Tab 5: Presets ==================================
+            with gr.Tab("5. Presets", id=5):
+                gr.Markdown(
+                    "### Saved configurations\n"
+                    "Save parameter sets for asset types you create repeatedly "
+                    "(e.g., \"character_UE5_hero\", \"prop_UE5_standard\"). "
+                    "Wired up properly in Milestone 9."
+                )
+                with gr.Row():
+                    with gr.Column():
+                        pr_list = gr.Dropdown(
+                            choices=workspace.list_presets(),
+                            label="Saved presets",
+                        )
+                        pr_refresh = gr.Button("Refresh list", size="sm")
+                        with gr.Row():
+                            pr_name = gr.Textbox(label="New preset name", scale=2)
+                            pr_save = gr.Button("Save current settings", variant="primary", scale=1)
+                        pr_delete = gr.Button("Delete selected", variant="stop")
+                        pr_status = gr.Markdown()
+                pr_refresh.click(fn=ui_refresh_presets, outputs=pr_list)
+                pr_save.click(fn=ui_save_preset, inputs=pr_name, outputs=[pr_status, pr_list])
+                pr_delete.click(fn=ui_delete_preset, inputs=pr_list, outputs=[pr_status, pr_list])
+
+            # ============ Tab 6: Diagnostics (hidden in prod, useful now) =
+            with gr.Tab("Diagnostics", id=99):
+                gr.Markdown(
+                    "### Milestone 1 — Foundation check\n"
+                    "Verify the Space environment is working correctly before "
+                    "building out the pipeline."
+                )
+                with gr.Row():
+                    with gr.Column():
+                        diag_btn = gr.Button("🧪 Run GPU smoke test", variant="primary")
+                        diag_out = gr.Markdown()
+                    with gr.Column():
+                        gr.Markdown("**Workspace state:**")
+                        diag_state = gr.JSON(value=workspace.get_state().to_dict())
+                        diag_refresh = gr.Button("Refresh state", size="sm")
+                diag_btn.click(fn=zerogpu_smoke_test, outputs=diag_out)
+                diag_refresh.click(
+                    fn=lambda: workspace.get_state().to_dict(),
+                    outputs=diag_state,
+                )
+
+        # --- Persistent right-side viewer + asset summary ------------------
+        gr.Markdown("---")
+        with gr.Row():
+            with gr.Column(scale=2):
+                viewer = gr.Model3D(
+                    label="3D viewer",
+                    value=ui_helpers.get_viewer_model_path(),
+                    clear_color=[0.1, 0.1, 0.12, 1.0],
+                    height=500,
+                )
+            with gr.Column(scale=1):
+                summary = gr.Markdown(
+                    value=ui_helpers.get_asset_summary(),
+                    elem_classes=["asset-summary"],
+                )
+                refresh_summary = gr.Button("🔄 Refresh viewer", size="sm")
+                refresh_summary.click(
+                    fn=lambda: (
+                        ui_helpers.get_viewer_model_path(),
+                        ui_helpers.get_asset_summary(),
+                    ),
+                    outputs=[viewer, summary],
+                )
+
+        # --- Global status bar --------------------------------------------
+        status_bar = gr.Markdown(
+            value=ui_helpers.get_status_bar(),
+            elem_classes=["status-bar"],
+        )
+
+        # --- Global refresh: every pipeline action updates the viewer, summary,
+        # and status bar. We chain .then() onto each button so the refresh runs
+        # only after the action completes (and even if it fails — Gradio still
+        # fires .then() after exceptions inside the main handler).
+        def _global_refresh():
+            return (
+                ui_helpers.get_viewer_model_path(),
+                ui_helpers.get_asset_summary(),
+                ui_helpers.get_status_bar(),
+            )
+
+        for trigger_btn in (gen_btn, pp_btn, rig_btn, ex_btn,
+                            diag_btn, diag_refresh, refresh_summary):
+            trigger_btn.click(
+                fn=_global_refresh,
+                outputs=[viewer, summary, status_bar],
+            )
+
+    return demo
+
+
+# ---------------------------------------------------------------------------
+# Entrypoint
+# ---------------------------------------------------------------------------
+# On HF Spaces, app.py is executed directly. We construct the demo and call
+# .launch() at module level. The HF Space runtime handles all networking;
+# we just need to bind to 0.0.0.0:7860.
+
+workspace.ensure_dirs()
+demo = build_ui()
+demo.queue(default_concurrency_limit=1).launch(
+    server_name="0.0.0.0",
+    server_port=7860,
+    show_error=True,
+    # Gradio 6: show_api removed. Use footer_links instead.
+    # Equivalent of old show_api=False:
+    footer_links=["gradio", "settings"],
+    theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
+    css=CUSTOM_CSS,
+)