Update app.py

app.py

@@ -1,10 +1,9 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
-import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
 from skyfield.api import load, Topos
+from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # Define Zodiac signs and their boundaries
@@ -23,32 +22,72 @@ ZODIAC_SIGNS = [
     ("Pisces", 330, 360),
 ]
 
+# Define Moon phases
+MOON_PHASES = [
+    ("New Moon", 0, 45),
+    ("Waxing Crescent", 45, 90),
+    ("First Quarter", 90, 135),
+    ("Waxing Gibbous", 135, 180),
+    ("Full Moon", 180, 225),
+    ("Waning Gibbous", 225, 270),
+    ("Last Quarter", 270, 315),
+    ("Waning Crescent", 315, 360),
+]
+
 @tool
-def get_moon_position_in_zodiac() -> str:
-    """A tool that calculates the current position of the Moon in the Zodiac."""
-    # Load astronomical data
-    planets = load('de421.bsp')  # Ensure this file is accessible
-    ts = load.timescale()
+def get_moon_info(date_time: str) -> dict:
+    """
+    A tool that calculates the Moon's Zodiac sign and phase for a given date and time.
+    
+    Args:
+        date_time: A string in ISO 8601 format (YYYY-MM-DDTHH:MM:SS), e.g., '2023-10-05T12:00:00'.
+    
+    Returns:
+        A dictionary containing the Moon's Zodiac sign and phase.
+    """
+    try:
+        # Parse input date and 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')  # Ensure this file is accessible
+        ts = load.timescale()
 
-    # Get the current time in UTC
-    now = datetime.datetime.now(pytz.utc)
-    t = ts.from_datetime(now)
+        # Convert user time to Skyfield timescale
+        t = ts.from_datetime(user_time)
 
-    # Get the geocentric position of the Moon
-    earth = planets['earth']
-    moon = planets['moon']
-    astrometric = earth.at(t).observe(moon)
-    ra, dec, distance = astrometric.radec()
+        # Get the geocentric position of the Moon
+        earth = planets['earth']
+        moon = planets['moon']
+        astrometric = earth.at(t).observe(moon)
+        ra, dec, distance = astrometric.radec()
 
-    # Convert RA to degrees
-    ra_in_degrees = ra._degrees
+        # Convert RA to degrees
+        ra_in_degrees = ra._degrees
 
-    # Determine the Zodiac sign based on RA
-    for sign, start, end in ZODIAC_SIGNS:
-        if start <= ra_in_degrees < end:
-            return f"The Moon is currently in {sign} ({start}° - {end}°)"
+        # Determine the Zodiac sign based on RA
+        zodiac_sign = "Unknown"
+        for sign, start, end in ZODIAC_SIGNS:
+            if start <= ra_in_degrees < end:
+                zodiac_sign = f"{sign} ({start}° - {end}°)"
+                break
 
-    return "Unknown"
+        # Calculate the Moon's phase
+        moon_phase_angle = earth.at(t).observe(moon).apparent().phase_angle().degrees
+        moon_phase = "Unknown"
+        for phase, start, end in MOON_PHASES:
+            if start <= moon_phase_angle < end:
+                moon_phase = phase
+                break
+
+        return {
+            "zodiac_sign": zodiac_sign,
+            "moon_phase": moon_phase,
+        }
+
+    except Exception as e:
+        return {"error": f"Invalid input or calculation error: {str(e)}"}
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -83,7 +122,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_moon_position_in_zodiac],  # Add the updated tool here
+    tools=[final_answer, get_moon_info],  # Add the updated tool here
     max_steps=6,
     verbosity_level=1,
     grammar=None,