Update app.py

app.py

@@ -693,58 +693,146 @@ sudo iptables-save > /etc/iptables/rules.v4  # Save rules
             
             # Handle other servers (Terraform, Cisco)
             else:
-                # Get available tools for the selected server
-                tools = self.get_available_tools(selected_server)
-                
-                if tools:
-                    # Show available tools and generate helpful response
-                    tools_info = "\n".join([f"- {tool.get('name', 'Unknown')}: {tool.get('description', 'No description')}" 
-                                          for tool in tools[:5]])
-                    
-                    server_info = {
-                        "server": self.mcp_servers[selected_server]['name'],
-                        "available_tools": len(tools),
-                        "sample_tools": tools_info
-                    }
+                # Try to get server information first
+                try:
+                    # Check if server provides any status info
+                    server_url = self.mcp_servers[selected_server]['space_url']
+                    status_response = requests.get(server_url, timeout=10)
                     
-                    response = self.generate_response(user_input, server_info)
-                else:
-                    # Provide guidance for servers without direct HTTP tool access
-                    if selected_server == "terraform":
-                        response = f"""**Terraform MCP Server**
+                    if status_response.status_code == 200:
+                        try:
+                            server_data = status_response.json()
+                            tools_count = server_data.get('tools', 'unknown')
+                            service_name = server_data.get('service', 'unknown')
+                            
+                            if selected_server == "cisco":
+                                response = f"""**Cisco MCP Server** ✅ Connected!
 
-I can help you with Terraform infrastructure management, but this server uses MCP protocol over SSE.
+**Server Status**: Online 
+**Service**: {service_name}
+**Available Tools**: {tools_count}
+**MCP Endpoint**: {server_url}/mcp/sse
 
-**Common Terraform operations:**
-- Plan infrastructure changes (`terraform plan`)
-- Apply configurations (`terraform apply`) 
-- Destroy resources (`terraform destroy`)
-- Validate configurations (`terraform validate`)
-- Import existing resources (`terraform import`)
+**What you can ask for with Cisco:**
+- "Show me Cisco interface configuration commands"
+- "Help with VLAN configuration"
+- "Generate Cisco routing commands"
+- "Show network troubleshooting commands"
+- "Configure access control lists"
+- "Help with Cisco switch configuration"
 
-**Server Status**: {health_check['status']}
-**MCP Endpoint**: {self.mcp_servers[selected_server]['space_url']}/mcp/sse
+**Note**: This server has {tools_count} tools available via MCP protocol. The tools likely include:
+- Interface configuration management
+- VLAN configuration and management  
+- Routing protocol setup
+- ACL (Access Control List) configuration
+- Network troubleshooting commands
+- Device status and monitoring
 
-*Note: For full functionality, this server requires MCP protocol connection.*
+Would you like me to help you with any specific Cisco networking task?
 """
-                    
-                    elif selected_server == "cisco":
-                        response = f"""**Cisco MCP Server**
+                            elif selected_server == "terraform":
+                                response = f"""**Terraform MCP Server** ✅ Connected!
 
-I can help you with Cisco network device management, but this server uses MCP protocol over SSE.
+**Server Status**: Online
+**Service**: {service_name}  
+**Available Tools**: {tools_count}
+**MCP Endpoint**: {server_url}/mcp/sse
+
+**What you can ask for with Terraform:**
+- "Show me Terraform plan commands"
+- "Help with infrastructure deployment"
+- "Generate Terraform configuration"
+- "Show resource management commands"
+- "Help with Terraform state management"
+
+**Note**: This server has {tools_count} tools available via MCP protocol.
+
+Would you like me to help you with any specific Terraform task?
+"""
+                        except json.JSONDecodeError:
+                            # Server responded but not with JSON
+                            if selected_server == "cisco":
+                                response = f"""**Cisco MCP Server** ⚠️ Connected but Limited Info
+
+**Server Status**: Online (non-JSON response)
+**MCP Endpoint**: {server_url}/mcp/sse
+
+**Common Cisco operations I can help with:**
+- Interface configuration commands
+- VLAN setup and management
+- Routing protocol configuration
+- Access Control Lists (ACLs)
+- Network troubleshooting
+- Switch and router configuration
+
+Would you like me to help with any specific Cisco networking task?
+"""
+                            else:
+                                response = f"""**Terraform MCP Server** ⚠️ Connected but Limited Info
 
-**Common Cisco operations:**
-- Show interface configurations
-- Configure VLANs 
-- Set up routing protocols
-- Manage access control lists
-- Check device status
-- Configure port settings
+**Server Status**: Online (non-JSON response)  
+**MCP Endpoint**: {server_url}/mcp/sse
+
+**Common Terraform operations I can help with:**
+- Infrastructure planning and deployment
+- Resource configuration and management
+- State file management
+- Module development
+- Provider configuration
+
+Would you like me to help with any specific Terraform task?
+"""
+                    else:
+                        # Server not responding properly
+                        if selected_server == "cisco":
+                            response = f"""**Cisco MCP Server** ❌ Connection Issue
 
 **Server Status**: {health_check['status']}
+**Error**: HTTP {status_response.status_code}
+**MCP Endpoint**: {server_url}/mcp/sse
+
+The server appears to be having issues. Please try again later or check if the server is running properly.
+"""
+                        else:
+                            response = f"""**Terraform MCP Server** ❌ Connection Issue
+
+**Server Status**: {health_check['status']}  
+**Error**: HTTP {status_response.status_code}
+**MCP Endpoint**: {server_url}/mcp/sse
+
+The server appears to be having issues. Please try again later or check if the server is running properly.
+"""
+                            
+                except Exception as e:
+                    # Connection failed
+                    if selected_server == "cisco":
+                        response = f"""**Cisco MCP Server** ❌ Connection Failed
+
+**Error**: {str(e)}
 **MCP Endpoint**: {self.mcp_servers[selected_server]['space_url']}/mcp/sse
 
-*Note: For full functionality, this server requires MCP protocol connection.*
+**Common Cisco operations (offline reference):**
+- Interface configuration: `interface GigabitEthernet0/1`
+- VLAN configuration: `vlan 10`
+- Routing: `ip route 0.0.0.0 0.0.0.0 192.168.1.1`
+- ACL: `access-list 10 permit 192.168.1.0 0.0.0.255`
+
+Please check if the server is running and try again.
+"""
+                    else:
+                        response = f"""**Terraform MCP Server** ❌ Connection Failed
+
+**Error**: {str(e)}
+**MCP Endpoint**: {self.mcp_servers[selected_server]['space_url']}/mcp/sse
+
+**Common Terraform operations (offline reference):**
+- Plan: `terraform plan`
+- Apply: `terraform apply`
+- Destroy: `terraform destroy`
+- Validate: `terraform validate`
+
+Please check if the server is running and try again.
 """
             
             self.conversation_history.append({"role": "assistant", "content": response})