Delete app.py

app.py

@@ -1,449 +0,0 @@
-"""
-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,
-    show_api=False,
-    theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
-    css=CUSTOM_CSS,
-)