Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 from langchain_google_genai import ChatGoogleGenerativeAI
-from tools import InternetSearchTool, WebscraperTool, ExcelTool, WikipediaTool
+from tools import InternetSearchTool, WebscraperTool, ExcelTool, WikipediaTool, MediaTool
 from smolagents import CodeAgent, InferenceClientModel
 
 # (Keep Constants as is)
@@ -23,12 +23,12 @@ class BasicAgent:
         search_tool = InternetSearchTool()
         webscraper_tool = WebscraperTool()
         excel_tool = ExcelTool()
-        python_tool = PythonTool()
+        media_tool = MediaTool()
         wikipedia_tool = WikipediaTool()
 
         # Create agent
         self.agent= CodeAgent(
-            tools=[search_tool, webscraper_tool, excel_tool, python_tool, wikipedia_tool],
+            tools=[search_tool, webscraper_tool, excel_tool, media_tool, wikipedia_tool],
             model=InferenceClientModel(token=os.getenv("HF_TOKEN"), provider="together"),
             add_base_tools=True,  # Add any additional base tools
             planning_interval=3,   # Enable planning every 3 steps
@@ -49,7 +49,7 @@ class BasicAgent:
         return summary
     
     def __call__(self, question: str) -> str:
-        prompt = """You are a general AI assistant. I will ask you a question. 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.
+        prompt = """You are a general AI assistant. I will ask you a question. 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.
         
         QUESTION:
         """