Update app.py

app.py

@@ -126,7 +126,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False, document=
     tools = [
         Tool(
             name="Image",
-            func=ImageGeneration(),
+            func=ImageGeneration(),  # Pass the class instance, not ImageGeneration()._run
             description="Useful for generating images based on text descriptions"
         ),
     ]
@@ -134,7 +134,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False, document=
     # Add the numpy tool, but with a more specific description
     tools.append(Tool(
         name="Calculator",
-        func=NumpyCodeCalculator(),
+        func=NumpyCodeCalculator(),  # Pass the class instance, not NumpyCodeCalculator()._run
         description="Useful only for performing numerical computations, not for general searches"
     ))
 
@@ -142,7 +142,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False, document=
     if websearch:
         tools.append(Tool(
             name="Web",
-            func=WebSearch(),
+            func=WebSearch(),  # Pass the class instance, not WebSearch()._run
             description="Useful for advanced web searching beyond general information"
         ))
 
@@ -150,7 +150,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False, document=
     if document:
         tools.append(Tool(
             name="Document",
-            func=DocumentQuestionAnswering(document),
+            func=DocumentQuestionAnswering(document),  # This is already correct
             description="Useful for answering questions about a specific document"
         ))