Update app.py

app.py

@@ -6,7 +6,7 @@ import requests
 import pandas as pd
 import base64
 import librosa
-import openpyxl
+import chess
 
 from typing import TypedDict, Annotated
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
@@ -65,7 +65,7 @@ def search_web(query: str) -> str:
     """Retrieves information about the topic."""
     results = DuckDuckGoSearchRun().invoke(query)
     if results:
-        return "\n\n".join([doc.text for doc in results[:1]])
+        return "\n\n".join([doc.text for doc in results[:2]])
     else:
         return "No matching content found."
 
@@ -85,6 +85,8 @@ def extract_text_from_image(img_path: str) -> str:
         print(error_msg)
         return ""
 
+def play_chess()
+    return chess.Board()
 
 # Initialize the tool
 get_file_path_tool = Tool(
@@ -111,7 +113,13 @@ extract_text_tool = Tool(
     description="Retrieves text from an image."
 )
 
-tools = [get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool]
+play_chess_tool = Tool(
+    name="chess_board_retriever",
+    func=play_chess,
+    description="Sets a chess board."
+)
+
+tools = [get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool, play_chess_tool]
 chat_with_tools = chat.bind_tools(tools, parallel_tool_calls=True)
 
 
@@ -131,7 +139,7 @@ builder = StateGraph(AgentState)
 
 # Define nodes: these do the work
 builder.add_node("assistant", assistant)
-builder.add_node("tools", ToolNode([get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool]))
+builder.add_node("tools", ToolNode([get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool, play_chess_tool]))
 
 # Define edges: these determine how the control flow moves
 builder.add_edge(START, "assistant")