Try add user_input in tools

app.py

@@ -1,9 +1,10 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool,HfApiModel,load_tool,tool,UserInputTool
+from smolagents import CodeAgent,DuckDuckGoSearchTool,HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.user_input import UserInputTool
 
 from kaggle.api.kaggle_api_extended import KaggleApi
 import os
@@ -77,6 +78,7 @@ def download_kaggle_dataset(
     return f"Dataset '{dataset_ref}' downloaded to '{download_path}'."
 
 final_answer = FinalAnswerTool()
+user_input = UserInputTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
@@ -98,7 +100,7 @@ agent = CodeAgent(
     model=model,
     tools=[final_answer,
            search_kaggle_datasets, 
-           UserInputTool(),
+           user_input,
            download_kaggle_dataset,
            image_generation_tool],
     max_steps=6,

tools/user_input.py

@@ -0,0 +1,15 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+
+class UserInputTool(Tool):
+    name = "user_input"
+    description = "Asks for user's input on a specific question"
+    inputs = {"question": {"type": "string", "description": "The question to ask the user"}}
+    output_type = "string"
+
+    def forward(self, question):
+        user_input = input(f"{question} => Type your answer here:")
+        return user_input
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False