Fix endpoint issue

app.py

@@ -497,75 +497,45 @@ async def chat_ui():
 @app.post("/chat", response_model=ChatOutput)
 async def chat_endpoint(chat_input: ChatInput) -> ChatOutput:
     """
-    Chat endpoint for the web interface - calls agent directly
+    Chat endpoint for the web interface - uses MCP client
     """
     logger.info(f"💬 Direct chat request: {chat_input.message[:100]}...")
     
     try:
-        # Call the agent API directly
-        agent_url = "https://agent.mcp.integration.ambris.com/index.php?fc=module&module=ambaiagent&controller=ajax"
-        
-        # Prepare the request for the agent
-        agent_payload = {
-            "action": "send_prompt",
-            "prompt": chat_input.message,
-            "context": {
-                "history": [msg.dict() for msg in chat_input.conversation_history],
-                "session_id": chat_input.session_id or f"hf_chat_{int(time.time())}"
-            }
-        }
+        # Use the MCP client to process the request
+        if not mcp_client:
+            raise HTTPException(status_code=500, detail="MCP client not initialized")
+        
+        # Process the message through MCP
+        # For now, let's use a simple search as an example
+        search_result = await mcp_client.call_tool(
+            "search_products",
+            {"query": chat_input.message, "client_token": chat_input.client_token}
+        )
+        
+        # Format the response
+        response_content = f"Based on your request '{chat_input.message}', here's what I found:\n\n{search_result}"
+        
+        # Update conversation history
+        updated_history = list(chat_input.conversation_history)
+        updated_history.append(ChatMessage(role="user", content=chat_input.message))
+        updated_history.append(ChatMessage(role="assistant", content=response_content))
+        
+        return ChatOutput(
+            response=response_content,
+            tool_used="search_products",
+            conversation_history=updated_history
+        )
         
-        logger.info(f"🚀 Calling agent API: {agent_url}")
-        logger.info(f"📦 Agent payload: {json.dumps(agent_payload, indent=2)}")
-        
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            response = await client.post(
-                agent_url,
-                json=agent_payload,
-                headers={
-                    "Content-Type": "application/json",
-                    "Accept": "application/json",
-                    "User-Agent": "HF-Space-Chat/1.0"
-                }
-            )
-            
-            logger.info(f"📡 Agent response status: {response.status_code}")
-            
-            if response.status_code == 200:
-                data = response.json()
-                logger.info(f"✅ Agent response received: {json.dumps(data, indent=2)}")
-                
-                # Extract the response from agent
-                if data.get('success'):
-                    agent_response = data.get('response', 'No response provided')
-                    
-                    # Update conversation history
-                    updated_history = list(chat_input.conversation_history)
-                    updated_history.append(ChatMessage(role="user", content=chat_input.message))
-                    updated_history.append(ChatMessage(role="assistant", content=agent_response))
-                    
-                    return ChatOutput(
-                        response=agent_response,
-                        tool_used=data.get('tool_used'),
-                        conversation_history=updated_history
-                    )
-                else:
-                    error_msg = data.get('error', 'Unknown error from agent')
-                    logger.error(f"❌ Agent returned error: {error_msg}")
-                    return ChatOutput(
-                        response=f"Sorry, I encountered an error: {error_msg}",
-                        conversation_history=list(chat_input.conversation_history)
-                    )
-            else:
-                logger.error(f"❌ Agent API call failed: {response.status_code} - {response.text}")
-                return ChatOutput(
-                    response=f"Sorry, I'm unable to connect to the AI agent (HTTP {response.status_code})",
-                    conversation_history=list(chat_input.conversation_history)
-                )
-                
     except Exception as e:
-        logger.error(f"💥 Error calling agent API: {e}")
-        import traceback
+        logger.error(f"❌ Chat endpoint error: {e}")
+        return ChatOutput(
+            response=f"Sorry, I encountered an error: {str(e)}",
+            conversation_history=list(chat_input.conversation_history)
+        )
+
+
+# Helper function to identify tool needed based on message
         logger.error(f"🔬 Full traceback: {traceback.format_exc()}")
         
         return ChatOutput(