Update app.py

app.py

@@ -1,53 +1,24 @@
-"""
-Streamlined Streamlit application for AI Astrology app with automatic timezone detection.
-"""
-
 import streamlit as st
 from datetime import datetime
 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  # Assuming you have the astro_core.py file
-from ai_interpreter import AstroAI  # Assuming you have the ai_interpreter.py file
+from astro_core import ChartCalculator
 
-# Load environment variables (if you are using .env for API keys, etc.)
 load_dotenv()
 
-def init_session_state():
-    """Initialize session state variables"""
-    if 'birth_chart' not in st.session_state:
-        st.session_state.birth_chart = None
-    if 'chat_history' not in st.session_state:
-        st.session_state.chat_history = []
-    if 'user_info' not in st.session_state:
-        st.session_state.user_info = None
-    if 'calculation_error' not in st.session_state:
-        st.session_state.calculation_error = None
-
 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:
             latitude, longitude = location.latitude, location.longitude
-
-            # Get timezone string for the coordinates
             timezone_str = tf.timezone_at(lng=longitude, lat=latitude)
             if not timezone_str:
-                timezone_str = 'UTC'  # Fallback to UTC if timezone not found
-
-            return {
-                'latitude': latitude,
-                'longitude': longitude,
-                'timezone': timezone_str,
-                'error': None
-            }
+                timezone_str = 'UTC'
+            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)}'}
@@ -55,113 +26,57 @@ def get_location_details(location_name: str) -> dict:
         return {'error': f'Unexpected error: {str(e)}'}
 
 def calculate_chart(user_data: dict):
-    """Calculate birth chart based on user data"""
-    try:
-        calculator = ChartCalculator()
-        birth_datetime = datetime.combine(user_data['birth_date'], user_data['birth_time'])
-
-        chart_data = calculator.calculate_birth_chart(
-            birth_datetime,
-            user_data['latitude'],
-            user_data['longitude'],
-            user_data['timezone']
-        )
-
-        # Check for calculation errors
-        if 'error' in chart_data:
-            st.session_state.calculation_error = chart_data['error']
-            return None
-
-        return chart_data
-    except Exception as e:
-        st.session_state.calculation_error = str(e)
-        return None
+    calculator = ChartCalculator()
+    birth_datetime = datetime.combine(user_data['birth_date'], user_data['birth_time'])
+    return calculator.calculate_birth_chart(
+        birth_datetime,
+        user_data['latitude'],
+        user_data['longitude'],
+        user_data['timezone']
+    )
 
 def main():
     st.title("AI Astrology Assistant")
 
-    # Initialize session state
-    init_session_state()
-
-    # User Information Form
     with st.form("user_info_form"):
-        st.subheader("Personal Information")
-
-        col1, col2 = st.columns(2)
-        with col1:
-            name = st.text_input("Full Name")
-            gender = st.selectbox("Gender", ["Male", "Female", "Non-binary", "Prefer not to say"])
-            birth_date = st.date_input("Birth Date", min_value=datetime(1900, 1, 1))
-
-        with col2:
-            birth_time = st.time_input("Birth Time")
-            birth_place = st.text_input("Birth Place (City, Country)")
-
+        name = st.text_input("Full Name")
+        birth_date = st.date_input("Birth Date", min_value=datetime(1900, 1, 1))
+        birth_time = st.time_input("Birth Time")
+        birth_place = st.text_input("Birth Place (City, Country)")
         submit_button = st.form_submit_button("Generate Birth Chart")
 
         if submit_button:
             if not birth_place:
                 st.error("Please enter your birth place.")
             else:
-                with st.spinner("Looking up location details..."):
+                with st.spinner("Fetching 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
-
-                    # Store user information
-                    st.session_state.user_info = {
+                    user_data = {
                         'name': name,
-                        'gender': gender,
                         'birth_date': birth_date,
                         'birth_time': birth_time,
-                        'birth_place': birth_place,
                         '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)
+                    with st.spinner("Calculating birth chart..."):
+                        birth_chart = calculate_chart(user_data)
 
-                    if st.session_state.birth_chart:
+                    if 'error' in birth_chart:
+                        st.error(f"Error: {birth_chart['error']}")
+                    else:
                         st.success("Birth chart calculated successfully!")
-                    elif st.session_state.calculation_error:
-                        st.error(f"Error calculating birth chart: {st.session_state.calculation_error}")
-
-    # Display areas (only show if birth chart is calculated)
-    if st.session_state.birth_chart and st.session_state.user_info:
-        col1, col2 = st.columns([2, 3])
-
-        with col1:
-            st.subheader(f"Birth Chart for {st.session_state.user_info['name']}")
-
-            # Display birth details
-            st.write("**Birth Details:**")
-            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:**")
-            for planet, data in st.session_state.birth_chart['planets'].items():
-                st.write(f"{planet}: {data['degrees']:.2f}° {data['sign']}")
-
-        with col2:
-            st.subheader("Graphical Astrology Chart")
-            if st.button("Show Astrology Chart"):
-                # Generate and display graphical chart
-                calculator = ChartCalculator()
-                with st.spinner("Generating graphical chart..."):
-                    calculator.draw_chart(st.session_state.birth_chart)
+                        st.write("**Planetary Positions:**")
+                        for planet, data in birth_chart['planets'].items():
+                            if 'error' in data:
+                                st.write(f"{planet}: {data['error']}")
+                            else:
+                                st.write(f"{planet}: {data['degrees']:.2f}° {data['sign']}")
 
 if __name__ == "__main__":
     main()