Update app.py

app.py

@@ -3,107 +3,71 @@ import requests
 from smolagents import CodeAgent
 import logging
 from tenacity import retry, stop_after_attempt, wait_exponential
-from typing import Optional
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Correct URL based on your Space
 HF_SPACE_URL = "https://manavraj-troubleshoot-mcp.hf.space"
 
 @retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=4, max=10))
 def call_mcp_server(message: str, tool_type: str = "knowledge_base") -> str:
-    """Call MCP server with proper endpoint routing"""
+    """Call MCP server endpoint"""
     try:
-        endpoint_map = {
-            "knowledge_base": "api/knowledge_base",
-            "web_search": "api/web_search",
-            "formatter": "api/formatter"
-        }
-        
-        endpoint = endpoint_map.get(tool_type, "api/knowledge_base")
-        url = f"{HF_SPACE_URL}/{endpoint}"
-        
-        response = requests.post(
-            url,
-            json={"data": [message]},
-            timeout=30
-        )
-        
-        if response.status_code != 200:
-            raise Exception(f"MCP server returned {response.status_code}")
-            
-        data = response.json()
-        if not isinstance(data, dict) or 'data' not in data:
-            raise Exception("Invalid MCP response format")
-            
-        return str(data['data'][0])
-        
+        endpoint = f"{HF_SPACE_URL}/api/{tool_type}"
+        response = requests.post(endpoint, json={"data": [message]}, timeout=30)
+        response.raise_for_status()
+        return response.json()['data'][0]
     except Exception as e:
-        logger.error(f"MCP call failed: {str(e)}")
+        logger.error(f"API call failed: {str(e)}")
         return f"Error: {str(e)}"
 
-# Initialize CodeAgent with proper system prompt including required tags
 agent = CodeAgent(
-    tools=[call_mcp_server],
+    tools=[],
     model="microsoft/DialoGPT-medium",
     system_prompt="""{{authorized_imports}}
 - requests for API calls
 - standard Python libraries
 {{managed_agents_descriptions}}
-You are an advanced Technical Support Assistant with these capabilities:
-1. Troubleshooting technical issues (wifi, hardware, software)
-2. Finding information through web search
-3. Formatting instructions into clear steps
 
-Available Tools:
-- knowledge_base: For technical issues (wifi, screen, sound problems)
-- web_search: For finding latest information or non-technical queries
-- formatter: To organize raw steps into numbered lists
+You are a Technical Support Assistant with these capabilities:
+1. Troubleshooting technical issues
+2. Finding information via web search
+3. Formatting instructions
 
-Response Guidelines:
-1. INTERNALLY follow THOUGHT/ACTION/OBSERVATION cycle
-2. Use knowledge_base for technical troubleshooting
-3. Use web_search for information lookup
-4. Use formatter when user provides unorganized steps
-5. Provide FINAL RESPONSE in clear, helpful language
+Access tools through MCP server:
+- knowledge_base: For technical issues
+- web_search: For information lookup
+- formatter: To organize steps
 
-Example workflow:
-THOUGHT: User has wifi issue, should check knowledge base
-ACTION: knowledge_base("wifi keeps disconnecting")
-OBSERVATION: Found 4 troubleshooting steps
-FINAL RESPONSE: Here's how to fix your wifi: [steps]
+Response workflow:
+1. Analyze user request
+2. Choose appropriate tool
+3. Return clear response
 
-Never show THOUGHT/ACTION/OBSERVATION to user - only final response."""
+Example:
+USER: My wifi disconnected
+THOUGHT: Should use knowledge_base
+ACTION: knowledge_base("wifi disconnection")
+RESPONSE: Try these steps: [solution steps]
+
+Never show internal workflow to user."""
 )
 
-def chat_interface(message: str, history: Optional[list] = None) -> str:
-    """Simplified chat interface"""
+def chat_interface(message: str, history: list) -> str:
+    """Handle chat interaction"""
     try:
-        # Let agent handle the complete process
         response = agent.run(message)
-        
-        # Extract final response and clean it up
-        if isinstance(response, str):
-            if "FINAL RESPONSE:" in response:
-                return response.split("FINAL RESPONSE:")[-1].strip()
-            return response
-        return str(response)
+        final = str(response).split("RESPONSE:")[-1].strip()
+        return final if final else "I couldn't process that request."
     except Exception as e:
         logger.error(f"Chat error: {str(e)}")
-        return f"Error processing request: {str(e)}"
+        return f"Error: {str(e)}"
 
-# Gradio interface
 demo = gr.ChatInterface(
     fn=chat_interface,
-    title="🔧 Technical Support Agent",
-    examples=[
-        "My wifi keeps disconnecting",
-        "Find the latest Windows 11 update",
-        "Format: Restart. Check connections. Update drivers"
-    ],
-    description="A technical support assistant that can troubleshoot issues, search the web, and format instructions."
+    title="🔧 Technical Support",
+    examples=["Wifi not working", "Find Windows 11 specs", "Format: Turn off. Wait. Restart"]
 )
 
 if __name__ == "__main__":