Update astro_core.py

astro_core.py

@@ -18,6 +18,7 @@ class ChartCalculator:
     """Handles astrological calculations"""
 
     def __init__(self):
+        # List of planets, including modern ones like Uranus, Neptune, and Pluto
         self.planets = [
             'Sun', 'Moon', 'Mercury', 'Venus', 'Mars',
             'Jupiter', 'Saturn', 'Uranus', 'Neptune', 'Pluto'
@@ -62,12 +63,21 @@ class ChartCalculator:
             # Extract planetary positions
             positions = {}
             for planet in self.planets:
-                obj = chart.get(planet)
-                if obj:
-                    lon = self._normalize_longitude(obj.lon)
+                try:
+                    obj = chart.get(planet)
+                    if obj:
+                        lon = self._normalize_longitude(obj.lon)
+                        positions[planet] = {
+                            'sign': self._get_zodiac_sign(lon),
+                            'degrees': lon
+                        }
+                    else:
+                        positions[planet] = {
+                            'error': f"{planet} not supported by the astrological library"
+                        }
+                except KeyError:
                     positions[planet] = {
-                        'sign': self._get_zodiac_sign(lon),
-                        'degrees': lon
+                        'error': f"{planet} not supported or unavailable"
                     }
 
             # Extract house cusps
@@ -113,9 +123,10 @@ class ChartCalculator:
 
         # Plot planets
         for planet, data in chart_data['planets'].items():
-            degrees = data['degrees']
-            angle = np.deg2rad(degrees)
-            ax.plot([angle], [0.5], marker='o', label=planet, markersize=10)
+            if 'error' not in data:
+                degrees = data['degrees']
+                angle = np.deg2rad(degrees)
+                ax.plot([angle], [0.5], marker='o', label=planet, markersize=10)
 
         # Draw the chart
         ax.legend(loc='upper right', bbox_to_anchor=(1.3, 1))