Update app.py

app.py

@@ -18,6 +18,44 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+def ai_fortune_teller(category: str) -> str:
+    """Predicts the future with extreme sarcasm and humor.
+    
+    Args:
+        category: The category to predict ('career', 'love', 'finance').
+    
+    Returns:
+        str: A sarcastic and funny prediction.
+    """
+    fortunes = {
+        'career': [
+            "Your dream job? Oh, you mean winning the lottery and never working again? Keep dreaming. 😂",
+            "You're destined for greatness... if scrolling social media for 5 hours a day counts as a skill.",
+            "One day, you will lead a company. That company may just be your own failed startup, but hey, leadership is leadership. 👏",
+            "Your boss will finally appreciate you! Just kidding. They’ll steal your ideas and take all the credit. Classic. 🤡",
+            "A new job opportunity is coming your way. Too bad it requires 15 years of experience for an entry-level position. 🤦‍♂️"
+        ],
+        'love': [
+            "What is love? I was hoping you’d tell me. Go ask your mom, she might know. 😂",
+            "You will meet the love of your life... and they will promptly ghost you after three messages. 💔",
+            "A grand romantic gesture is in your future! Unfortunately, it’s from someone you find incredibly annoying. 🤷",
+            "Your soulmate is out there... probably swiping left on your profile as we speak. 😬",
+            "True love is coming. It might just be your delivery guy, but hey, a connection is a connection. 🚀"
+        ],
+        'finance': [
+            "Your financial future is bright! Unless, of course, you continue buying overpriced coffee every day. ☕💸",
+            "You will soon receive a large sum of money. Oh wait, I misread. You will soon receive a large sum of bills. Oops. 🤡",
+            "Invest in crypto! Or don’t. Either way, you’ll still be confused about how it actually works. 🧐",
+            "You are about to make a life-changing financial decision... by finally unsubscribing from that gym membership you never use. 👏",
+            "A golden financial opportunity is coming. Unfortunately, it’s an MLM pitch from your cousin. 🚩"
+        ]
+    }
+
+    if category in fortunes:
+        return random.choice(fortunes[category])
+    else:
+        return "Oh, you want a special prediction? Sure. I predict you’ll ignore this advice and continue making questionable life choices. 😂"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +93,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
+    tools=[ai_fortune_teller,get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,