Spaces:
Sleeping
Sleeping
| """ | |
| UI helpers — shared functions across Gradio tabs. | |
| Keeps `app.py` focused on layout, not formatting logic. | |
| """ | |
| from __future__ import annotations | |
| from . import quota, workspace | |
| def get_status_bar() -> str: | |
| """Build the global status bar shown at the bottom of every tab. | |
| Markdown-formatted single line with quota + workspace info. | |
| """ | |
| state = workspace.get_state() | |
| parts = [ | |
| quota.format_status(), | |
| f"📁 Workspace: {workspace.current_size_mb():.1f} MB", | |
| f"📦 Exports: {workspace.export_count()}", | |
| ] | |
| if state.model_used: | |
| parts.append(f"🔧 Last: {state.model_used}") | |
| return " · ".join(parts) | |
| def get_asset_summary() -> str: | |
| """Multi-line summary of what's currently in `workspace/current/`. | |
| Used in the viewer panel to show pipeline progress. | |
| """ | |
| state = workspace.get_state() | |
| lines = [f"**Asset:** `{state.asset_name}`"] | |
| if state.face_count: | |
| lines.append(f"- Faces: {state.face_count:,}") | |
| lines.append(f"- Vertices: {state.vertex_count:,}") | |
| stages = [] | |
| if state.high_poly_glb: | |
| stages.append("✓ Generated") | |
| if state.cleaned_glb: | |
| stages.append("✓ Cleaned") | |
| if state.low_poly_glb: | |
| stages.append("✓ Decimated") | |
| if state.unwrapped_glb: | |
| stages.append("✓ UV unwrapped") | |
| if state.normal_dx_png or state.normal_gl_png: | |
| stages.append("✓ Normal baked") | |
| if state.albedo_png: | |
| stages.append("✓ Albedo baked") | |
| if state.orm_png or (state.roughness_png and state.metallic_png): | |
| stages.append("✓ PBR maps") | |
| if state.lod_glbs: | |
| stages.append(f"✓ LODs ({len(state.lod_glbs)})") | |
| if state.collision_glb: | |
| stages.append("✓ Collision") | |
| if state.rigged_glb or state.rigged_fbx: | |
| stages.append("✓ Rigged") | |
| if stages: | |
| lines.append("") | |
| lines.append("**Progress:**") | |
| for s in stages: | |
| lines.append(f"- {s}") | |
| else: | |
| lines.append("") | |
| lines.append("*No asset loaded yet. Start at the Generate tab.*") | |
| return "\n".join(lines) | |
| def get_viewer_model_path() -> str | None: | |
| """Pick the best GLB to show in the 3D viewer. | |
| Reads the filesystem directly so it works across the ZeroGPU subprocess | |
| boundary (the in-memory state written inside @spaces.GPU is invisible to | |
| the parent Gradio process). | |
| """ | |
| from .workspace import CURRENT | |
| # Order: most processed → least processed | |
| candidates = [ | |
| CURRENT / "rigged.glb", | |
| CURRENT / "scaled.glb", | |
| CURRENT / "pivoted.glb", | |
| CURRENT / "lods" / "LOD0.glb", | |
| CURRENT / "unwrapped.glb", | |
| CURRENT / "low_poly.glb", | |
| CURRENT / "cleaned.glb", | |
| CURRENT / "repaired.glb", | |
| CURRENT / "raw_gen.glb", | |
| CURRENT / "high_poly.glb", | |
| ] | |
| for path in candidates: | |
| if path.exists(): | |
| return str(path) | |
| return None | |
| def quota_warning(operation: str) -> str: | |
| """Generate a warning if an operation would exceed the daily quota. | |
| Returns an empty string if the operation fits comfortably. | |
| """ | |
| estimated = quota.estimate(operation) | |
| state = quota.get_state() | |
| remaining = state.remaining_seconds() | |
| if estimated > remaining: | |
| overage = estimated - remaining | |
| cost = overage * quota.OVERAGE_RATE_PER_SECOND | |
| return ( | |
| f"⚠️ **Quota warning:** This will use ~{estimated}s but you only " | |
| f"have {remaining:.0f}s left today. " | |
| f"Overage cost: ~${cost:.2f}" | |
| ) | |
| elif estimated > remaining * 0.5: | |
| return ( | |
| f"ℹ️ Estimated GPU time: ~{estimated}s " | |
| f"({remaining:.0f}s remaining today)" | |
| ) | |
| return "" | |