Update app.py

app.py

@@ -1,6 +1,6 @@
 # app.py
 """
-Streamlined Streamlit application for AI Astrology app with simplified user input.
+Streamlined Streamlit application for AI Astrology app with automatic timezone detection.
 """
 
 import streamlit as st
@@ -8,6 +8,7 @@ from datetime import datetime
 import pytz
 from geopy.geocoders import Nominatim
 from geopy.exc import GeocoderTimedOut, GeocoderServiceError
+from timezonefinder import TimezoneFinder
 from dotenv import load_dotenv
 from astro_core import ChartCalculator
 from ai_interpreter import AstroAI
@@ -28,16 +29,34 @@ def init_session_state():
     if 'calculation_error' not in st.session_state:
         st.session_state.calculation_error = None
 
-def get_coordinates(location_name: str) -> tuple:
-    """Get coordinates from location name using geocoding"""
+def get_location_details(location_name: str) -> dict:
+    """Get coordinates and timezone from location name"""
     try:
+        # Initialize geocoder and timezone finder
         geolocator = Nominatim(user_agent="ai_astrology_app")
+        tf = TimezoneFinder()
+        
+        # Get location
         location = geolocator.geocode(location_name)
         if location:
-            return location.latitude, location.longitude
-        return None
-    except (GeocoderTimedOut, GeocoderServiceError):
-        return None
+            latitude, longitude = location.latitude, location.longitude
+            
+            # Get timezone string for the coordinates
+            timezone_str = tf.timezone_at(lat=latitude, lng=longitude)
+            if not timezone_str:
+                timezone_str = 'UTC'  # Fallback to UTC if timezone not found
+            
+            return {
+                'latitude': latitude,
+                'longitude': longitude,
+                'timezone': timezone_str,
+                'error': None
+            }
+        return {'error': 'Location not found'}
+    except (GeocoderTimedOut, GeocoderServiceError) as e:
+        return {'error': f'Geocoding error: {str(e)}'}
+    except Exception as e:
+        return {'error': f'Unexpected error: {str(e)}'}
 
 def calculate_chart(user_data: dict):
     """Calculate birth chart based on user data"""
@@ -81,12 +100,6 @@ def main():
         with col2:
             birth_time = st.time_input("Birth Time")
             birth_place = st.text_input("Birth Place (City, Country)")
-            # Get local timezone based on location if possible, otherwise show selector
-            timezone = st.selectbox(
-                "Timezone",
-                pytz.common_timezones,  # Using common timezones instead of all timezones
-                index=pytz.common_timezones.index('UTC')
-            )
         
         submit_button = st.form_submit_button("Generate Birth Chart")
         
@@ -94,11 +107,12 @@ def main():
             if not birth_place:
                 st.error("Please enter your birth place.")
             else:
-                coordinates = get_coordinates(birth_place)
-                if coordinates:
-                    latitude, longitude = coordinates
-                    st.session_state.location_coords = coordinates
-                    
+                with st.spinner("Looking up location details..."):
+                    location_details = get_location_details(birth_place)
+                
+                if location_details.get('error'):
+                    st.error(location_details['error'])
+                else:
                     # Reset error state
                     st.session_state.calculation_error = None
                     
@@ -109,11 +123,14 @@ def main():
                         'birth_date': birth_date,
                         'birth_time': birth_time,
                         'birth_place': birth_place,
-                        'latitude': latitude,
-                        'longitude': longitude,
-                        'timezone': timezone
+                        'latitude': location_details['latitude'],
+                        'longitude': location_details['longitude'],
+                        'timezone': location_details['timezone']
                     }
                     
+                    # Display detected timezone
+                    st.info(f"Detected timezone for {birth_place}: {location_details['timezone']}")
+                    
                     # Calculate birth chart
                     with st.spinner("Calculating your birth chart..."):
                         st.session_state.birth_chart = calculate_chart(st.session_state.user_info)
@@ -122,8 +139,6 @@ def main():
                         st.success("Birth chart calculated successfully!")
                     elif st.session_state.calculation_error:
                         st.error(f"Error calculating birth chart: {st.session_state.calculation_error}")
-                else:
-                    st.error("Could not find coordinates for the given location. Please check the location name.")
     
     # Display areas (only show if birth chart is calculated)
     if st.session_state.birth_chart and st.session_state.user_info:
@@ -137,6 +152,7 @@ def main():
             st.write(f"Date: {st.session_state.user_info['birth_date'].strftime('%B %d, %Y')}")
             st.write(f"Time: {st.session_state.user_info['birth_time'].strftime('%H:%M')}")
             st.write(f"Location: {st.session_state.user_info['birth_place']}")
+            st.write(f"Timezone: {st.session_state.user_info['timezone']}")
             
             # Display planetary positions
             st.write("**Planetary Positions:**")