Update app.py

app.py

@@ -22,7 +22,7 @@ ZODIAC_SIGNS = [
     ("Pisces", 330, 360),
 ]
 
-# Moon phase boundaries (0° to 360° phase angle)
+# Moon phase boundaries (0° to 360° phase angle) for display purposes
 MOON_PHASES = [
     ("New Moon", 0, 45),
     ("Waxing Crescent", 45, 90),
@@ -34,36 +34,100 @@ MOON_PHASES = [
     ("Waning Crescent", 315, 360),
 ]
 
+# Fertility sign coefficients (applicable to all plants)
+FERTILITY_SIGN_COEFFS = {
+    "Aries": 1,
+    "Taurus": 2,
+    "Gemini": 0,
+    "Cancer": 2,
+    "Leo": 1,
+    "Virgo": 0,
+    "Libra": 0.5,
+    "Scorpio": 1.5,
+    "Sagittarius": 1,
+    "Capricorn": 1,
+    "Aquarius": 0,
+    "Pisces": 2,
+}
+
+# Pruning sign coefficients (applicable to all plants)
+PRUNING_SIGN_COEFFS = {
+    "Aries": 1,
+    "Taurus": 0,
+    "Gemini": 2,
+    "Cancer": 0,
+    "Leo": 1,
+    "Virgo": 2,
+    "Libra": 1.5,
+    "Scorpio": 0.5,
+    "Sagittarius": 1,
+    "Capricorn": 1,
+    "Aquarius": 2,
+    "Pisces": 0,
+}
+
+# Fertility phase coefficients for above-ground plants
+FERTILITY_PHASE_COEFFS_ABOVE = {
+    "New Moon": 0,
+    "Waxing Moon": 1,
+    "Full Moon": 0,
+    "Waning Moon": 0.5,
+}
+
+# Fertility phase coefficients for root crops
+FERTILITY_PHASE_COEFFS_ROOT = {
+    "New Moon": 0,
+    "Waxing Moon": 0.5,
+    "Full Moon": 0,
+    "Waning Moon": 1,
+}
+
+# Pruning phase coefficients
+PRUNING_PHASE_COEFFS = {
+    "New Moon": 0,
+    "Waxing Moon": 1,
+    "Full Moon": 0,
+    "Waning Moon": 0.5,
+}
+
 @tool
 def get_moon_info(date_time: str) -> dict:
     """
-    Returns Moon's Zodiac position with degrees/minutes and phase for given date/time.
+    Returns Moon's Zodiac position, phase, and fertility and pruning indices for the given date/time.
 
     Args:
         date_time (str): ISO 8601 formatted datetime (YYYY-MM-DDTHH:MM:SS)
 
     Returns:
-        dict: {"zodiac_position": "Leo 15°30'",
-               "moon_phase": "Waxing Gibbous"}
+        dict: {
+            "zodiac_position": "Leo 15°30'",
+            "moon_phase": "Waxing Gibbous",
+            "fertility_above_ground": 2.0,
+            "fertility_root_crop": 1.5,
+            "pruning": 2.0
+        }
     """
     try:
+        # Parse input datetime and localize to UTC
         user_time = datetime.datetime.strptime(date_time, "%Y-%m-%dT%H:%M:%S")
         user_time = pytz.utc.localize(user_time)
 
+        # Load ephemeris and timescale
         planets = load('de421.bsp')
         ts = load.timescale()
         t = ts.from_datetime(user_time)
 
+        # Define celestial bodies
         earth = planets['earth']
         moon = planets['moon']
         sun = planets['sun']
 
-        # Calculate ecliptic longitude for geocentric position
+        # Calculate Moon's ecliptic longitude
         astrometric = earth.at(t).observe(moon)
         _, _, ecliptic_lon = astrometric.ecliptic_latlon()
         lon_deg = ecliptic_lon.degrees % 360
 
-        # Calculate phase angle
+        # Calculate phase angle (angle between Moon and Sun from Earth)
         phase_angle = earth.at(t).observe(moon).apparent().phase_angle(sun).degrees
 
         # Determine Zodiac sign and position
@@ -75,21 +139,46 @@ def get_moon_info(date_time: str) -> dict:
                 position_degrees = lon_deg - start
                 break
 
-        # Format position to degrees/minutes
+        # Format position to degrees and minutes
         degrees = int(position_degrees)
         minutes = int((position_degrees % 1) * 60)
         position_str = f"{zodiac_sign} {degrees}°{minutes:02}'"
 
-        # Determine phase
+        # Determine moon phase for display
         moon_phase = "Unknown"
         for phase, start, end in MOON_PHASES:
             if start <= phase_angle < end:
                 moon_phase = phase
                 break
 
+        # Determine phase category for indices with 15° orbis for New and Full Moon
+        if (phase_angle >= 345 or phase_angle < 15):
+            phase_category = "New Moon"  # 345° to 15° (30° total orbis)
+        elif 15 <= phase_angle < 165:
+            phase_category = "Waxing Moon"
+        elif 165 <= phase_angle < 195:
+            phase_category = "Full Moon"  # 165° to 195° (30° total orbis)
+        elif 195 <= phase_angle < 345:
+            phase_category = "Waning Moon"
+        else:
+            phase_category = "Unknown"
+
+        # Calculate fertility and pruning indices
+        if zodiac_sign in FERTILITY_SIGN_COEFFS and phase_category in FERTILITY_PHASE_COEFFS_ABOVE:
+            fertility_above_ground = FERTILITY_SIGN_COEFFS[zodiac_sign] + FERTILITY_PHASE_COEFFS_ABOVE[phase_category]
+            fertility_root_crop = FERTILITY_SIGN_COEFFS[zodiac_sign] + FERTILITY_PHASE_COEFFS_ROOT[phase_category]
+            pruning = PRUNING_SIGN_COEFFS[zodiac_sign] + PRUNING_PHASE_COEFFS[phase_category]
+        else:
+            fertility_above_ground = None
+            fertility_root_crop = None
+            pruning = None
+
         return {
             "zodiac_position": position_str,
-            "moon_phase": moon_phase
+            "moon_phase": moon_phase,
+            "fertility_above_ground": fertility_above_ground,
+            "fertility_root_crop": fertility_root_crop,
+            "pruning": pruning
         }
 
     except Exception as e: