Update app.py

app.py

@@ -1,103 +1,186 @@
-import spaces  # MUST be imported before any CUDA-touching package (torch/diffusers)
+"""
+integrated_nexus_app.py
+NEXUS Visual Weaver - Integrated App (Phase A + HF MCP Bridge)
 
-import gradio as gr
-import numpy as np
-import random
-import torch
-from diffusers import Flux2KleinPipeline
-
-# ---------------------------------------------------------------------------
-# Model: FLUX.2 [klein] 4B
-#   - Apache-2.0, 4B params, BFL's fastest small model (sub-second, ~13GB VRAM)
-#   - Unified text-to-image + multi-reference editing
-#   - Released Jan 2026 (current BFL small-model generation)
-# ---------------------------------------------------------------------------
-MODEL_REPO_ID = "black-forest-labs/FLUX.2-klein-4B"
-
-dtype = torch.bfloat16
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-# Load on cuda at module level. (No enable_model_cpu_offload() on ZeroGPU —
-# the GPU is only attached inside @spaces.GPU; module-level cuda uses the
-# ZeroGPU CUDA-emulation, and offload would conflict.)
-pipe = Flux2KleinPipeline.from_pretrained(MODEL_REPO_ID, torch_dtype=dtype).to(device)
-
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 2048
-
-
-@spaces.GPU(duration=60)
-def infer(
-    prompt,
-    seed,
-    randomize_seed,
-    width,
-    height,
-    num_inference_steps,
-    guidance_scale,
-    progress=gr.Progress(track_tqdm=True),
-):
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator(device=device).manual_seed(seed)
-    image = pipe(
-        prompt=prompt,
-        width=width,
-        height=height,
-        num_inference_steps=num_inference_steps,
-        guidance_scale=guidance_scale,
-        generator=generator,
-    ).images[0]
-    return image, seed
-
-
-examples = [
-    "A magical city at twilight, glowing windows, storybook illustration, warm light",
-    "A cat holding a sign that says hello world",
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
-]
-
-css = """
-#col-container { margin: 0 auto; max-width: 640px; }
+This is the main Gradio application for the hackathon Space.
+It combines:
+- Phase A: Orchestration + Governance Checkpoint UI
+- HF MCP Bridge: Authentication + Repository tools exposed as MCP tools
+
+Launch with mcp_server=True for agent integration.
 """
 
-with gr.Blocks(css=css) as demo:
-    with gr.Column(elem_id="col-container"):
-        gr.Markdown("# 🖼️ NEXUS Visual Weaver — FLUX.2 [klein] 4B")
-        with gr.Row():
-            prompt = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=1,
-                placeholder="Enter your prompt",
-                container=False,
-            )
-            run_button = gr.Button("Run", scale=0, variant="primary")
+import gradio as gr
+from nexus_hf_mcp_bridge import nexus_hf_bridge
+
+
+# ============================================================
+# Phase A - Original Orchestration Logic (Stub for now)
+# ============================================================
+
+def run_refine_and_plan(prompt: str, mode: str):
+    """Phase A orchestration stub."""
+    return {
+        "input_prompt": prompt,
+        "reasoning_mode": mode,
+        "selected_backbone": "flux_klein_9b" if "image" in prompt.lower() else "ltx_2.3",
+        "plan_summary": f"Plan generated for: {prompt[:80]}...",
+        "next_step": "Run judgment checkpoint or use HF Bridge"
+    }
+
+
+def run_judgment_checkpoint(plan: dict):
+    """Phase A governance checkpoint stub."""
+    return {
+        "checkpoint_id": "phase_a_checkpoint",
+        "trust_score": 82,
+        "recommendation": "proceed",
+        "justification": "Phase A governance stub. Real thermodynamic detectors coming in next phase.",
+        "local_mode": True
+    }
+
+
+# ============================================================
+# HF MCP Bridge UI Functions
+# ============================================================
+
+def hf_authenticate():
+    """Trigger HF OAuth Device Code flow."""
+    result = nexus_hf_bridge.tool_authenticate()
+    return result
+
+
+def hf_create_or_update_file(repo_id: str, path_in_repo: str, content: str):
+    """Create or update a file in a Hugging Face repo via the bridge."""
+    if not repo_id or not path_in_repo or not content:
+        return {"status": "error", "message": "Please fill all fields."}
+    
+    result = nexus_hf_bridge.tool_create_or_update_file(
+        repo_id=repo_id,
+        path_in_repo=path_in_repo,
+        content=content
+    )
+    return result
+
+
+def hf_list_files(repo_id: str):
+    """List files in a repository."""
+    if not repo_id:
+        return {"status": "error", "message": "Please provide a repository ID."}
+    
+    result = nexus_hf_bridge.tool_list_repo_files(repo_id=repo_id)
+    return result
+
+
+# ============================================================
+# Gradio UI
+# ============================================================
+
+with gr.Blocks(
+    title="NEXUS Visual Weaver",
+    theme=gr.themes.Soft(),
+    css="""
+    .nexus-header { font-size: 2.1rem; font-weight: 700; }
+    .section-box { border: 1px solid #374151; border-radius: 10px; padding: 16px; margin-top: 10px; }
+    """
+) as demo:
+
+    gr.HTML("""
+    <div style="text-align: center; margin-bottom: 1rem;">
+        <h1 class="nexus-header">NEXUS Visual Weaver</h1>
+        <span style="background:#1f2937; color:#93c5fd; padding:2px 10px; border-radius:9999px; font-size:0.75rem;">
+            Phase A + HF MCP Bridge
+        </span>
+    </div>
+    """)
+
+    with gr.Tabs():
+        # ===================== TAB 1: Phase A Orchestration =====================
+        with gr.Tab("🎛️ Orchestration (Phase A)"):
+            with gr.Row():
+                with gr.Column(scale=3):
+                    prompt = gr.Textbox(
+                        label="Creative Direction",
+                        placeholder="A Slavic model in patent leather trench coat with crimson hardware...",
+                        lines=5
+                    )
+                    mode = gr.Radio(
+                        ["Strict (High Fidelity)", "Frontier (Creative Exploration)"],
+                        value="Strict (High Fidelity)",
+                        label="Reasoning Mode"
+                    )
+                with gr.Column(scale=2):
+                    gr.Markdown("### Current Status")
+                    gr.Markdown("- Phase A orchestration active\n- HF MCP Bridge ready\n- Governance layer: Stub mode")
+
+            with gr.Row():
+                refine_btn = gr.Button("🔄 Refine & Create Plan", variant="primary")
+                judge_btn = gr.Button("⚖️ Run Judgment Checkpoint", variant="secondary")
+
+            with gr.Row():
+                plan_output = gr.JSON(label="Orchestration Plan")
+                checkpoint_output = gr.JSON(label="Governance Checkpoint")
+
+            refine_btn.click(run_refine_and_plan, inputs=[prompt, mode], outputs=plan_output)
+            judge_btn.click(run_judgment_checkpoint, inputs=[plan_output], outputs=checkpoint_output)
 
-        result = gr.Image(label="Result", show_label=False)
+        # ===================== TAB 2: HF MCP Bridge =====================
+        with gr.Tab("🤖 HF MCP Bridge"):
+            gr.Markdown("## Hugging Face MCP Bridge")
+            gr.Markdown("Authenticate once, then use the tools below. All operations go through the sovereign bridge.")
 
-        with gr.Accordion("Advanced Settings", open=False):
-            seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
             with gr.Row():
-                width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=1024)
-                height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=1024)
+                auth_btn = gr.Button("🔐 Authenticate with Hugging Face (Device Code)", variant="primary", scale=1)
+                auth_status = gr.JSON(label="Authentication Status")
+
+            auth_btn.click(hf_authenticate, outputs=auth_status)
+
+            gr.Markdown("---")
+
             with gr.Row():
-                num_inference_steps = gr.Slider(
-                    label="Inference steps", minimum=1, maximum=8, step=1, value=4
-                )
-                guidance_scale = gr.Slider(
-                    label="Guidance scale", minimum=0.0, maximum=5.0, step=0.1, value=1.0
-                )
-
-        gr.Examples(examples=examples, inputs=[prompt])
-
-    gr.on(
-        triggers=[run_button.click, prompt.submit],
-        fn=infer,
-        inputs=[prompt, seed, randomize_seed, width, height, num_inference_steps, guidance_scale],
-        outputs=[result, seed],
-    )
+                with gr.Column():
+                    repo_id = gr.Textbox(label="Repository ID", placeholder="username/my-model")
+                    path_in_repo = gr.Textbox(label="Path in Repository", placeholder="README.md or folder/file.txt")
+                    content = gr.Textbox(label="File Content", lines=8, placeholder="Write your content here...")
+                    update_btn = gr.Button("📝 Create / Update File", variant="primary")
+                    update_result = gr.JSON(label="Result")
+
+                with gr.Column():
+                    list_repo = gr.Textbox(label="Repository ID to List")
+                    list_btn = gr.Button("📂 List Files")
+                    list_result = gr.JSON(label="Repository Files")
+
+            update_btn.click(
+                hf_create_or_update_file,
+                inputs=[repo_id, path_in_repo, content],
+                outputs=update_result
+            )
 
+            list_btn.click(
+                hf_list_files,
+                inputs=[list_repo],
+                outputs=list_result
+            )
+
+        # ===================== TAB 3: System =====================
+        with gr.Tab("📊 System"):
+            gr.Markdown("### NEXUS Visual Weaver Status")
+            gr.Markdown("""
+            - **Phase A**: Orchestration + Governance Checkpoint (active)
+            - **HF MCP Bridge**: Authentication + Repository tools (integrated)
+            - **MCP Server**: Enabled (`mcp_server=True`)
+            - **OAuth App**: X - GROK integration for HF
+            - **Next**: Thermodynamic governance layer + Zapier MCP integration
+            """)
+
+# ============================================================
+# Launch
+# ============================================================
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_api=True,
+        mcp_server=True,           # Important for MCP tools
+        ssr_mode=False
+    )