Update app.py

app.py

@@ -34,38 +34,34 @@ MOON_PHASES = [
     ("Waning Crescent", 315, 360),
 ]
 
-
 @tool
 def get_moon_info(date_time: str) -> dict:
     """
     Returns Moon's Zodiac position with degrees/minutes and phase for given date/time.
 
     Args:
-        date_time (str): ISO 8601 format (YYYY-MM-DDTHH:MM:SS)
+        date_time (str): ISO 8601 formatted datetime (YYYY-MM-DDTHH:MM:SS)
 
     Returns:
         dict: {"zodiac_position": "Leo 15°30'",
                "moon_phase": "Waxing Gibbous"}
     """
     try:
-        # Parse input time
         user_time = datetime.datetime.strptime(date_time, "%Y-%m-%dT%H:%M:%S")
         user_time = pytz.utc.localize(user_time)
 
-        # Load astronomical data
         planets = load('de421.bsp')
         ts = load.timescale()
         t = ts.from_datetime(user_time)
 
-        # Get celestial bodies
         earth = planets['earth']
         moon = planets['moon']
         sun = planets['sun']
 
-        # Calculate geocentric ecliptic longitude
+        # Calculate ecliptic longitude for geocentric position
         astrometric = earth.at(t).observe(moon)
-        _, _, ecliptic_lon = astrometric.ecliptic_latlon()  # Use ecliptic longitude
-        lon_deg = ecliptic_lon.degrees % 360  # Normalize to 0-360°
+        _, _, ecliptic_lon = astrometric.ecliptic_latlon()
+        lon_deg = ecliptic_lon.degrees % 360
 
         # Calculate phase angle
         phase_angle = earth.at(t).observe(moon).apparent().phase_angle(sun).degrees
@@ -79,7 +75,7 @@ def get_moon_info(date_time: str) -> dict:
                 position_degrees = lon_deg - start
                 break
 
-        # Convert to degrees/minutes
+        # Format position to degrees/minutes
         degrees = int(position_degrees)
         minutes = int((position_degrees % 1) * 60)
         position_str = f"{zodiac_sign} {degrees}°{minutes:02}'"
@@ -99,26 +95,43 @@ def get_moon_info(date_time: str) -> dict:
     except Exception as e:
         return {"error": f"Error: {str(e)}"}
 
-# Existing timezone tool remains unchanged
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
-    Returns the current local time in the specified timezone.
+    Returns the current local time in the specified timezone with description.
+
+    Args:
+        timezone (str): A string representing a valid timezone (e.g., 'UTC')
+
+    Returns:
+        str: Formatted local time with timezone description
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        now = datetime.datetime.now(tz)
+        return f"Local time in {timezone}: {now.strftime('%Y-%m-%d %H:%M:%S')}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+@tool
+def get_current_time_raw(timezone: str) -> str:
+    """
+    Returns current local time in specified timezone as ISO 8601 string.
 
     Args:
-        timezone (str): A string representing a valid timezone (e.g., 'America/New_York').
+        timezone (str): A string representing a valid timezone (e.g., 'UTC')
 
     Returns:
-        str: Formatted local time string.
+        str: Datetime in ISO 8601 format (YYYY-MM-DDTHH:MM:SS)
     """
     try:
         tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"Local time in {timezone}: {local_time}"
+        now = datetime.datetime.now(tz)
+        return now.strftime("%Y-%m-%dT%H:%M:%S")
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Model configuration as specified
+# Model configuration
 final_answer = FinalAnswerTool()
 model = HfApiModel(
     max_tokens=2096,
@@ -127,15 +140,17 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Setup agent with updated tools
+# Load image tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+# Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
+# Initialize agent with all tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_moon_info, get_current_time_in_timezone, image_generation_tool],
+    tools=[final_answer, get_moon_info, get_current_time_in_timezone, get_current_time_raw, image_generation_tool],
     max_steps=6,
     verbosity_level=1,
     prompt_templates=prompt_templates