Update astro_core.py

astro_core.py

@@ -1,5 +1,6 @@
 from datetime import datetime
 from typing import Dict, Any
+import pytz
 import flatlib
 from flatlib.datetime import Datetime
 from flatlib.geopos import GeoPos
@@ -12,6 +13,23 @@ class ChartCalculator:
         self.planets = ['Sun', 'Moon', 'Mercury', 'Venus', 'Mars', 
                        'Jupiter', 'Saturn', 'Uranus', 'Neptune', 'Pluto']
     
+    def _get_utc_offset(self, dt: datetime, timezone: str) -> str:
+        """
+        Get UTC offset for a given datetime and timezone
+        Returns offset in format '+HH:MM' or '-HH:MM'
+        """
+        tz = pytz.timezone(timezone)
+        offset = tz.utcoffset(dt)
+        
+        # Convert timedelta to hours and minutes
+        total_seconds = int(offset.total_seconds())
+        hours = total_seconds // 3600
+        minutes = (abs(total_seconds) % 3600) // 60
+        
+        # Format as string
+        sign = '+' if hours >= 0 else '-'
+        return f"{sign}{abs(hours):02d}:{minutes:02d}"
+    
     def calculate_birth_chart(self, 
                             birth_datetime: datetime,
                             latitude: float,
@@ -25,8 +43,11 @@ class ChartCalculator:
         date_str = birth_datetime.strftime("%Y/%m/%d")
         time_str = birth_datetime.strftime("%H:%M")
         
+        # Get UTC offset for the timezone
+        utc_offset = self._get_utc_offset(birth_datetime, timezone)
+        
         # Create flatlib datetime object
-        date = Datetime(date_str, time_str, timezone)
+        date = Datetime(date_str, time_str, utc_offset)
         pos = GeoPos(latitude, longitude)
         
         # Calculate chart