Update app.py

app.py

@@ -95,69 +95,63 @@ class MCPClient:
     
     def analyze_user_request(self, user_input: str) -> Dict[str, Any]:
         """Use OpenAI to analyze user request and determine which MCP server to use"""
-        system_prompt = f"""
+        system_prompt = """
         You are an intelligent assistant that helps users interact with MCP (Model Context Protocol) servers.
         
         Available MCP servers (use these EXACT keys in your response):
-        1. terraform: {self.mcp_servers['terraform']['description']}
-        2. linux: {self.mcp_servers['linux']['description']}  
-        3. cisco: {self.mcp_servers['cisco']['description']}
+        1. terraform: Terraform infrastructure management
+        2. linux: Linux system operations  
+        3. cisco: Cisco network management
         
         IMPORTANT: For the "recommended_server" field, you MUST use one of these exact keys: "terraform", "linux", or "cisco"
         
-        Analyze the user's request and determine:
-        1. Which MCP server(s) would be most appropriate
-        2. What specific tools or operations might be needed
-        3. Any parameters or arguments that should be passed
+        Analyze the user's request and determine which MCP server would be most appropriate.
         
         Respond in JSON format with:
-        {{
+        {
             "recommended_server": "terraform|linux|cisco",
             "reasoning": "explanation of why this server was chosen",
-            "suggested_action": "what action to take",
-            "parameters": {{"key": "value"}}
-        }}
+            "suggested_action": "what action to take"
+        }
         """
         
         try:
+            response = self.openai_client.chat.completions.create(
+                model="gpt-4",
+                messages=[
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": user_input}
+                ],
+                temperature=0.3
+            )
+            
+            content = response.choices[0].message.content.strip()
+            
+            # Try to parse JSON, handle cases where GPT returns non-JSON
             try:
-                response = self.openai_client.chat.completions.create(
-                    model="gpt-4",
-                    messages=[
-                        {"role": "system", "content": system_prompt},
-                        {"role": "user", "content": user_input}
-                    ],
-                    temperature=0.3
-                )
-                
-                content = response.choices[0].message.content.strip()
+                return json.loads(content)
+            except json.JSONDecodeError:
+                # If not valid JSON, extract server recommendation manually
+                content_lower = content.lower()
+                if "cisco" in content_lower:
+                    recommended_server = "cisco"
+                elif "linux" in content_lower:
+                    recommended_server = "linux"
+                elif "terraform" in content_lower:
+                    recommended_server = "terraform"
+                else:
+                    recommended_server = "cisco"  # Default for network questions
                 
-                # Try to parse JSON, handle cases where GPT returns non-JSON
-                try:
-                    return json.loads(content)
-                except json.JSONDecodeError:
-                    # If not valid JSON, extract server recommendation manually
-                    content_lower = content.lower()
-                    if "cisco" in content_lower:
-                        recommended_server = "cisco"
-                    elif "linux" in content_lower:
-                        recommended_server = "linux"
-                    elif "terraform" in content_lower:
-                        recommended_server = "terraform"
-                    else:
-                        recommended_server = "cisco"  # Default for network questions
-                    
-                    return {
-                        "recommended_server": recommended_server,
-                        "reasoning": "Based on content analysis",
-                        "suggested_action": content,
-                        "parameters": {}
-                    }
+                return {
+                    "recommended_server": recommended_server,
+                    "reasoning": "Based on content analysis",
+                    "suggested_action": content
+                }
                 
         except Exception as e:
             return {
                 "error": f"Failed to analyze request: {str(e)}",
-                "recommended_server": "cisco"  # Default fallback
+                "recommended_server": "cisco"
             }
     
     def generate_response(self, user_input: str, mcp_results: Dict[str, Any] = None) -> str:
@@ -323,7 +317,7 @@ def create_gradio_interface():
             history.append([message, error_msg])
             return history, ""
     
-    def get_server_status(self):
+    def get_server_status():
         status_info = "## MCP Servers Status\n\n"
         for key, config in (mcp_client.mcp_servers.items() if mcp_client else []):
             try:
@@ -472,5 +466,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=True,
         debug=True
-    )
-    # Now?
+    )