Fix import error and update Gradio interface with all MCP tools

app.py

@@ -1,35 +1,55 @@
 import gradio as gr
 import json
-from task_manager import TaskManagerServer
-import asyncio
+import task_manager
 
-# Create MCP server instance
-server = TaskManagerServer()
-
-async def call_mcp_tool(tool_name, arguments_json):
+def call_mcp_tool(tool_name, arguments_json):
     """Call MCP tools via web interface"""
     try:
         args = json.loads(arguments_json) if arguments_json else {}
         
-        if tool_name == "add_task":
-            result = await server.add_task(args.get("title", ""), args.get("description", ""))
+        if tool_name == "create_task":
+            result = task_manager.create_task(
+                title=args.get("title", ""),
+                description=args.get("description", ""),
+                priority=args.get("priority", "medium"),
+                tags=args.get("tags", ""),
+                due_date=args.get("due_date", "")
+            )
         elif tool_name == "list_tasks":
-            result = await server.list_tasks()
-        elif tool_name == "complete_task":
-            result = await server.complete_task(args.get("task_id", 0))
+            result = task_manager.list_tasks(
+                status=args.get("status", "all"),
+                priority=args.get("priority", "all"),
+                tag=args.get("tag", ""),
+                limit=args.get("limit", 10)
+            )
+        elif tool_name == "get_task":
+            result = task_manager.get_task(args.get("task_id", 0))
+        elif tool_name == "update_task":
+            result = task_manager.update_task(
+                task_id=args.get("task_id", 0),
+                title=args.get("title", ""),
+                description=args.get("description", ""),
+                status=args.get("status", ""),
+                priority=args.get("priority", ""),
+                tags=args.get("tags", ""),
+                due_date=args.get("due_date", "")
+            )
         elif tool_name == "delete_task":
-            result = await server.delete_task(args.get("task_id", 0))
+            result = task_manager.delete_task(args.get("task_id", 0))
+        elif tool_name == "search_tasks":
+            result = task_manager.search_tasks(
+                query=args.get("query", ""),
+                limit=args.get("limit", 10)
+            )
+        elif tool_name == "get_stats":
+            result = task_manager.get_stats()
         else:
             return "Unknown tool"
             
-        return json.dumps(result, indent=2)
+        return result
     except Exception as e:
         return f"Error: {str(e)}"
 
-def sync_call_mcp_tool(tool_name, arguments_json):
-    """Sync wrapper for async function"""
-    return asyncio.run(call_mcp_tool(tool_name, arguments_json))
-
 # Create Gradio interface
 with gr.Blocks(title="Task Manager MCP Server") as demo:
     gr.Markdown("# Task Manager MCP Server")
@@ -37,7 +57,7 @@ with gr.Blocks(title="Task Manager MCP Server") as demo:
     
     with gr.Row():
         tool_dropdown = gr.Dropdown(
-            choices=["add_task", "list_tasks", "complete_task", "delete_task"],
+            choices=["create_task", "list_tasks", "get_task", "update_task", "delete_task", "search_tasks", "get_stats"],
             label="Select Tool",
             value="list_tasks"
         )
@@ -51,8 +71,21 @@ with gr.Blocks(title="Task Manager MCP Server") as demo:
     submit_btn = gr.Button("Call Tool")
     output = gr.Textbox(label="Result", lines=10)
     
+    # Add example buttons for common operations
+    with gr.Row():
+        gr.Examples(
+            examples=[
+                ["create_task", '{"title": "Learn MCP", "description": "Study Model Context Protocol", "priority": "high"}'],
+                ["list_tasks", '{}'],
+                ["get_stats", '{}'],
+                ["search_tasks", '{"query": "learn"}']
+            ],
+            inputs=[tool_dropdown, arguments_input],
+            label="Quick Examples"
+        )
+    
     submit_btn.click(
-        sync_call_mcp_tool,
+        call_mcp_tool,
         inputs=[tool_dropdown, arguments_input],
         outputs=output
     )

requirements.txt

@@ -1,3 +1,4 @@
 gradio
 mcp
-pydantic
+pydantic
+fastmcp