Update app.py

app.py

@@ -6,6 +6,14 @@ import pandas as pd
 import time
 from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, GoogleSearchTool, WikipediaSearchTool, VisitWebpageTool, PythonInterpreterTool
 from smolagents.agents import PromptTemplates
+from tool import (save_and_read_file,
+                 download_file_from_url,
+                 extract_text_from_image,
+                 analyze_csv_file, 
+                 analyze_excel_file,
+                 extract_video_id,
+                 get_youtube_transcript)
+
 
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
@@ -24,9 +32,20 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         self.agent = CodeAgent(
-            tools = [GoogleSearchTool(), WikipediaSearchTool(), VisitWebpageTool(), PythonInterpreterTool()],
+            tools = [GoogleSearchTool(), 
+                     WikipediaSearchTool(), 
+                     VisitWebpageTool(), 
+                     PythonInterpreterTool(), 
+                     save_and_read_file,
+                     download_file_from_url,
+                     extract_text_from_image,
+                     analyze_csv_file, 
+                     analyze_excel_file,
+                     extract_video_id,
+                     get_youtube_transcript],
             model = model,
             add_base_tools=True,
+            additional_authorized_imports=["os","tempfile","requests", "urllib", "pytesseract", "PIL", "pandas", "uuid", "youtube_transcript_api"]
         )
 
         print("BasicAgent initialized.")