Update app.py

app.py

@@ -4,7 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-import numpy as np
+import numpy as np$
+from typing import Tuple
 
 from Gradio_UI import GradioUI
 
@@ -45,6 +46,38 @@ def compute_vacuum_energy(L: float, N_max: int, hbar: float=1.0545718e-34, c: fl
 
     return energy
 
+@tool
+def compute_universe_color(age_in_gyr: float) -> Tuple[int, int, int]:
+    """
+    Estimate the average color of the universe based on its age.
+
+    Parameters:
+        age_in_gyr (float): Age of the universe in billions of years (Gyr).
+
+    Returns:
+        Tuple[int, int, int]: RGB color representing the average color of the universe.
+    """
+
+    # Clamp age to a plausible range (0 - 20 Gyr)
+    age = max(0.0, min(age_in_gyr, 20.0))
+
+    # Approximate color evolution over time in RGB
+    # 0 Gyr → blue-white (hot stars), 13.8 Gyr → cosmic latte (pale beige), 20 Gyr → reddish
+    if age < 1.0:
+        color = (180, 210, 255)  # bright blue-white
+    elif age < 5.0:
+        color = (240, 230, 255)  # softer white
+    elif age < 9.0:
+        color = (255, 240, 220)  # yellowish
+    elif age < 13.0:
+        color = (255, 230, 210)  # pale beige
+    elif age <= 14.0:
+        color = (255, 223, 196)  # "cosmic latte"
+    else:
+        color = (240, 200, 180)  # redder with star aging
+
+    return color
+
 def convert_energy(value: float, current_unit: str, target_unit: str) -> float:
     """
     Convert an energy value from one unit to another.
@@ -131,7 +164,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_age_univers, get_current_time_in_timezone, convert_energy, compute_vacuum_energy], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_age_univers, get_current_time_in_timezone, convert_energy, compute_vacuum_energy, compute_universe_color, DuckDuckGoSearchTool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,