Update app.py

app.py

@@ -43,7 +43,8 @@ ZODIAC_SIGNS = [
     ("Gemini", 60, 90),
     ("Cancer", 90, 120),
     ("Leo", 120, 150),
-    ("Virgo", 150, 180),
+    ("Virgo"
+, 150, 180),
     ("Libra", 180, 210),
     ("Scorpio", 210, 240),
     ("Sagittarius", 240, 270),
@@ -120,6 +121,38 @@ PRUNING_PHASE_COEFFS = {
     "Waning Moon": 0.5,
 }
 
+# Translation dictionaries
+classification_ru = {
+    'Swallowed': 'проглоченная',
+    'Tiny': 'сверхмалая',
+    'Small': 'малая',
+    'Normal': 'нормальная',
+    'Ideal': 'идеальная',
+    'Big': 'большая'
+}
+
+planet_ru = {
+    'Sun': 'Солнце',
+    'Moon': 'Луна',
+    'Mercury': 'Меркурий',
+    'Venus': 'Венера',
+    'Mars': 'Марс',
+    'Jupiter': 'Юпитер',
+    'Saturn': 'Сатурн'
+}
+
+planet_symbols = {
+    'Sun': '☉',
+    'Moon': '☾',
+    'Mercury': '☿',
+    'Venus': '♀',
+    'Mars': '♂',
+    'Jupiter': '♃',
+    'Saturn': '♄'
+}
+
+
+
 @tool
 def get_moon_info(date_time: str) -> dict:
     """
@@ -291,6 +324,149 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
+# Function to calculate PLadder and zone sizes
+@tool
+def PLadder_ZSizes(date_time_iso: str):
+    """
+    Calculate the planetary ladder and zone sizes for a given date and time.
+    
+    Args:
+        date_time_iso (str): Date and time in ISO format (e.g., '2023-10-10T12:00:00')
+    
+    Returns:
+        dict: Contains 'PLadder' (list of planets) and 'ZSizes' (list of zone sizes with classifications)
+              or an error message if unsuccessful
+    """
+    try:
+        dt = datetime.fromisoformat(date_time_iso)
+        if dt.year < 1900 or dt.year > 2050:
+            return {"error": "Дата вне диапазона. Должна быть между 1900 и 2050 годами."}
+        
+        # Load ephemeris
+        planets = load('de421.bsp')
+        earth = planets['earth']
+        
+        # Define planet objects
+        planet_objects = {
+            'Sun': planets['sun'],
+            'Moon': planets['moon'],
+            'Mercury': planets['mercury'],
+            'Venus': planets['venus'],
+            'Mars': planets['mars'],
+            'Jupiter': planets['jupiter barycenter'],
+            'Saturn': planets['saturn barycenter']
+        }
+        
+        # Create time object
+        ts = load.timescale()
+        t = ts.utc(dt.year, dt.month, dt.day, dt.hour, dt.minute, dt.second)
+        
+        # Compute ecliptic longitudes
+        longitudes = {}
+        for planet in planet_objects:
+            apparent = earth.at(t).observe(planet_objects[planet]).apparent()
+            _, lon, _ = apparent.ecliptic_latlon()
+            longitudes[planet] = lon.degrees
+        
+        # Sort planets by longitude to form PLadder
+        sorted_planets = sorted(longitudes.items(), key=lambda x: x[1])
+        PLadder = [p for p, _ in sorted_planets]
+        sorted_lons = [lon for _, lon in sorted_planets]
+        
+        # Calculate zone sizes
+        zone_sizes = [sorted_lons[0]] + [sorted_lons[i+1] - sorted_lons[i] for i in range(6)] + [360 - sorted_lons[6]]
+        
+        # Determine bordering planets for classification
+        bordering = [[PLadder[0]]] + [[PLadder[i-1], PLadder[i]] for i in range(1, 7)] + [[PLadder[6]]]
+        
+        # Classify each zone
+        ZSizes = []
+        for i, size in enumerate(zone_sizes):
+            bord = bordering[i]
+            if any(p in ['Sun', 'Moon'] for p in bord):
+                X = 7
+            elif any(p in ['Mercury', 'Venus', 'Mars'] for p in bord):
+                X = 6
+            else:
+                X = 5
+            
+            if size <= 1:
+                classification = 'Swallowed'
+            elif size <= X:
+                classification = 'Tiny'
+            elif size <= 40:
+                classification = 'Small'
+            elif size < 60:
+                if 50 <= size <= 52:
+                    classification = 'Ideal'
+                else:
+                    classification = 'Normal'
+            else:
+                classification = 'Big'
+            
+            # Convert size to degrees and minutes
+            d = int(size)
+            m = int((size - d) * 60)
+            size_str = f"{d}°{m}'"
+            ZSizes.append((size_str, classification))
+        
+        return {'PLadder': PLadder, 'ZSizes': ZSizes}
+    
+    except ValueError:
+        return {"error": "Неверный формат даты и времени. Используйте ISO формат, например, '2023-10-10T12:00:00'"}
+    except Exception as e:
+        return {"error": f"Ошибка при вычислении: {str(e)}"}
+
+
+# Function to parse date and time into ISO format
+def parse_date_time(date_time_str):
+    try:
+        dt = parser.parse(date_time_str)
+        return dt.isoformat()
+    except ValueError:
+        return None
+
+# Function to convert longitude to zodiac sign and degrees
+def lon_to_sign(lon):
+    signs = ["Овен", "Телец", "Близнецы", "Рак", "Лев", "Дева", 
+             "Весы", "Скорпион", "Стрелец", "Козерог", "Водолей", "Рыбы"]
+    sign_index = int(lon // 30)
+    sign = signs[sign_index]
+    degrees = int(lon % 30)
+    minutes = int((lon % 1) * 60)
+    return f"{sign} {degrees}°{minutes}'"
+
+
+@tool
+def plot_pladder(PLadder):
+    """
+    Plot the planetary ladder as a right triangle with planet symbols.
+    
+    Args:
+        PLadder (list): List of planet names in order
+    
+    Returns:
+        matplotlib.figure.Figure: The generated plot
+    """
+    fig, ax = plt.subplots()
+    # Plot triangle with right angle on top: vertices at (0,0), (1.5,3), (3,0)
+    ax.plot([0, 1.5, 3, 0], [0, 3, 0, 0], 'k-')
+    # Draw horizontal lines dividing height into three equal parts
+    ax.plot([0, 3], [1, 1], 'k--')
+    ax.plot([0, 3], [2, 2], 'k--')
+    # Define positions for planets 1 to 7, adjusted to avoid overlap
+    positions = [(0.2, 0.2), (0.2, 1.2), (0.2, 2.2), (1.5, 3.2), (2.8, 2.2), (2.8, 1.2), (2.8, 0.2)]
+    for i, pos in enumerate(positions):
+        symbol = planet_symbols[PLadder[i]]
+        ax.text(pos[0], pos[1], symbol, ha='center', va='center', fontsize=24)  # Doubled font size
+    ax.set_xlim(-0.5, 3.5)
+    ax.set_ylim(-0.5, 3.5)
+    ax.set_aspect('equal')
+    ax.axis('off')
+    return fig
+
+
+
 # Load image tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -301,7 +477,7 @@ with open("prompts.yaml", 'r') as stream:
 # Initialize agent with all tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_moon_info, get_current_time_in_timezone, get_current_time_raw, image_generation_tool],
+    tools=[final_answer, get_moon_info, get_current_time_in_timezone, get_current_time_raw, plot_pladder, PLadder_ZSizes, image_generation_tool],
     max_steps=6,
     verbosity_level=1,
     prompt_templates=prompt_templates