Update app.py

app.py

@@ -1,7 +1,5 @@
 """NEXUS Visual Weaver - Build Small Hackathon command center."""
-
 from __future__ import annotations
-
 import os
 import sys
 import hashlib
@@ -9,7 +7,6 @@ import secrets
 from pathlib import Path
 from typing import Any
 from urllib.parse import urlparse
-
 import gradio as gr
 
 ROOT = Path(__file__).resolve().parent
@@ -18,8 +15,8 @@ if str(SRC) not in sys.path:
     sys.path.insert(0, str(SRC))
 
 try:
-    import spaces  # type: ignore  # noqa: F401
-except Exception:  # pragma: no cover - local development does not require Spaces.
+    import spaces
+except Exception:
     spaces = None
 
 from nexus_visual_weaver.catalog import catalog_summary
@@ -28,12 +25,10 @@ from nexus_visual_weaver.hf_runtime import generate_flux_image
 from nexus_visual_weaver.model_relay import WeaverModelRelay
 from nexus_visual_weaver.planner import build_command_center_run
 from nexus_visual_weaver.provider_runtime import judge_with_minicpm, judge_with_nemotron
-from nexus_visual_weaver.render import render_catalog_table, render_command_header, render_dashboard_regions
+from nexus_visual_weaver.render import render_catalog_table, render_dashboard_regions
 from nexus_visual_weaver.security import scan_file
 from nexus_visual_weaver.styles import APP_CSS
 
-# Use default theme to avoid Gradio 6.18 Font comparison bug
-# Custom theme with any parameter triggers: AttributeError: 'str' object has no attribute 'name'
 APP_THEME = gr.themes.Soft()
 
 DEFAULT_PROMPT = (
@@ -55,7 +50,6 @@ ASPECT_DIMENSIONS = {
     "Portrait": (832, 1216),
 }
 
-
 def _default_operator_state() -> dict[str, Any]:
     return {
         "provider_state": "idle",
@@ -64,18 +58,15 @@ def _default_operator_state() -> dict[str, Any]:
         "message": "No operator action yet.",
     }
 
-
 def _zero_gpu_entrypoint(fn: Any) -> Any:
     gpu_decorator = getattr(spaces, "GPU", None) if spaces is not None else None
     if gpu_decorator is None:
         return fn
     return gpu_decorator(duration=300)(fn)
 
-
 def _relay_snapshot(adult_mode: bool = False) -> dict[str, Any]:
     return MODEL_RELAY.dashboard_snapshot(public_demo=not adult_mode)
 
-
 def _file_path(uploaded: Any) -> str | None:
     if uploaded is None:
         return None
@@ -84,7 +75,6 @@ def _file_path(uploaded: Any) -> str | None:
     path = getattr(uploaded, "name", None)
     return str(path) if path else None
 
-
 def _safe_file_hash(path: str | None) -> tuple[str | None, int | None]:
     if not path:
         return None, None
@@ -100,7 +90,6 @@ def _safe_file_hash(path: str | None) -> tuple[str | None, int | None]:
         return None, None
     return sha256.hexdigest(), size
 
-
 def _safe_reference_url_metadata(reference_url: str | None) -> dict[str, Any] | None:
     if not reference_url:
         return None
@@ -116,30 +105,26 @@ def _safe_reference_url_metadata(reference_url: str | None) -> dict[str, Any] |
         "message": "URL stored as metadata only; Space runtime does not crawl or copy shop images.",
     }
 
-
 def _reference_metadata(uploaded: Any, reference_url: str | None, scan: dict[str, Any]) -> list[dict[str, Any]]:
     records: list[dict[str, Any]] = []
     path = _file_path(uploaded)
     if path:
         file_hash, size = _safe_file_hash(path)
-        records.append(
-            {
-                "source": "upload",
-                "basename": Path(path).name,
-                "sha256": file_hash,
-                "size_bytes": size,
-                "st3gg_status": scan.get("status"),
-                "export_gate": scan.get("export_gate"),
-                "magic": scan.get("magic"),
-                "extension": scan.get("extension"),
-            }
-        )
+        records.append({
+            "source": "upload",
+            "basename": Path(path).name,
+            "sha256": file_hash,
+            "size_bytes": size,
+            "st3gg_status": scan.get("status"),
+            "export_gate": scan.get("export_gate"),
+            "magic": scan.get("magic"),
+            "extension": scan.get("extension"),
+        })
     url_record = _safe_reference_url_metadata(reference_url)
     if url_record:
         records.append(url_record)
     return records
 
-
 def _creator_controls(
     reasoning_mode: str,
     video_preset: str,
@@ -178,7 +163,6 @@ def _creator_controls(
         },
     }
 
-
 def _resolve_seed(seed_value: Any) -> int:
     try:
         if seed_value is None or str(seed_value).strip() == "":
@@ -188,15 +172,12 @@ def _resolve_seed(seed_value: Any) -> int:
         return secrets.randbelow(1_000_000_000)
     return secrets.randbelow(1_000_000_000) if seed < 0 else seed
 
-
 def _generation_dimensions(aspect: str | None) -> tuple[int, int]:
     return ASPECT_DIMENSIONS.get(str(aspect or "Portrait"), ASPECT_DIMENSIONS["Portrait"])
 
-
 def _style_modifier(style_strength: str | None) -> str:
     return STYLE_MODIFIERS.get(str(style_strength or "High Fashion"), STYLE_MODIFIERS["High Fashion"])
 
-
 def _prompt_with_controls(prompt: str, controls: dict[str, Any]) -> str:
     wardrobe = controls.get("wardrobe", {})
     additions = [
@@ -215,13 +196,11 @@ def _prompt_with_controls(prompt: str, controls: dict[str, Any]) -> str:
     prompt = f"{prompt}\nWardrobe controls: {suffix}" if suffix else prompt
     return f"{prompt}\nStyle direction: {style}"
 
-
 def _generated_output_path(operator_state: dict[str, Any] | None) -> str | None:
     generation = (operator_state or {}).get("generation") or {}
     output_path = generation.get("output_path")
     return str(output_path) if output_path else None
 
-
 def _authoritative_generated_scan(operator_state: dict[str, Any] | None) -> dict[str, Any]:
     output_path = _generated_output_path(operator_state)
     if output_path:
@@ -229,7 +208,6 @@ def _authoritative_generated_scan(operator_state: dict[str, Any] | None) -> dict
     stored_scan = (operator_state or {}).get("generated_scan")
     return stored_scan if isinstance(stored_scan, dict) else scan_file(None)
 
-
 def _checkpoint_seed(checkpoint_id: str) -> int:
     suffix = "".join(char for char in checkpoint_id[-8:] if char in "0123456789abcdefABCDEF")
     if not suffix:
@@ -239,7 +217,6 @@ def _checkpoint_seed(checkpoint_id: str) -> int:
     except ValueError:
         return 0
 
-
 def _wardrobe_summary(run: Any) -> str:
     slots = getattr(getattr(run, "outfit", None), "slots", []) or []
     return "; ".join(
@@ -247,10 +224,8 @@ def _wardrobe_summary(run: Any) -> str:
         for slot in slots
     )
 
-
 SECTIONS = ["Forge", "Wardrobe", "Lore", "Models", "Security", "Runs"]
 
-
 def _button_updates(run: Any | None, operator_state: dict[str, Any] | None) -> tuple[Any, Any, Any]:
     state = operator_state or {}
     generated = bool(_generated_output_path(state)) and (state.get("generation") or {}).get("status") == "success"
@@ -262,7 +237,6 @@ def _button_updates(run: Any | None, operator_state: dict[str, Any] | None) -> t
         gr.update(interactive=False),
     )
 
-
 def _dashboard_regions(
     run: Any | None = None,
     adult_mode: bool = False,
@@ -279,7 +253,6 @@ def _dashboard_regions(
         operator_state=operator_state,
     )
 
-
 # ─── Modal Integration ───
 MODAL_AVAILABLE = False
 try:
@@ -288,7 +261,6 @@ try:
 except ImportError:
     pass
 
-# LoRA Registry (mirrors modal_nexus_refine_v2.py)
 LORA_ADAPTERS = {
     "garment": {"repo": "NO8D/BodyControl", "desc": "Body/garment shape control", "weight": 0.75},
     "hardware": {"repo": "NO8D/ExpressionControl", "desc": "Expression/hardware detail", "weight": 0.70},
@@ -307,15 +279,12 @@ GPU_OPTIONS = {
 
 MODAL_COST_TRACKER = {"credits_remaining": 250.88, "total_spent": 0.0, "refinements": 0}
 
-
 def _modal_refine_image(image_bytes: bytes, user_addition: str, gpu_type: str = "A100-80GB",
                          strength: float = 0.58, steps: int = 32, guidance_scale: float = 3.8,
                          seed: int = -1, lora_adapters: list | None = None,
                          negative_prompt: str = "blurry, low quality, deformed, extra limbs") -> tuple:
-    """Real Modal refinement call — wires to nexus-couture-refine-v2 on Modal."""
     if not MODAL_AVAILABLE:
-        return None, "❌ Modal not installed. Add 'modal' to requirements.txt"
-
+        return None, "❌ Modal not installed"
     try:
         fn = modal.Function.lookup("nexus-couture-refine-v2", "refine_couture")
         result_bytes = fn.remote(
@@ -329,28 +298,24 @@ def _modal_refine_image(image_bytes: bytes, user_addition: str, gpu_type: str =
             negative_prompt=negative_prompt,
             gpu_type=gpu_type,
         )
-        # Update cost tracker
         gpu_info = GPU_OPTIONS.get(gpu_type, GPU_OPTIONS["A100-80GB"])
         est_cost = round(gpu_info["price"] * (steps / 60), 4)
         MODAL_COST_TRACKER["total_spent"] += est_cost
         MODAL_COST_TRACKER["credits_remaining"] -= est_cost
         MODAL_COST_TRACKER["refinements"] += 1
-        return result_bytes, f"✅ Modal refinement complete on {gpu_type} (est. ${est_cost:.4f})"
+        return result_bytes, f"✅ Modal refinement complete on {gpu_type}"
     except Exception as e:
         return None, f"❌ Modal error: {str(e)[:200]}"
 
-
 def _modal_health_check() -> dict:
-    """Check Modal connectivity and GPU availability."""
     if not MODAL_AVAILABLE:
-        return {"status": "unavailable", "message": "Modal package not installed"}
+        return {"status": "unavailable", "message": "Modal not installed"}
     try:
         fn = modal.Function.lookup("nexus-couture-refine-v2", "check_modal_health")
         return fn.remote()
     except Exception as e:
         return {"status": "error", "message": str(e)[:200]}
 
-
 @_zero_gpu_entrypoint
 def run_weave(
     prompt, reasoning_mode, video_preset, adult_mode, upload, active_section,
@@ -390,6 +355,7 @@ def run_weave(
         provider_state = generation.provider_state
     else:
         provider_state = "checkpointed"
+
     operator_state = {
         "provider_state": provider_state,
         "checkpoint": "pending_review",
@@ -417,7 +383,6 @@ def run_weave(
         *_button_updates(run, operator_state),
     )
 
-
 def toggle_adult_visibility(adult_mode, active_section, upload):
     scan = scan_file(_file_path(upload))
     operator_state = {
@@ -431,7 +396,6 @@ def toggle_adult_visibility(adult_mode, active_section, upload):
         render_catalog_table(adult_mode=adult_mode), catalog_summary(adult_mode), scan, operator_state,
     )
 
-
 def refresh_section(active_section, adult_mode, run, scan, operator_state):
     scan = scan or scan_file(None)
     regions = _dashboard_regions(
@@ -440,7 +404,6 @@ def refresh_section(active_section, adult_mode, run, scan, operator_state):
     )
     return regions["command_rail"], regions["operations"], regions["inspector"], regions["artifacts"], regions["providers"], scan
 
-
 def _render_stateful(run, adult_mode, scan, active_section, operator_state):
     scan = scan or scan_file(None)
     regions = _dashboard_regions(
@@ -457,7 +420,6 @@ def _render_stateful(run, adult_mode, scan, active_section, operator_state):
         *_button_updates(run, operator_state),
     )
 
-
 def scan_reference(run, adult_mode, upload, active_section, operator_state, reference_url=None):
     state = operator_state or _default_operator_state()
     reference_path = _file_path(upload)
@@ -486,7 +448,6 @@ def scan_reference(run, adult_mode, upload, active_section, operator_state, refe
     rendered = _render_stateful(run, adult_mode, generated_scan, active_section, next_state)
     return (*rendered, generated_scan)
 
-
 def approve_checkpoint(run, adult_mode, scan, active_section, operator_state):
     state = operator_state or _default_operator_state()
     scan = _authoritative_generated_scan(state)
@@ -508,7 +469,6 @@ def approve_checkpoint(run, adult_mode, scan, active_section, operator_state):
         }
     return _render_stateful(run, adult_mode, scan, active_section, next_state)
 
-
 def export_packet(run, adult_mode, scan, active_section, operator_state, override_reason=None):
     state = operator_state or _default_operator_state()
     scan = _authoritative_generated_scan(state)
@@ -533,7 +493,6 @@ def export_packet(run, adult_mode, scan, active_section, operator_state, overrid
         }
     return _render_stateful(run, adult_mode, scan, active_section, next_state)
 
-
 def stop_provider_job(run, adult_mode, scan, active_section, operator_state):
     scan = scan or scan_file(None)
     next_state = {
@@ -543,7 +502,6 @@ def stop_provider_job(run, adult_mode, scan, active_section, operator_state):
     }
     return _render_stateful(run, adult_mode, scan, active_section, next_state)
 
-
 def reset_demo(adult_mode, active_section):
     scan = scan_file(None)
     operator_state = _default_operator_state()
@@ -557,11 +515,8 @@ def reset_demo(adult_mode, active_section):
         gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False),
     )
 
-
 # ─── Modal Tab Handlers ───
-
 def modal_refine_handler(input_image, user_addition, gpu_type, strength, steps, guidance, seed, lora_choices, negative_prompt):
-    """Handle Modal refinement from the Space UI."""
     if input_image is None:
         return None, "❌ No input image provided"
     from PIL import Image as PILImage
@@ -573,7 +528,6 @@ def modal_refine_handler(input_image, user_addition, gpu_type, strength, steps,
         img = PILImage.open(input_image)
     img.save(buf, format="PNG")
     image_bytes = buf.getvalue()
-
     result_bytes, message = _modal_refine_image(
         image_bytes=image_bytes, user_addition=user_addition,
         gpu_type=gpu_type, strength=strength, steps=int(steps),
@@ -587,18 +541,15 @@ def modal_refine_handler(input_image, user_addition, gpu_type, strength, steps,
         return result_img, f"{message}\n{cost_info}"
     return None, message
 
-
 def modal_health_handler():
-    """Check Modal connectivity."""
     result = _modal_health_check()
     if result.get("status") == "healthy":
-        return f"✅ Modal connected\nGPU: {result.get('gpu', 'N/A')}\nVRAM: {result.get('gpu_memory_gb', 0)}GB\nLoRAs: {', '.join(result.get('lora_registry', []))}"
+        return f"✅ Modal connected\nGPU: {result.get('gpu', 'N/A')}"
     elif result.get("status") == "unavailable":
         return f"⚠️ {result.get('message', 'Modal not available')}"
     else:
         return f"❌ Modal error: {result.get('message', 'Unknown')}"
 
-
 initial_operator_state = _default_operator_state()
 initial_regions = _dashboard_regions(scan=scan_file(None), operator_state=initial_operator_state)
 
@@ -609,7 +560,7 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
     topbar_html = gr.HTML(initial_regions["topbar"], container=False, visible=False)
 
     with gr.Tabs():
-        # ═══ Tab 1: Studio (existing creator workbench) ═══
+        # ═══ Tab 1: Studio ═══
         with gr.Tab("🧵 Studio"):
             with gr.Row(elem_id="nw-creator-workbench", elem_classes=["nw-creator-workbench"]):
                 with gr.Column(scale=5, min_width=520, elem_id="nw-creator-panel"):
@@ -644,7 +595,6 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
                         with gr.Row():
                             scan_btn = gr.Button("Scan Uploaded File", scale=1)
                             stop_btn = gr.Button("Stop Job", variant="stop", interactive=False, scale=1)
-
                 with gr.Column(scale=4, min_width=460, elem_id="nw-output-panel"):
                     gr.Markdown("### Output")
                     artifact_html = gr.HTML(initial_regions["artifacts"], container=False)
@@ -658,7 +608,7 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
                     )
                     gr.Markdown("Generation is not export. Every artifact stays behind ST3GG review and human checkpoint.")
 
-        # ═══ Tab 2: Modal Refinement ═══
+        # ═══ Tab 2: Modal Refinement (ACTIVE) ═══
         with gr.Tab("⚡ Modal"):
             gr.Markdown("## ⚡ Modal GPU Refinement")
             gr.Markdown("Send a generated image to Modal for FLUX.1-Kontext-dev refinement with multi-LoRA on dedicated GPU.")
@@ -694,7 +644,7 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
             )
             modal_health_btn.click(fn=modal_health_handler, inputs=[], outputs=[modal_status])
 
-        # ═══ Tab 3: LoRA Lab ═══
+        # ═══ Tab 3: LoRA Lab (ACTIVE) ═══
         with gr.Tab("🧪 LoRA Lab"):
             gr.Markdown("## 🧪 LoRA Training Lab")
             gr.Markdown("Train custom LoRA adapters on Modal GPU. Connect a dataset repo and configure training parameters.")
@@ -719,7 +669,6 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
                         for k, v in LORA_ADAPTERS.items()
                     )
                     gr.Markdown(lora_catalog_md)
-
             def lora_train_handler(dataset_repo, output_name, rank, lr, steps, batch, push, hub_repo):
                 if not MODAL_AVAILABLE:
                     return "❌ Modal not installed"
@@ -733,7 +682,6 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
                     return f"✅ Training triggered on Modal!\nDataset: {dataset_repo}\nOutput: {output_name}\nRank: {rank}, Steps: {steps}, LR: {lr}"
                 except Exception as e:
                     return f"❌ Training error: {str(e)[:300]}"
-
             lora_train_btn.click(
                 fn=lora_train_handler,
                 inputs=[lora_dataset_repo, lora_output_name, lora_rank, lora_lr,
@@ -781,6 +729,7 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
         outputs=stateful_outputs, api_name="run_active_weave",
         concurrency_limit=1, concurrency_id="flux-gpu",
     )
+
     run_submit = prompt.submit(
         fn=run_weave,
         inputs=[prompt, reasoning_mode, video_preset, adult_mode, upload, section_nav,
@@ -789,6 +738,7 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
         outputs=stateful_outputs, api_name=False,
         concurrency_limit=1, concurrency_id="flux-gpu",
     )
+
     adult_mode.change(
         fn=toggle_adult_visibility,
         inputs=[adult_mode, section_nav, upload],
@@ -796,47 +746,53 @@ with gr.Blocks(title="NEXUS Visual Weaver") as demo:
                  artifact_html, provider_html, catalog_html, catalog_json, scan_json, operator_state],
         api_name="toggle_adult_catalog", queue=False,
     )
+
     section_nav.change(
         fn=refresh_section,
         inputs=[section_nav, adult_mode, active_run_state, scan_state, operator_state],
         outputs=[command_rail_html, operations_html, inspector_html, artifact_html, provider_html, scan_json],
         api_name=False, queue=False,
     )
+
     scan_btn.click(
         fn=scan_reference,
         inputs=[active_run_state, adult_mode, upload, section_nav, operator_state, reference_url],
         outputs=dashboard_outputs + [operator_state, checkpoint_btn, export_btn, stop_btn, scan_state],
         api_name="scan_reference", queue=False,
     )
+
     checkpoint_btn.click(
         fn=approve_checkpoint,
         inputs=[active_run_state, adult_mode, scan_state, section_nav, operator_state],
         outputs=operator_outputs, api_name="approve_checkpoint", queue=False,
     )
+
     export_btn.click(
         fn=export_packet,
         inputs=[active_run_state, adult_mode, scan_state, section_nav, operator_state, override_reason],
         outputs=operator_outputs, api_name="prepare_export_packet", queue=False,
     )
+
     stop_btn.click(
         fn=stop_provider_job,
         inputs=[active_run_state, adult_mode, scan_state, section_nav, operator_state],
         outputs=operator_outputs, api_name="stop_provider_job", queue=False,
         cancels=[run_click, run_submit],
     )
+
     reset_btn.click(
         fn=reset_demo,
         inputs=[adult_mode, section_nav],
         outputs=stateful_outputs, api_name="reset_demo_state", queue=False,
         cancels=[run_click, run_submit],
     )
+
     demo.load(
         fn=lambda: (render_catalog_table(False), catalog_summary(False), scan_file(None), scan_file(None), _default_operator_state()),
         outputs=[catalog_html, catalog_json, scan_json, scan_state, operator_state],
         api_name=False,
     )
 
-
 if __name__ == "__main__":
     if hasattr(sys.stdout, "reconfigure"):
         sys.stdout.reconfigure(encoding="utf-8", errors="replace")
@@ -845,6 +801,8 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=int(os.environ.get("NEXUS_PORT", os.environ.get("PORT", "7860"))),
-        quiet=True, mcp_server=True, ssr_mode=False,
+        quiet=True,
+        mcp_server=True,
+        ssr_mode=False,
         css=APP_CSS,
-    )
+    )