Update app.py

app.py

@@ -1,46 +1,57 @@
-#!/usr/bin/env python3
-""" MCP Server using FastMCP + Hugging Face Hub SDK (No .env, uses Space secrets) """
-
 import os
 import json
+import gradio as gr
 from fastmcp import FastMCP
 from huggingface_hub import HfApi, model_info
-from huggingface_hub.utils import HfHubHTTPError
 
-# — Configuration —
-HF_TOKEN = os.environ.get("HF_TOKEN")  # Set this in Space → Settings → Variables and secrets
+# — Hugging Face Token from Space Secrets —
+HF_TOKEN = os.environ.get("HF_TOKEN")
 hf_api = HfApi(token=HF_TOKEN) if HF_TOKEN else None
 
-# — Create MCP Server —
+# — FastMCP Server Setup —
 mcp = FastMCP("hf-tagging-bot")
 
-# — Tool: Get Current Tags —
 @mcp.tool()
 def get_current_tags(repo_id: str) -> str:
     """Get current tags from a HuggingFace model repository"""
-    print(f"🔧 get_current_tags called with repo_id: {repo_id}")
     if not hf_api:
-        error_result = {"error": "HF token not configured"}
-        return json.dumps(error_result)
+        return json.dumps({"error": "HF token not configured"})
 
     try:
         info = model_info(repo_id=repo_id, token=HF_TOKEN)
         current_tags = info.tags if info.tags else []
-        result = {
+        return json.dumps({
             "status": "success",
             "repo_id": repo_id,
             "current_tags": current_tags,
             "count": len(current_tags),
-        }
-        return json.dumps(result)
+        })
     except Exception as e:
-        error_result = {
+        return json.dumps({
             "status": "error",
             "repo_id": repo_id,
             "error": str(e),
-        }
-        return json.dumps(error_result)
+        })
+
+# — Gradio UI Wrapper —
+def gradio_tag_checker(repo_id):
+    return get_current_tags(repo_id)
+
+# — Launch Gradio Interface —
+demo = gr.Interface(
+    fn=gradio_tag_checker,
+    inputs=gr.Textbox(label="Enter HuggingFace Model Repo ID", placeholder="e.g. bert-base-uncased"),
+    outputs=gr.Textbox(label="Current Tags (JSON)"),
+    title="🔖 HuggingFace Tag Checker",
+    description="Uses FastMCP + HuggingFace Hub SDK to fetch current tags from any model repo."
+)
 
-# — Run MCP Server —
+# — Run Both MCP Server and Gradio —
 if __name__ == "__main__":
-    mcp.run()
+    import threading
+
+    # Run FastMCP in background thread
+    threading.Thread(target=mcp.run, daemon=True).start()
+
+    # Run Gradio UI
+    demo.launch()