Update app.py

app.py

@@ -19,6 +19,7 @@ from scripts.text_web_browser import (
     VisitTool,
 )
 from scripts.visual_qa import visualizer
+from scripts.text_cleaner import TextCleanerTool
 
 from smolagents import (
     CodeAgent,
@@ -169,6 +170,15 @@ 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,9 +196,10 @@ 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]
+    all_tools = [visualizer] + web_tools + [image_generator] + clean_text
 
     # Validate tools before creating agent
     for tool in all_tools: