Update app.py

app.py

@@ -19,7 +19,6 @@ from scripts.text_web_browser import (
     VisitTool,
 )
 from scripts.visual_qa import visualizer
-from scripts.text_cleaner_tool import TextCleanerTool
 
 from smolagents import (
     CodeAgent,
@@ -170,15 +169,6 @@ class ToolRegistry:
             print(f"✗ Couldn't initialize image generation tool: {e}")
             raise
 
-    @staticmethod
-    def load_clean_text_tool():
-        """Initialize and return image generation tools."""
-        try:
-            return TextCleanerTool
-        except Exception as e:
-            print(f"✗ Couldn't initialize clean text tool: {e}")
-            raise
-
 
 # ------------------------ Agent Creation and Execution ------------------------
 def create_agent():
@@ -186,7 +176,7 @@ def create_agent():
     # Initialize model
     model = LiteLLMModel(
         custom_role_conversions=custom_role_conversions,
-        model_id="openrouter/deepseek/deepseek-chat-v3-0324:free",  # currently serving:
+        model_id="openrouter/google/gemini-2.0-flash-001",  # currently serving:
     )  # DEEPSEEK = openrouter/perplexity/r1-1776 <--- boss model
 
     # Initialize tools
@@ -196,10 +186,9 @@ def create_agent():
     # Collect all tools in a single list
     web_tools = ToolRegistry.load_web_tools(model, browser, text_limit)
     image_generator = ToolRegistry.load_image_generation_tools()
-    clean_text = ToolRegistry.load_clean_text_tool()
 
     # Combine all tools into a single list (not a tuple)
-    all_tools = [visualizer, web_tools, image_generator, clean_text]
+    all_tools = [visualizer] + web_tools + [image_generator]
 
     # Validate tools before creating agent
     for tool in all_tools:
@@ -420,7 +409,7 @@ class GradioUI:
             with gr.Sidebar():
                 gr.Markdown(
                     """#OpenDeepResearch - 3theSmolagents!
-                Model_id: deepseek/deepseek-chat-v3-0324:free"""
+                Model_id: google/gemini-2.0-flash-001"""
                 )
                 with gr.Group():
                     gr.Markdown("**What's on your mind mate?**", container=True)