Update app.py

app.py

@@ -1,34 +1,33 @@
 import os
 import datetime
-import requests
 import pytz
 import yaml
-from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 
-# --- STEP 1: CREATE NECESSARY FILES (For Colab/VS Code safety) ---
-# This creates a basic prompts.yaml if it doesn't exist
+# --- STEP 1: CONFIGURATION & MOCK FILES ---
+# This creates a prompts.yaml so the script doesn't crash
 prompt_content = {
-    "system_prompt": "You are a helpful agent that can use tools to answer questions. Solve the task step by step."
+    "system_prompt": "You are a helpful AI agent. Answer the question using tools if necessary."
 }
 with open("prompts.yaml", "w") as f:
     yaml.dump(prompt_content, f)
 
-# --- STEP 2: DEFINE TOOLS ---
+# --- STEP 2: DEFINE YOUR TOOLS ---
 
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that returns a creative message combining a string and an integer.
+    """A tool that does nothing yet 
     Args:
-        arg1: A descriptive string or name
-        arg2: A lucky number or quantity
+        arg1: the first argument
+        arg2: the second argument
     """
-    return f"Magic is happening! You provided '{arg1}' and the number {arg2}. Your agent is ready!"
+    return f"Magic confirmed with {arg1} and {arg2}!"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York' or 'Europe/London').
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
         tz = pytz.timezone(timezone)
@@ -37,43 +36,42 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# --- STEP 3: SETUP MODEL AND AGENT ---
+# --- STEP 3: INITIALIZE MODEL & AGENT ---
 
-# If you have an HF_TOKEN in your env variables, it will be picked up automatically
-model = InferenceClientModel(
+# Use HfApiModel (The updated name for InferenceClientModel)
+model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 )
 
-# Load external tools
+# Load the image tool from the Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-search_tool = DuckDuckGoSearchTool()
 
-# Load system prompt from the file we created
+# Load the prompt template
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the Agent with ALL tools included
+# Create the Agent
 agent = CodeAgent(
     model=model,
     tools=[
         get_current_time_in_timezone, 
         my_custom_tool, 
         image_generation_tool, 
-        search_tool
+        DuckDuckGoSearchTool()
     ], 
     max_steps=6,
     verbosity_level=1,
     prompt_templates=prompt_templates
 )
 
-# --- STEP 4: RUN THE AGENT ---
+# --- STEP 4: RUNNING THE AGENT ---
 
-# In VS Code/Local: This will open a browser window
-# In Colab: This will show the UI in the cell output
-# Note: Since Gradio_UI is a custom local file in the course, 
-# we use the standard agent.run() or a basic Gradio interface if that file is missing.
+# If you are in the Hugging Face Space/Course environment, use:
+# from Gradio_UI import GradioUI
+# GradioUI(agent).launch()
 
-print("Agent is initialized. You can now ask questions like: 'What time is it in Tokyo?'")
-agent.run("What is the current time in Paris?")
+# If you are testing in a standard Python script/Colab:
+print("--- Starting Agent Test ---")
+agent.run("What is the current time in New York and then create an image of a futuristic clock.")