Update chat_mcp_server.py

chat_mcp_server.py

@@ -1,9 +1,24 @@
-from mcp.server.fastmcp import FastMCP
+import gradio as gr
+from web_scraper import search_your_product
 
-mcp = FastMCP("chatbot")
+
+def product_search(query: str) -> str:
+    """
+    Search for products based on the provided query.
+
+    Args:
+        query (str): The search query for products.
+
+    Returns:
+        str: A message with the search results.
+    """
+
+    results = search_your_product(query)
+    if not results:
+        return f"No products found for your search: '{query}'. Please try a different query."
+    return f"Here are the products matching your search for '{query}': {results}."
 
 
-@mcp.tool()
 def product_recommendation(product_name: str) -> str:
     """
     Recommend a product based on the provided product name.
@@ -17,8 +32,7 @@ def product_recommendation(product_name: str) -> str:
     return f"I recommend you try {product_name}."
 
 
-@mcp.tool()
-def cart_managemnt(action: str, product_name: str) -> str:
+def cart_management(action: str, product_name: str) -> str:
     """
     Manage the shopping cart by adding or removing products.
 
@@ -37,7 +51,6 @@ def cart_managemnt(action: str, product_name: str) -> str:
         return "Invalid action. Please specify 'add' or 'remove'."
 
 
-@mcp.tool()
 def abondoned_cart_recovery() -> str:
     """
     Recover abandoned carts by sending a reminder to the user.
@@ -48,7 +61,6 @@ def abondoned_cart_recovery() -> str:
     return "Don't forget to complete your purchase! Your cart is waiting for you."
 
 
-@mcp.tool()
 def order_tracking(order_id: str) -> str:
     """
     Track an order based on the provided order ID.
@@ -62,7 +74,6 @@ def order_tracking(order_id: str) -> str:
     return f"Your order {order_id} is currently being processed and will be shipped soon."
 
 
-@mcp.tool()
 def checkout() -> str:
     """
     Perform the checkout process.
@@ -73,7 +84,6 @@ def checkout() -> str:
     return "Checkout completed successfully! Thank you for your purchase."
 
 
-@mcp.tool()
 def customer_support(issue: str) -> str:
     """
     Provide customer support for the specified issue.
@@ -87,27 +97,12 @@ def customer_support(issue: str) -> str:
     return f"Our customer support team is looking into your issue: {issue}. We will get back to you shortly."
 
 
-@mcp.tool()
-def product_search(query: str) -> str:
-    """
-    Search for products based on the provided query.
-
-    Args:
-        query (str): The search query for products.
-
-    Returns:
-        str: A message with the search results.
-    """
-    return f"Here are the products matching your search for '{query}': [Product List]."
-
-
-@mcp.tool()
 def multi_channel_support(channel: str, message: str) -> str:
     """
     Provide support through multiple channels.
 
     Args:
-        channel (str): The support channel (e.g., 'email', 'chat', 'phone').
+        channel (str): The support channel (e.g., 'email', 'chat', 'phone', 'whatsapp').
         message (str): The message or issue to address.
 
     Returns:
@@ -116,7 +111,6 @@ def multi_channel_support(channel: str, message: str) -> str:
     return f"We have received your message via {channel}: '{message}'. Our team will respond shortly."
 
 
-@mcp.tool()
 def personalized_recommendations(user_id: str) -> str:
     """
     Provide personalized product recommendations based on the user's ID.
@@ -130,7 +124,6 @@ def personalized_recommendations(user_id: str) -> str:
     return f"Based on your preferences, we recommend the following products for you, User {user_id}: [Personalized Product List]."
 
 
-@mcp.tool()
 def inventory_management(product_name: str, action: str) -> str:
     """
     Manage inventory by adding or removing products.
@@ -150,7 +143,6 @@ def inventory_management(product_name: str, action: str) -> str:
         return "Invalid action. Please specify 'add' or 'remove'."
 
 
-@mcp.tool()
 def stock_awareness(product_name: str) -> str:
     """
     Check the stock availability of a product.
@@ -163,5 +155,5 @@ def stock_awareness(product_name: str) -> str:
     """
     return f"The current stock status for {product_name} is: [Stock Status]."
 
-    if __name__ == "__main__":
-    mcp.run(transport="stdio")
+
+gr.launch(mcp_server=True)   # exposes an MCP schema automatically