Update app.py

app.py

@@ -1,82 +1,71 @@
 import gradio as gr
 import spaces
-from nexus_hf_mcp_bridge import nexus_hf_bridge
-from nexus_hf_mcp_bridge.hf_oauth import HFOAuthClient
-from nexus_hf_mcp_bridge.token_manager import HFTokenManager
+import os
+from huggingface_hub import HfApi
 
 @spaces.GPU
 def _init_gpu():
     pass
 
-client = HFOAuthClient()
-device_data = gr.State({})   # Better than global for Gradio
+# Read token from Space Secret (recommended) or environment
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
-def start_auth():
-    try:
-        data = client.start_device_flow()
-        return data, {"status": "pending", "message": "Open the URL and enter the code"}
-    except Exception as e:
-        return {}, {"status": "error", "message": str(e)}
+def get_status():
+    if HF_TOKEN:
+        return "✅ Token loaded from Space Secrets"
+    return "❌ No HF_TOKEN found. Please add it in Space Settings → Secrets"
+
+def create_or_update_file(repo_id: str, path_in_repo: str, content: str):
+    if not HF_TOKEN:
+        return {"status": "error", "message": "HF_TOKEN not found. Add it in Space Secrets."}
+    if not repo_id or not path_in_repo or not content:
+        return {"status": "error", "message": "Please fill all fields"}
 
-def complete_auth(data):
-    if not data:
-        return {"status": "error", "message": "Please start authentication first"}
     try:
-        token = client.poll_for_token(data["device_code"])
-        if token:
-            tm = HFTokenManager()
-            tm.save_token(token)
-            return {"status": "success", "message": "✅ Authentication successful!"}
-        else:
-            return {"status": "error", "message": "Not authorized yet or timed out"}
+        api = HfApi(token=HF_TOKEN)
+        api.upload_file(
+            path_or_fileobj=content.encode("utf-8"),
+            path_in_repo=path_in_repo,
+            repo_id=repo_id,
+            commit_message="Updated via NEXUS HF Bridge"
+        )
+        return {"status": "success", "message": f"File '{path_in_repo}' updated successfully"}
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
-def get_status():
-    tm = HFTokenManager()
-    return "✅ Authenticated" if tm.get_valid_token() else "Not authenticated"
-
-def safe_write(repo_id, path, content):
-    tm = HFTokenManager()
-    if not tm.get_valid_token():
-        return {"error": "Please authenticate first"}
-    if not all([repo_id, path, content]):
-        return {"error": "Please fill all fields"}
-    return nexus_hf_bridge.tool_create_or_update_file(repo_id, path, content)
-
-def safe_list(repo_id):
-    tm = HFTokenManager()
-    if not tm.get_valid_token():
-        return {"error": "Please authenticate first"}
+def list_repo_files(repo_id: str):
+    if not HF_TOKEN:
+        return {"status": "error", "message": "HF_TOKEN not found"}
     if not repo_id:
-        return {"error": "Please provide repository ID"}
-    return nexus_hf_bridge.tool_list_repo_files(repo_id)
+        return {"status": "error", "message": "Please provide repository ID"}
+
+    try:
+        api = HfApi(token=HF_TOKEN)
+        files = api.list_repo_files(repo_id=repo_id)
+        return {"status": "success", "files": files}
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
 
+# ===================== UI =====================
 with gr.Blocks(title="NEXUS Visual Weaver") as demo:
     gr.Markdown("# NEXUS Visual Weaver - HF MCP Bridge")
-
-    with gr.Tab("Authentication"):
-        gr.Markdown("### Step 1: Start Authentication")
-        start_btn = gr.Button("Start Authentication", variant="primary")
-        device_info = gr.JSON(label="Device Code Info")
-
-        gr.Markdown("### Step 2: Complete Authentication")
-        complete_btn = gr.Button("I have authorized in browser", variant="secondary")
-        result = gr.JSON(label="Result")
-
-        start_btn.click(start_auth, outputs=[device_info, result])
-        complete_btn.click(complete_auth, inputs=[device_info], outputs=result)
-
-        gr.Markdown("### Status")
-        gr.Button("Check Authentication Status").click(get_status, outputs=gr.Textbox())
+    gr.Markdown("**Using Space Secrets (secure)**")
 
     with gr.Tab("Repository Tools"):
-        repo = gr.Textbox(label="Repository ID")
-        path = gr.Textbox(label="Path in Repo")
+        repo_id = gr.Textbox(label="Repository ID", placeholder="username/model-name")
+        path = gr.Textbox(label="Path in Repo", placeholder="folder/file.txt")
         content = gr.Textbox(label="Content", lines=6)
 
         with gr.Row():
-            gr.Button("Create / Update File").click(safe_write, inputs=[repo, path, content], outputs=gr.JSON())
-            gr.Button("List Files").click(safe_list, inputs=[repo], outputs=gr.JSON())
+            write_btn = gr.Button("Create / Update File", variant="primary")
+            list_btn = gr.Button("List Files")
+
+        result = gr.JSON(label="Result")
+
+        write_btn.click(create_or_update_file, inputs=[repo_id, path, content], outputs=result)
+        list_btn.click(list_repo_files, inputs=[repo_id], outputs=result)
+
+    with gr.Tab("Status"):
+        gr.Button("Check Token Status").click(get_status, outputs=gr.Textbox())
 
 demo.launch(mcp_server=True)