Update app.py

app.py

@@ -1,26 +1,14 @@
-import argparse
 import json
 import mimetypes  # Added missing import
 import os
 import re  # Added missing import
 import shutil  # Added missing import
-import threading
-from concurrent.futures import ThreadPoolExecutor, as_completed
-from datetime import datetime
-from pathlib import Path
 from typing import Dict, List, Optional, Any
 
-import datasets
-import pandas as pd
 from dotenv import load_dotenv
 from huggingface_hub import login
 import gradio as gr
 
-from scripts.reformulator import prepare_response
-from scripts.run_agents import (
-    get_single_file_description,
-    get_zip_description,
-)
 from scripts.text_inspector_tool import TextInspectorTool
 from scripts.text_web_browser import (
     ArchiveSearchTool,
@@ -34,16 +22,13 @@ from scripts.text_web_browser import (
 from scripts.visual_qa import visualizer
 
 # from scripts.flux_lora_tool import FluxLoRATool
-from tqdm import tqdm
 
 from smolagents import (
     CodeAgent,
     HfApiModel,
     LiteLLMModel,
-    Model,
     OpenAIServerModel,  # Added missing model
     TransformersModel,  # Added missing model
-    ToolCallingAgent,
     Tool,
 )
 from smolagents.agent_types import AgentText, AgentImage, AgentAudio
@@ -283,7 +268,7 @@ class ToolRegistry:
 
 # Agent creation in a factory function
 def create_agent():
-    """Creates a fresh agent instance for each session."""
+    """Creates a fresh agent instance with properly configured tools."""
     # Initialize model
     model = LiteLLMModel(
         custom_role_conversions=custom_role_conversions,
@@ -294,12 +279,23 @@ def create_agent():
     text_limit = 20000
     browser = SimpleTextBrowser(**BROWSER_CONFIG)
 
+    # Collect all tools in a single list
     web_tools = ToolRegistry.load_web_tools(model, browser, text_limit)
     gemma_vision_tool = ToolRegistry.load_vision_tools()
 
+    # Combine all tools into a single list (not a tuple)
+    all_tools = [visualizer] + web_tools + [gemma_vision_tool]
+
+    # Validate tools before creating agent
+    for tool in all_tools:
+        if not isinstance(tool, Tool):
+            raise ValueError(
+                f"Invalid tool type: {type(tool)}. All tools must be instances of Tool class."
+            )
+
     return CodeAgent(
         model=model,
-        tools=([visualizer] + web_tools, gemma_vision_tool),  # Fixed the missing comma
+        tools=all_tools,  # Pass a single list containing all tools
         max_steps=10,
         verbosity_level=1,
         additional_authorized_imports=AUTHORIZED_IMPORTS,
@@ -536,7 +532,7 @@ class GradioUI:
             # Add session state to store session-specific data
             session_state = gr.State({})  # Initialize empty state for each session
             stored_messages = gr.State([])
-            if not "file_uploads_log" in locals():
+            if "file_uploads_log" not in locals():
                 file_uploads_log = gr.State([])
 
             chatbot = gr.Chatbot(