Create app.py

app.py

@@ -0,0 +1,153 @@
+import gradio as gr
+import matplotlib.pyplot as plt
+from skyfield.api import load, Topos
+from datetime import datetime
+from dateutil import parser
+from io import BytesIO
+from PIL import Image
+
+# Russian translations for planets
+planet_ru = {
+    'Sun': 'Солнце', 'Moon': 'Луна', 'Mercury': 'Меркурий', 'Venus': 'Венера',
+    'Mars': 'Марс', 'Jupiter': 'Юпитер', 'Saturn': 'Сатурн'
+}
+
+# Planet symbols for plotting
+planet_symbols = {
+    'Sun': '☉', 'Moon': '☾', 'Mercury': '☿', 'Venus': '♀',
+    'Mars': '♂', 'Jupiter': '♃', 'Saturn': '♄'
+}
+
+# Parse date-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
+
+# Convert longitude to zodiac sign and degrees
+def lon_to_sign(lon):
+    signs = ["Овен", "Телец", "Близнецы", "Рак", "Лев", "Дева", 
+             "Весы", "Скорпион", "Стрелец", "Козерог", "Водолей", "Рыбы"]
+    sign_index = int(lon // 30)
+    degrees = int(lon % 30)
+    minutes = int((lon % 1) * 60)
+    return f"{signs[sign_index]} {degrees}°{minutes}'"
+
+# Calculate PLadder and zone sizes
+def PLadder_ZSizes(date_time_iso, lat, lon):
+    try:
+        dt = datetime.fromisoformat(date_time_iso)
+        if not 1900 <= dt.year <= 2050:
+            return {"error": "Дата вне диапазона (1900–2050)."}
+        
+        planets = load('de421.bsp')
+        earth = planets['earth']
+        observer = earth + Topos(latitude_degrees=float(lat), longitude_degrees=float(lon))
+        
+        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']
+        }
+        
+        ts = load.timescale()
+        t = ts.utc(dt.year, dt.month, dt.day, dt.hour, dt.minute, dt.second)
+        
+        longitudes = {}
+        for planet, obj in planet_objects.items():
+            astrometric = observer.at(t).observe(obj)
+            _, lon, _ = astrometric.apparent().ecliptic_latlon()
+            longitudes[planet] = lon.degrees
+        
+        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]
+        
+        zone_sizes = [sorted_lons[0]] + [sorted_lons[i+1] - sorted_lons[i] for i in range(6)] + [360 - sorted_lons[6]]
+        bordering = [[PLadder[0]]] + [[PLadder[i-1], PLadder[i]] for i in range(1, 7)] + [[PLadder[6]]]
+        
+        ZSizes = []
+        for i, size in enumerate(zone_sizes):
+            bord = bordering[i]
+            X = 7 if any(p in ['Sun', 'Moon'] for p in bord) else 6 if any(p in ['Mercury', 'Venus', 'Mars'] for p in bord) else 5
+            classification = ('Swallowed' if size <= 1 else 'Tiny' if size <= X else 'Small' if size <= 40 else
+                             'Ideal' if 50 <= size <= 52 else 'Normal' if size < 60 else 'Big')
+            d, m = int(size), int((size - int(size)) * 60)
+            ZSizes.append((f"{d}°{m}'", classification))
+        
+        return {'PLadder': PLadder, 'ZSizes': ZSizes, 'longitudes': longitudes}
+    
+    except ValueError:
+        return {"error": "Неверный формат даты и времени."}
+    except Exception as e:
+        return {"error": f"Ошибка: {str(e)}"}
+
+# Plot the planetary ladder
+def plot_pladder(PLadder):
+    fig, ax = plt.subplots()
+    ax.plot([0, 1.5, 3, 0], [0, 3, 0, 0], 'k-')  # Triangle
+    ax.plot([0, 3], [1, 1], 'k--')  # Horizontal lines
+    ax.plot([0, 3], [2, 2], 'k--')
+    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, (x, y) in enumerate(positions):
+        ax.text(x, y, planet_symbols[PLadder[i]], ha='center', va='center', fontsize=24)
+    ax.set_xlim(-0.5, 3.5)
+    ax.set_ylim(-0.5, 3.5)
+    ax.set_aspect('equal')
+    ax.axis('off')
+    return fig
+
+# Main interface function
+def chat_interface(query, lat, lon):
+    if not query.startswith("PLadder "):
+        return "Запрос должен начинаться с 'PLadder' и содержать дату/время.", None
+    
+    date_time_str = query.split(" ", 1)[1]
+    date_time_iso = parse_date_time(date_time_str)
+    if not date_time_iso:
+        return "Неверный формат даты и времени.", None
+    
+    result = PLadder_ZSizes(date_time_iso, lat, lon)
+    if "error" in result:
+        return result["error"], None
+    
+    PLadder = result["PLadder"]
+    ZSizes = result["ZSizes"]
+    longitudes = result["longitudes"]
+    
+    planet_list = "\n".join([f"{planet_ru[p]}: {lon_to_sign(longitudes[p])}" for p in PLadder])
+    zones_text = "\n".join([f"Зона {i+1}: {size} ({cls})" for i, (size, cls) in enumerate(ZSizes)])
+    
+    fig = plot_pladder(PLadder)
+    buf = BytesIO()
+    fig.savefig(buf, format='png', bbox_inches='tight')
+    buf.seek(0)
+    img = Image.open(buf)
+    plt.close(fig)
+    
+    text = f"Планетарная лестница:\n{planet_list}\n\nРазмеры зон:\n{zones_text}"
+    return text, img
+
+# Gradio UI
+with gr.Blocks() as interface:
+    with gr.Row():
+        with gr.Column(scale=2):
+            output_text = gr.Textbox(label="Ответ", lines=10)
+        with gr.Column(scale=1):
+            output_image = gr.Image(label="График планетарной лестницы")
+    with gr.Row():
+        with gr.Column(scale=1):
+            query_text = gr.Textbox(label="Запрос", placeholder="Пример: PLadder 2023-10-10 12:00")
+            location_lat = gr.Textbox(label="Широта", placeholder="Пример: 37.7749")
+            location_lon = gr.Textbox(label="Долгота", placeholder="Пример: -122.4194")
+            submit_button = gr.Button("Отправить")
+    
+    submit_button.click(
+        fn=chat_interface,
+        inputs=[query_text, location_lat, location_lon],
+        outputs=[output_text, output_image]
+    )
+
+interface.launch()