Update ui_components.py

ui_components.py

@@ -8,7 +8,6 @@ from config import AppConfig, CUSTOM_CSS
 from chat_handler import ChatHandler
 from server_manager import ServerManager
 from mcp_client import UniversalMCPClient
-from mcp_spaces_finder import create_simple_mcp_selector
 
 class UIComponents:
     """Manages Gradio UI components and event handlers"""
@@ -70,11 +69,17 @@ class UIComponents:
         with gr.Accordion("⚙️ Manage Servers", open=False):
             gr.Markdown("**Add MCP Servers**")
             
-            # Create MCP spaces dropdown selector
-            self.mcp_dropdown, self.selected_space_textbox = create_simple_mcp_selector(
-                dropdown_label="🤖 Select MCP Server",
-                textbox_label="Selected HuggingFace Space",
-                placeholder="No server selected"
+            # Get MCP spaces count for dropdown label
+            from mcp_spaces_finder import _finder
+            spaces = _finder.get_mcp_spaces()
+            spaces_count = len(spaces)
+            
+            # Create MCP spaces dropdown (without the textbox)
+            self.mcp_dropdown = gr.Dropdown(
+                choices=spaces,
+                label=f"🤖 Select from {spaces_count} Available MCP Server Spaces",
+                value=None,
+                info="Choose a HuggingFace space that provides MCP server functionality"
             )
             
             # Server title input
@@ -157,10 +162,10 @@ class UIComponents:
             """Handle adding a server and update displays"""
             # Check if both title and space are provided
             if not server_title or not server_title.strip():
-                return "❌ Please provide a server title", "", *update_server_display(), "", ""
+                return "❌ Please provide a server title", "", *update_server_display(), ""
             
-            if not selected_space or selected_space == "No server selected":
-                return "❌ Please select a HuggingFace space from the dropdown", "", *update_server_display(), "", ""
+            if not selected_space:
+                return "❌ Please select a HuggingFace space from the dropdown", "", *update_server_display(), ""
             
             # Use the selected space from dropdown
             status_msg, details_html = self.server_manager.add_custom_server(server_title.strip(), selected_space)
@@ -168,7 +173,7 @@ class UIComponents:
             # Update sidebar server display
             count_text, list_text = update_server_display()
             
-            return status_msg, details_html, count_text, list_text, "", ""  # Clear inputs
+            return status_msg, details_html, count_text, list_text, ""  # Clear server name input
         
         def handle_refresh_status():
             """Handle refresh status button"""
@@ -182,21 +187,26 @@ class UIComponents:
             # Clear cache and get fresh spaces
             refresh_mcp_spaces()
             spaces = _finder.get_mcp_spaces()
-            
-            # Update dropdown choices
-            return gr.Dropdown(choices=spaces, value=None, label=f"🤖 Select MCP Server ({len(spaces)} available)")
+            spaces_count = len(spaces)
+            
+            # Update dropdown choices with new count
+            return gr.Dropdown(
+                choices=spaces, 
+                value=None, 
+                label=f"🤖 Select from {spaces_count} Available MCP Server Spaces",
+                info="Choose a HuggingFace space that provides MCP server functionality"
+            )
         
         # Connect server management events
         self.add_server_btn.click(
             handle_add_server,
-            inputs=[self.server_name, self.selected_space_textbox],  # Use the textbox that shows selected space
+            inputs=[self.server_name, self.mcp_dropdown],  # Use dropdown directly
             outputs=[
                 self.add_server_output, 
                 self.add_server_details, 
                 self.server_count_display, 
                 self.server_list_display, 
-                self.server_name, 
-                self.selected_space_textbox  # Clear the selected space display
+                self.server_name  # Clear server name input
             ]
         )