Update app.py

app.py

@@ -13,7 +13,11 @@ from langchain_core.tools import tool
 from langchain_core.messages import AIMessage, ToolMessage, HumanMessage, BaseMessage, SystemMessage
 from random import randint
 
+import requests
+from bs4 import BeautifulSoup
+import openpyxl
 import wikipedia
+import pandas as pd
 
 import gradio as gr
 import logging
@@ -24,27 +28,58 @@ class OrderState(TypedDict):
     order: list[str]
     finished: bool
 
-# System instruction for the BaristaBot
+# System instruction for the Agent
 SYSINT = (
     "system",
-    "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: "
+    "You are a general AI assistant. I will ask you a question."
+    "The question requires a tool to solve. You must attempt to use at least one of the available tools before returning an answer."
+    "Report your thoughts, and finish your answer with the following template: "
     "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings."
     "If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise."
     "If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise."
     "If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
-    "If a tool required for task completion is unavailable after multiple tries, return 0."
+    "If a tool required for task completion is not functioning, return 0."
     )
 
-WELCOME_MSG = "Welcome to the BaristaBot cafe. Type `q` to quit. How may I serve you today?"
+WELCOME_MSG = "Welcome to my general-purpose AI agent. Type `q` to quit. How shall I fail to serve you today?"
 
 # Initialize the Google Gemini LLM
 llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
 
 @tool
-def wikipedia_search(title: str) -> str:
-    """Provides a short snippet from a Wikipedia article with the given title"""
+def wikipedia_search_tool(title: str) -> str:
+    """Provides an excerpt from a Wikipedia article with the given title."""
     page = wikipedia.page(title, auto_suggest=False)
-    return page.content[:300]
+    return page.content[:3000]
+
+@tool
+def media_tool(file_path: str) -> str:
+    """Used for deciphering video and audio files."""
+    return "This tool hasn't been implemented yet. Please return 0 if the task cannot be solved without knowing the contents of this file."
+
+@tool
+def internet_search_tool(search_query: str) -> str:
+    """Does a google search with using the input as the search query. Returns a long batch of textual information related to the query."""
+    search_tool = DuckDuckGoSearchTool()
+    result = search_tool(question)
+    return result
+
+@tool
+def webscraper_tool(url: str) -> str:
+    """Returns the page's html content from the input url."""
+    response = requests.get(url, stream=True)
+    if response.status_code == 200:
+        soup = BeautifulSoup(response.content, 'html.parser')
+        html_text = soup.get_text()
+        return html_text
+    else:
+        raise Exception(f"Failed to retrieve the webpage. Status code: {response.status_code}")
+
+@tool
+def read_excel_tool(file_path: str) -> str:
+    """Returns the contents of an Excel file as a Pandas dataframe."""
+    df = pd.read_excel(path, engine = "openpyxl")
+    return df
 
 def agent_node(state: OrderState) -> OrderState:
     """agent with tool handling."""
@@ -80,10 +115,27 @@ def interactive_tools_node(state: OrderState) -> OrderState:
         tool_name = tool_call["name"]
         tool_args = tool_call["args"]
 
-        if tool_name == "wikipedia_search":
+        if tool_name == "wikipedia_search_tool":
             print(str(tool_args))
             page = wikipedia.page(tool_args.get("title"), auto_suggest=False)
-            response = page.content[:300]
+            response = page.content[:3000]
+        elif tool_name == "media_tool":
+            print(str(tool_args))
+            response = "This tool hasn't been implemented yet. Please return 0 if the task cannot be solved without knowing the contents of this file."
+        elif tool_name == "internet_search_tool":
+            search_tool = DuckDuckGoSearchTool()
+            response = search_tool(question)[:3000]
+        elif tool_name == "webscraper_tool":
+            response = requests.get(url, stream=True)
+            if response.status_code == 200:
+                soup = BeautifulSoup(response.content, 'html.parser')
+                html_text = soup.get_text()
+                response = html_text
+            else:
+                response = Exception(f"Failed to retrieve the webpage. Status code: {response.status_code}")
+        elif tool_name == "read_excel_tool":
+            df = pd.read_excel(path, engine = "openpyxl")
+            response = df
 
         else:
             raise NotImplementedError(f'Unknown tool call: {tool_name}')
@@ -239,12 +291,12 @@ def gradio_chat(message: str, history: list) -> str:
         return f"An error occurred: {str(e)}"
 
 # Gradio interface
-def launch_baristabot():
+def launch_agent():
     gr.ChatInterface(
         gradio_chat,
         type="messages",
-        title="BaristaBot",
-        description="Your friendly AI cafe assistant",
+        title="Agent",
+        description="An AI agent (work in progress)",
         theme="ocean"
     ).launch()
 
@@ -254,4 +306,4 @@ if __name__ == "__main__":
                     stream=sys.stdout,
                     level=logging.INFO,
                     format='%(asctime)s - %(levelname)s - %(message)s')
-    launch_baristabot()
+    launch_agent()