update app.py

app.py

@@ -4,24 +4,25 @@ import pytz
 import yaml
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 
-# --- STEP 1: CONFIGURATION & MOCK FILES ---
-# This creates a prompts.yaml so the script doesn't crash
+# --- STEP 1: SECURITY & CONFIGURATION ---
+# Replace with your NEW token. 
+# Best practice: Set this in your system environment variables as HF_TOKEN
+os.environ["HF_TOKEN"] = "PASTE_YOUR_NEW_TOKEN_HERE"
+
+# Creating a proper prompts.yaml to guide the agent's logic
 prompt_content = {
-    "system_prompt": "You are a helpful AI agent. Answer the question using tools if necessary."
+    "system_prompt": (
+        "You are an expert AI assistant that can use python code to solve tasks. "
+        "You have access to tools. For every step, explain your thought process in 'Thought:', "
+        "then write the code in a 'Code:' block. Once you have the final answer, "
+        "provide it clearly."
+    )
 }
+
 with open("prompts.yaml", "w") as f:
     yaml.dump(prompt_content, f)
 
-# --- STEP 2: DEFINE YOUR TOOLS ---
-
-@tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return f"Magic confirmed with {arg1} and {arg2}!"
+# --- STEP 2: DEFINE TOOLS ---
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -30,48 +31,72 @@ def get_current_time_in_timezone(timezone: str) -> str:
         timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
+        # Normalize common inputs
+        if timezone.lower() == "new york":
+            timezone = "America/New_York"
+        elif timezone.lower() == "london":
+            timezone = "Europe/London"
+            
         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}"
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"Error: '{timezone}' is not a recognized timezone. Please use format like 'Europe/Paris'."
 
-# --- STEP 3: INITIALIZE MODEL & AGENT ---
+@tool
+def my_custom_tool(arg1: str, arg2: int) -> str:
+    """A placeholder tool for future custom logic. 
+    Args:
+        arg1: A descriptive string.
+        arg2: An integer multiplier or count.
+    """
+    return f"Logic processed: {arg1} received with value {arg2}."
+
+# --- STEP 3: INITIALIZE MODEL & TOOLS ---
 
-# Use HfApiModel (The updated name for InferenceClientModel)
+# Use a highly available model. If Qwen 32B gives a 403, try 'meta-llama/Llama-3.3-70B-Instruct'
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct', 
 )
 
-# Load the image tool from the Hub
+# Load the image generation tool
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Load the prompt template
+# Load the prompt template from the file we just created
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the Agent
+# --- STEP 4: INITIALIZE AGENT ---
+
 agent = CodeAgent(
     model=model,
     tools=[
         get_current_time_in_timezone, 
-        my_custom_tool, 
-        image_generation_tool, 
-        DuckDuckGoSearchTool()
+        DuckDuckGoSearchTool(), 
+        image_generation_tool,
+        my_custom_tool
     ], 
     max_steps=6,
     verbosity_level=1,
     prompt_templates=prompt_templates
 )
 
-# --- STEP 4: RUNNING THE AGENT ---
+# --- STEP 5: EXECUTION ---
 
-# If you are in the Hugging Face Space/Course environment, use:
-# from Gradio_UI import GradioUI
-# GradioUI(agent).launch()
+if __name__ == "__main__":
+    print("--- Starting Agent Test ---")
+    
+    task = "What is the current time in New York? After you find the time, generate a beautiful image of a clock that reflects the style of that city."
+    
+    try:
+        result = agent.run(task)
+        print("\n--- FINAL ANSWER ---")
+        print(result)
+    except Exception as e:
+        print(f"\nAn error occurred during execution: {e}")
 
-# 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.")
+    # Optional: If you want to launch a UI (requires 'gradio' installed)
+    # from Gradio_UI import GradioUI
+    # GradioUI(agent).launch()