feat: Update MCP server logic and fix warnings

app.py

@@ -145,6 +145,24 @@ with gr.Blocks() as demo:
                         llm_last_frame_img = gr.Image(label="Last Frame", type="filepath", interactive=False)
                     llm_gif_output = gr.Image(label="Generated GIF", type="filepath", interactive=False)
 
+        # Hidden tab for exposing functions as MCP tools
+        with gr.Tab("MCP Tools", visible=False):
+            gr.Markdown("## Tools for MCP Server")
+            
+            # Input for video path for MCP tools
+            mcp_video_path_input = gr.Textbox(label="Video Path for MCP")
+            
+            # For letter_counter
+            lc_word_input = gr.Textbox(label="Word")
+            lc_letter_input = gr.Textbox(label="Letter")
+            lc_output = gr.Number(label="Count")
+            lc_btn = gr.Button("Count Letters")
+            
+            # For video tools
+            mcp_get_first_frame_btn = gr.Button("MCP Get First Frame")
+            mcp_get_last_frame_btn = gr.Button("MCP Get Last Frame")
+            gif_btn = gr.Button("Convert to GIF")
+
     # --- Event Handlers ---
 
     # --- System Events ---
@@ -178,13 +196,15 @@ with gr.Blocks() as demo:
     get_first_frame_btn.click(
         fn=video_tools.getFirstFrame,
         inputs=uploaded_video_path_state,
-        outputs=first_frame_img
+        outputs=first_frame_img,
+        api_name="getFirstFrame"
     )
 
     get_last_frame_btn.click(
         fn=video_tools.getLastFrame,
         inputs=uploaded_video_path_state,
-        outputs=last_frame_img
+        outputs=last_frame_img,
+        api_name="getLastFrame"
     )
 
     # --- LLM Events ---
@@ -221,6 +241,35 @@ with gr.Blocks() as demo:
         ]
     )
 
+    # --- Hidden MCP Tool Events ---
+    lc_btn.click(
+        fn=letter_counter,
+        inputs=[lc_word_input, lc_letter_input],
+        outputs=lc_output,
+        api_name="letter_counter"
+    )
+
+    mcp_get_first_frame_btn.click(
+        fn=video_tools.getFirstFrame,
+        inputs=mcp_video_path_input,
+        outputs=llm_first_frame_img, # Can reuse an existing image component
+        api_name="getFirstFrame"
+    )
+
+    mcp_get_last_frame_btn.click(
+        fn=video_tools.getLastFrame,
+        inputs=mcp_video_path_input,
+        outputs=llm_last_frame_img, # Can reuse an existing image component
+        api_name="getLastFrame"
+    )
+    
+    gif_btn.click(
+        fn=video_tools.convert_mp4_to_gif,
+        inputs=mcp_video_path_input,
+        outputs=llm_gif_output,
+        api_name="convert_mp4_to_gif"
+    )
+
     # --- App Load Event ---
     demo.load(
         fn=on_load_tasks,
@@ -239,16 +288,6 @@ with gr.Blocks() as demo:
 
 # --- MCP Server Launch ---
 if __name__ == "__main__":
-    # Expose the functions we want to be available as tools
-    demo.add_tools(
-        {
-            "letter_counter": letter_counter,
-            "getFirstFrame": video_tools.getFirstFrame,
-            "getLastFrame": video_tools.getLastFrame,
-            "convert_mp4_to_gif": video_tools.convert_mp4_to_gif,
-        }
-    )
-    # Launch the Gradio app with MCP server enabled
     demo.launch(mcp_server=True)
 
 

requirements.txt

@@ -1,2 +1,2 @@
-gradio[mcp]==4.29.0
+gradio[mcp]
 requests