Update app.py

app.py

@@ -34,6 +34,93 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+@tool
+def calculate_age(birth_date: str, reference_date: str = "today") -> str:
+    """Calculates precise age in years, months, and days from birth date to reference date.
+    Handles multiple date formats (YYYY-MM-DD, MM/DD/YYYY, DD.MM.YYYY, etc.).
+    
+    Args:
+        birth_date: The birth date in any common format (e.g., "1990-05-15", "05/15/1990", "15.05.1990")
+        reference_date: The date to calculate age at (defaults to today). Use "today" or specific date in same formats.
+    
+    Returns:
+        String describing age in years, months, and days (e.g., "34 years, 2 months, 15 days")
+    """
+
+    from datetime import datetime
+    import re
+    
+    def parse_date(date_str):
+        """Parse various date formats into datetime object"""
+        if date_str.lower() == "today":
+            return datetime.today()
+            
+        # Try different date patterns
+        patterns = [
+            (r"(\d{4})-(\d{1,2})-(\d{1,2})", "%Y-%m-%d"),  # YYYY-MM-DD
+            (r"(\d{1,2})/(\d{1,2})/(\d{4})", "%m/%d/%Y"),    # MM/DD/YYYY
+            (r"(\d{1,2})\.(\d{1,2})\.(\d{4})", "%d.%m.%Y"),  # DD.MM.YYYY
+            (r"(\d{1,2}) (\w{3,}) (\d{4})", "%d %B %Y"),     # 15 May 1990
+            (r"(\w{3,}) (\d{1,2}), (\d{4})", "%B %d, %Y"),   # May 15, 1990
+        ]
+        
+        for pattern, fmt in patterns:
+            if re.fullmatch(pattern, date_str):
+                try:
+                    return datetime.strptime(date_str, fmt)
+                except ValueError:
+                    continue
+                    
+        raise ValueError(f"Could not parse date: {date_str}")
+
+    try:
+        # Parse dates
+        birth_dt = parse_date(birth_date)
+        ref_dt = parse_date(reference_date)
+        
+        # Calculate age
+        if ref_dt < birth_dt:
+            return "Error: Reference date is before birth date"
+            
+        years = ref_dt.year - birth_dt.year
+        months = ref_dt.month - birth_dt.month
+        days = ref_dt.day - birth_dt.day
+        
+        # Handle negative months/days
+        if days < 0:
+            months -= 1
+            # Get days in previous month
+            if ref_dt.month == 1:
+                prev_month = 12
+                prev_year = ref_dt.year - 1
+            else:
+                prev_month = ref_dt.month - 1
+                prev_year = ref_dt.year
+            days_in_prev_month = (datetime(prev_year, prev_month + 1, 1) - datetime(prev_year, prev_month, 1)).days
+            days += days_in_prev_month
+            
+        if months < 0:
+            years -= 1
+            months += 12
+            
+        # Format output
+        parts = []
+        if years > 0:
+            parts.append(f"{years} year{'s' if years != 1 else ''}")
+        if months > 0:
+            parts.append(f"{months} month{'s' if months != 1 else ''}")
+        if days > 0:
+            parts.append(f"{days} day{'s' if days != 1 else ''}")
+            
+        if not parts:
+            return "0 days old (born today)"
+            
+        return ", ".join(parts) + " old"
+        
+    except Exception as e:
+        return f"Error calculating age: {str(e)}. Please check date formats."
+
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -55,7 +142,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=[final_answer, calculate_age], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,