Update app.py

Added a Zodiac function!

app.py

@@ -9,14 +9,88 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def get_zodiac_info(birth_month:str, birth_year:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+    """
+    A tool that returns your Zodiac sign and Chinese zodiac animal based on birth month and year.
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        birth_month (str): Month of birth (e.g., "February")
+        birth_year (int): Year of birth (e.g., 1988)
+    
+    Returns:
+        str: Information about zodiac sign and Chinese zodiac animal, or error message
     """
-    return "What magic will you build ?"
+    try:
+        # Validate month input
+        if not isinstance(birth_month, str):
+            return "Error: Birth month must be a string (e.g., 'February')."
+            
+        # Convert month to lowercase for case-insensitive comparison
+        birth_month = birth_month.lower()
+        
+        # Define months and their numerical values for easier date comparison
+        month_to_num = {
+            "january": 1, "february": 2, "march": 3, "april": 4,
+            "may": 5, "june": 6, "july": 7, "august": 8,
+            "september": 9, "october": 10, "november": 11, "december": 12
+        }
+        
+        # Get month number
+        month_num = month_to_num.get(birth_month)
+        if month_num is None:
+            return "Error: Invalid month name provided. Please use full month names (e.g., 'February')."
+        
+        # Validate year input
+        if not isinstance(birth_year, int):
+            try:
+                birth_year = int(birth_year)
+            except (ValueError, TypeError):
+                return "Error: Birth year must be a valid integer."
+                
+        if birth_year < 1900 or birth_year > 2100:
+            return "Error: Birth year should be between 1900 and 2100."
+        
+        # Determine Western zodiac sign
+        if (month_num == 1 and 20 <= 31) or (month_num == 2 and 1 <= 18):
+            zodiac_sign = "Aquarius"
+        elif (month_num == 2 and 19 <= 29) or (month_num == 3 and 1 <= 20):
+            zodiac_sign = "Pisces"
+        elif (month_num == 3 and 21 <= 31) or (month_num == 4 and 1 <= 19):
+            zodiac_sign = "Aries"
+        elif (month_num == 4 and 20 <= 30) or (month_num == 5 and 1 <= 20):
+            zodiac_sign = "Taurus"
+        elif (month_num == 5 and 21 <= 31) or (month_num == 6 and 1 <= 20):
+            zodiac_sign = "Gemini"
+        elif (month_num == 6 and 21 <= 30) or (month_num == 7 and 1 <= 22):
+            zodiac_sign = "Cancer"
+        elif (month_num == 7 and 23 <= 31) or (month_num == 8 and 1 <= 22):
+            zodiac_sign = "Leo"
+        elif (month_num == 8 and 23 <= 31) or (month_num == 9 and 1 <= 22):
+            zodiac_sign = "Virgo"
+        elif (month_num == 9 and 23 <= 30) or (month_num == 10 and 1 <= 22):
+            zodiac_sign = "Libra"
+        elif (month_num == 10 and 23 <= 31) or (month_num == 11 and 1 <= 21):
+            zodiac_sign = "Scorpio"
+        elif (month_num == 11 and 22 <= 30) or (month_num == 12 and 1 <= 21):
+            zodiac_sign = "Sagittarius"
+        else:
+            zodiac_sign = "Capricorn"
+        
+        # Determine Chinese zodiac animal
+        # Chinese zodiac repeats every 12 years
+        # Rat, Ox, Tiger, Rabbit, Dragon, Snake, Horse, Goat, Monkey, Rooster, Dog, Pig
+        chinese_zodiac = ["Monkey", "Rooster", "Dog", "Pig", "Rat", "Ox", 
+                          "Tiger", "Rabbit", "Dragon", "Snake", "Horse", "Goat"]
+        
+        # Calculate Chinese zodiac animal
+        zodiac_animal = chinese_zodiac[birth_year % 12]
+        
+        # Return formatted string
+        return f"You're a {zodiac_sign} and you were born in the year of the {zodiac_animal}."
+    
+    except Exception as e:
+        return f"Error: An unexpected error occurred: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +129,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_zodiac_info, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,