Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import pytz
 import yaml
@@ -6,70 +6,38 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 
-# Example of a creative custom tool
-@tool
-def random_welcome_message(name: str) -> str:
-    """Greet a user with a random, cheerful message.
-    Args:
-        name: The name of the user to greet.
-    """
-    import random
-    messages = [
-        f"Hey {name}, hope your day’s fantastic!",
-        f"Welcome aboard, {name}! Ready to explore?",
-        f"Good to see you, {name}! Let's create some magic!"
-    ]
-    return random.choice(messages)
-
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetch the current time in a specified timezone.
-    Args:
-        timezone: Example: 'Asia/Kolkata' or 'America/New_York'
-    """
+    """Return the current time in the specified timezone."""
     try:
         tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
+        time_now = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {time_now}"
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {e}"
+        return f"Error: {e}"
 
 
-# Initialize the tools
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Configure model (you can swap this for any inference endpoint)
-model = InferenceClientModel(
+# Replaced InferenceClientModel → HfApiModel
+model = HfApiModel(
     max_tokens=2048,
     temperature=0.7,
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
 )
 
-# Load agent prompt template file
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
+with open("prompts.yaml", "r") as f:
+    prompt_templates = yaml.safe_load(f)
 
-# Build the multi-tool agent
 agent = CodeAgent(
     model=model,
-    tools=[
-        final_answer,
-        search_tool,
-        image_generation_tool,
-        get_current_time_in_timezone,
-        random_welcome_message
-    ],
+    tools=[final_answer, search_tool, image_generation_tool, get_current_time_in_timezone],
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name="WebSearchAssistant",
-    description="Multi-modal assistant with search, chatbot, and image generation",
-    prompt_templates=prompt_templates
+    description="A multi-tool web chatbot and image generator",
+    prompt_templates=prompt_templates,
 )
 
-# Launch an interactive Gradio UI
 GradioUI(agent).launch()