Update app.py

app.py

@@ -4,19 +4,53 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+from skyfield.api import load, Topos
+from skyfield.almanac import find_discrete, moon_phase_fraction
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity!
+# Define Zodiac signs and their boundaries
+ZODIAC_SIGNS = [
+    ("Aries", 0, 30),
+    ("Taurus", 30, 60),
+    ("Gemini", 60, 90),
+    ("Cancer", 90, 120),
+    ("Leo", 120, 150),
+    ("Virgo", 150, 180),
+    ("Libra", 180, 210),
+    ("Scorpio", 210, 240),
+    ("Sagittarius", 240, 270),
+    ("Capricorn", 270, 300),
+    ("Aquarius", 300, 330),
+    ("Pisces", 330, 360),
+]
+
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: # it's important to specify the return type
-    # Keep this format for the tool description / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
+def get_moon_position_in_zodiac() -> str:
+    """A tool that calculates the current position of the Moon in the Zodiac.
     """
-    return "What magic will you build ?"
+    # Load astronomical data
+    planets = load('de421.bsp')  # Downloaded from JPL Horizons
+    ts = load.timescale()
+
+    # Get the current time in UTC
+    now = datetime.datetime.now(pytz.utc)
+    t = ts.from_datetime(now)
+
+    # 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
+
+    # 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}°)"
+
+    return "Unknown"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -36,10 +70,10 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
 )
 
 
@@ -51,7 +85,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, get_moon_position_in_zodiac],  # Add the new tool here
     max_steps=6,
     verbosity_level=1,
     grammar=None,