Dockerized

.dockerignore

@@ -0,0 +1,7 @@
+__pycache__/
+*.pyc
+*.log
+.venv/
+logs/
+*.egg-info/
+*.dist-info/

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.13-slim
+
+# Create non-root user
+RUN useradd -m -u 1000 user
+USER user
+
+# Environment
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+RUN chown -R user:1000 /app
+
+# Install pip and uv
+RUN pip install --upgrade pip && pip install uv
+
+# Copy project files
+COPY --chown=user:1000 . /app
+
+# Install dependencies
+RUN uv sync --frozen
+
+# Run the server
+CMD ["uv", "run", "/app/mcp_server.py", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -3,11 +3,8 @@ title: Unit3 MCP PR Agent
 emoji: 📚
 colorFrom: gray
 colorTo: blue
-sdk: gradio
-sdk_version: 5.34.1
-python_version: 3.13
-app_file: mcp_server.py
-pinned: false
+sdk: docker
+pinned: true
 license: apache-2.0
 short_description: MCP - Server - Gets/Adds tags - HuggingFace model repository
 tags:

mcp_server.py

@@ -3,8 +3,10 @@
 Simplified MCP Server for HuggingFace Hub Tagging Operations using FastMCP
 """
 
+import asyncio
 import os
 import json
+from fastapi import FastAPI
 from fastmcp import FastMCP
 from huggingface_hub import HfApi, model_info, ModelCard, ModelCardData
 from huggingface_hub.utils import HfHubHTTPError
@@ -21,7 +23,6 @@ hf_api = HfApi(token=HF_TOKEN) if HF_TOKEN else None
 # Create the FastMCP server
 mcp = FastMCP("hf-tagging-bot")
 
-
 @mcp.tool()
 def get_current_tags(repo_id: str) -> str:
     """Get current tags from a HuggingFace model repository"""
@@ -179,6 +180,43 @@ This PR adds the `{new_tag}` tag to the model repository.
         print(f"❌ add_new_tag error returning: {json_str}")
         return json_str
 
+# Mount the MCP app as a sub-application
+mcp_app = mcp.http_app()
+
+# Create FastAPI app
+app = FastAPI(
+    title="HF Tagging Bot - MCP Server",
+    description="A service that provides health endpoint and MCP tools",
+    version="1.0.0",
+    lifespan=mcp_app.router.lifespan_context,
+)
+
+app.mount("/mcp-server", mcp_app, "mcp")
+
+@app.get("/")
+async def root():
+    """Root endpoint with basic information"""
+    return {
+        "name": "HF Tagging Bot - MCP Server",
+        "status": "running",
+        "description": "mcp server - Webhook listener for automatic model tagging",
+        "tools": {
+            "get_current_tags": "Get current tags from a HuggingFace model repository",
+            "add_new_tag": "Add a new tag to a HuggingFace model repository via PR"
+        }
+    }
+
+@app.get("/health")
+async def health_check():
+    """Health check endpoint for monitoring"""
+    return {
+        "status": "healthy",
+        "components": {
+            "hf_token": "configured" if HF_TOKEN else "missing",
+        }
+    }
 
+# Add a simple main block for direct execution
 if __name__ == "__main__":
-    mcp.run(transport="sse")
+    import uvicorn
+    uvicorn.run("mcp_server:app", host="0.0.0.0", port=7860)

pyproject.toml

@@ -5,10 +5,9 @@ description = "MCP - Server - Gets/Adds tags - HuggingFace model repository"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
-    "huggingface_hub",
-    "gradio[mcp]",
+    "huggingface_hub[mcp]",
+    "fastapi",
     "fastmcp",
-    "python-dotenv",
 ]
 
 [build-system]