Working the deploy

app.py

@@ -1,134 +1,31 @@
-#!/usr/bin/env python3
-"""
-SmolAgents Authentication Fix
-Resolves 401 "Invalid username or password" errors
-"""
-
+import gradio as gr
 import os
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, InferenceClientModel
-
-# Method 1: Explicit token (most reliable)
-def create_agent_with_token():
-    """Create agent with explicit HF token - most reliable method"""
-    # Replace with your actual HuggingFace token
-    hf_token = "hf_xxxxxxxxxxxxxxxxxxxxxxxxxx"
-
-    model = HfApiModel(
-        model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-        token=hf_token
-    )
-
-    agent = CodeAgent(
-        tools=[DuckDuckGoSearchTool()],
-        model=model
-    )
-    return agent
-
-# Method 2: Environment variable (recommended for production)
-def create_agent_with_env_var():
-    """Create agent using HF_TOKEN environment variable"""
-    # Set environment variable first:
-    # export HF_TOKEN="hf_xxxxxxxxxxxxxxxxxxxxxxxxxx"
-
-    # Verify token is set
-    if not os.getenv("HF_TOKEN"):
-        raise ValueError("HF_TOKEN environment variable not set!")
-
-    model = HfApiModel(
-        model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
-        # token will be read from HF_TOKEN automatically
-    )
 
-    agent = CodeAgent(
-        tools=[DuckDuckGoSearchTool()],
-        model=model
-    )
-    return agent
+from smolagents import InferenceClientModel, CodeAgent, MCPClient
 
-# Method 3: InferenceClientModel (newest approach, 2025)
-def create_agent_with_inference_client():
-    """Create agent using newer InferenceClientModel - better error handling"""
-    hf_token = "hf_xxxxxxxxxxxxxxxxxxxxxxxxxx"
 
-    model = InferenceClientModel(
-        model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-        token=hf_token
+try:
+    mcp_client = MCPClient(
+        {"url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"}
     )
+    #    mcp_client = MCPClient(
+    #    {"url": " http://127.0.0.1:7860/gradio_api/mcp/sse"}
+    #)
 
-    agent = CodeAgent(
-        tools=[DuckDuckGoSearchTool()],
-        model=model
-    )
-    return agent
+    tools = mcp_client.get_tools()
 
-# Method 4: Login first (alternative approach)
-def create_agent_with_login():
-    """Create agent after logging in to HuggingFace Hub"""
-    from huggingface_hub import login
+    model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
 
-    # Login to HF (will prompt for token if not provided)
-    login(token="hf_xxxxxxxxxxxxxxxxxxxxxxxxxx")
+    agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
 
-    # Now HfApiModel should work without explicit token
-    model = HfApiModel(
-        model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
+    demo = gr.ChatInterface(
+        fn=lambda message, history: str(agent.run(message)),
+        type="messages",
+        examples=["Analyze the sentiment of the following text 'This is awesome'"],
+        title="Agent with MCP Tools",
+        description="This is a simple agent that uses MCP tools to answer questions.",
     )
 
-    agent = CodeAgent(
-        tools=[DuckDuckGoSearchTool()],
-        model=model
-    )
-    return agent
-
-def test_agent(agent):
-    """Test the agent with a simple query"""
-    try:
-        result = agent.run("What is the current time?")
-        print("✅ Success! Agent is working properly.")
-        print(f"Result: {result}")
-        return True
-    except Exception as e:
-        print(f"❌ Error: {e}")
-        return False
-
-if __name__ == "__main__":
-    print("Testing SmolAgents authentication fixes...\n")
-
-    # Try Method 1 first (most common solution)
-    print("Method 1: Explicit token")
-    try:
-        agent = create_agent_with_token()
-        if test_agent(agent):
-            print("✅ Method 1 successful!\n")
-        else:
-            print("❌ Method 1 failed, trying next method...\n")
-    except Exception as e:
-        print(f"❌ Method 1 failed: {e}\n")
-
-    # Try Method 3 (newest approach)
-    print("Method 3: InferenceClientModel")
-    try:
-        agent = create_agent_with_inference_client()
-        if test_agent(agent):
-            print("✅ Method 3 successful!\n")
-    except Exception as e:
-        print(f"❌ Method 3 failed: {e}\n")
-
-# Token Requirements Checklist:
-"""
-Your HuggingFace token must have these permissions:
-✅ "Make calls to the serverless Inference API"
-✅ "Read access to contents of all public gated repos" (for gated models)
-
-To get your token:
-1. Go to: https://huggingface.co/settings/tokens
-2. Click "New token"
-3. Select "Write" permissions
-4. Copy the token (starts with hf_)
-
-Common issues:
-- Token is expired or revoked
-- Token lacks proper permissions
-- Model is gated and requires special access
-- Network/firewall blocking HF API calls
-"""
+    demo.launch(share=True)
+finally:
+    mcp_client.disconnect()