Update app.py

app.py

@@ -1,61 +1,72 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, HfApiModel, load_tool, tool
 import datetime
-import requests
 import pytz
 import yaml
+import random
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Example custom tool: a magic placeholder for creative ideas!
 @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 "What magic will you build?"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    
+def time_traveling_fortune_teller(timezone: str) -> str:
+    """A fun tool that tells the current time in a given timezone and provides a random fortune.
     Args:
         timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     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}"
+        current_time = datetime.datetime.now(tz).strftime('%Y-%m-%d %H:%M:%S')
+        current_hour = datetime.datetime.now(tz).hour
+        
+        # Fun fortunes based on time of day
+        morning_fortunes = [
+            "☀️ Rise and shine! Today is a blank canvas, so go paint your masterpiece!",
+            "🌅 A surprise awaits you this morning… maybe it's coffee, maybe it's chaos. Who knows?",
+            "🍩 Today might be sweet! Maybe a donut is in your future? (Or at least a snack!)"
+        ]
+        
+        afternoon_fortunes = [
+            "🌞 Power through! Your afternoon holds the promise of success (or at least a good nap).",
+            "🍕 Lunchtime decisions determine your fate—choose wisely!",
+            "🚀 Something unexpected will happen soon. Could be good, could be hilarious!"
+        ]
+        
+        evening_fortunes = [
+            "🌙 The stars say… it's time to relax! Or binge-watch your favorite show.",
+            "🔥 A late-night snack might be calling your name. The universe approves.",
+            "🌌 Your dreams tonight will hold a secret message. If you remember them!"
+        ]
+        
+        # Choose fortune based on time of day
+        if 5 <= current_hour < 12:
+            fortune = random.choice(morning_fortunes)
+        elif 12 <= current_hour < 18:
+            fortune = random.choice(afternoon_fortunes)
+        else:
+            fortune = random.choice(evening_fortunes)
+        
+        return f"⏳ The local time in {timezone} is {current_time}.\n🔮 Your fortune: {fortune}"
+    
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"⚠️ Oops! The time-traveling fortune teller got lost in another dimension. Error: {str(e)}"
 
 final_answer = FinalAnswerTool()
-
-# Using a free model (e.g., 'gpt2') instead of a paid one.
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='gpt2',  # Free/open-source model; change as needed.
-    custom_role_conversions=None,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
 )
 
-# Import an image generation tool from the Hub as an example.
+# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool, get_current_time_in_timezone, DuckDuckGoSearchTool(), image_generation_tool],
+    tools=[final_answer, time_traveling_fortune_teller],  
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
     prompt_templates=prompt_templates
 )