Update app.py

app.py

@@ -1,12 +1,13 @@
 # app.py
 """
-Main Streamlit application file for the AI Astrology app.
-Handles the user interface and coordinates between astrology calculations and AI interpretation.
+Streamlined Streamlit application for AI Astrology app with simplified user input.
 """
 
 import streamlit as st
 from datetime import datetime
 import pytz
+from geopy.geocoders import Nominatim
+from geopy.exc import GeocoderTimedOut, GeocoderServiceError
 from dotenv import load_dotenv
 from astro_core import ChartCalculator
 from ai_interpreter import AstroAI
@@ -20,6 +21,33 @@ def init_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 'location_coords' not in st.session_state:
+        st.session_state.location_coords = None
+
+def get_coordinates(location_name: str) -> tuple:
+    """Get coordinates from location name using geocoding"""
+    try:
+        geolocator = Nominatim(user_agent="ai_astrology_app")
+        location = geolocator.geocode(location_name)
+        if location:
+            return location.latitude, location.longitude
+        return None
+    except (GeocoderTimedOut, GeocoderServiceError):
+        return None
+
+def calculate_chart(user_data: dict):
+    """Calculate birth chart based on user data"""
+    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")
@@ -27,64 +55,104 @@ def main():
     # Initialize session state
     init_session_state()
     
-    # Sidebar for user input
-    st.sidebar.header("Birth Information")
-    
-    date = st.sidebar.date_input("Birth Date", datetime.now())
-    time = st.sidebar.time_input("Birth Time", datetime.now().time())
+    # 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)")
+            timezone = st.selectbox("Timezone", pytz.all_timezones)
+        
+        submit_button = st.form_submit_button("Generate Birth Chart")
+        
+        if submit_button and birth_place:
+            coordinates = get_coordinates(birth_place)
+            if coordinates:
+                latitude, longitude = coordinates
+                st.session_state.location_coords = coordinates
+                
+                # Store user information
+                st.session_state.user_info = {
+                    'name': name,
+                    'gender': gender,
+                    'birth_date': birth_date,
+                    'birth_time': birth_time,
+                    'birth_place': birth_place,
+                    'latitude': latitude,
+                    'longitude': longitude,
+                    'timezone': timezone
+                }
+                
+                # Calculate birth chart
+                with st.spinner("Calculating your birth chart..."):
+                    st.session_state.birth_chart = calculate_chart(st.session_state.user_info)
+                st.success("Birth chart calculated successfully!")
+            else:
+                st.error("Could not find coordinates for the given location. Please check the location name.")
     
-    # Location input
-    latitude = st.sidebar.number_input("Latitude", -90.0, 90.0, 0.0)
-    longitude = st.sidebar.number_input("Longitude", -180.0, 180.0, 0.0)
-    timezone = st.sidebar.selectbox("Timezone", pytz.all_timezones)
-    
-    # Create calculator and AI interpreter instances
-    calculator = ChartCalculator()
-    interpreter = AstroAI()
-    
-    # Calculate button
-    if st.sidebar.button("Calculate Chart"):
-        with st.spinner("Calculating birth chart..."):
-            # Combine date and time
-            birth_datetime = datetime.combine(date, time)
+    # 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']}")
             
-            # Calculate birth chart
-            st.session_state.birth_chart = calculator.calculate_birth_chart(
-                birth_datetime, latitude, longitude, timezone
-            )
-            st.success("Chart calculated!")
-    
-    # Display areas
-    col1, col2 = st.columns([2, 3])
-    
-    with col1:
-        st.subheader("Birth Chart")
-        if st.session_state.birth_chart:
             # Display planetary positions
-            st.write("### 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']} in House {data['house']}")
-    
-    with col2:
-        st.subheader("AI Interpretation")
-        if st.session_state.birth_chart:
-            # Question input
-            user_question = st.text_input("Ask about your chart:")
-            if user_question:
+        
+        with col2:
+            st.subheader("AI Interpretation")
+            # Suggested questions
+            question_templates = [
+                "What are the main themes in my birth chart?",
+                f"How does my {st.session_state.user_info['gender'].lower()} energy manifest in this chart?",
+                "What career paths might suit me based on my chart?",
+                "What are my relationship patterns according to this chart?",
+                "What are my greatest strengths based on this chart?"
+            ]
+            
+            selected_question = st.selectbox(
+                "Choose a question or type your own below:",
+                [""] + question_templates
+            )
+            
+            custom_question = st.text_input(
+                "Ask about your chart:",
+                value=selected_question
+            )
+            
+            if custom_question:
                 with st.spinner("Generating interpretation..."):
+                    interpreter = AstroAI()
                     interpretation = interpreter.get_interpretation(
                         st.session_state.birth_chart,
-                        user_question
+                        custom_question
                     )
                     st.session_state.chat_history.append({
-                        "question": user_question,
+                        "question": custom_question,
                         "answer": interpretation
                     })
             
             # Display chat history
+            st.write("**Previous Interpretations:**")
             for exchange in st.session_state.chat_history:
-                st.write(f"**You:** {exchange['question']}")
-                st.write(f"**AI:** {exchange['answer']}")
+                with st.expander(f"Q: {exchange['question']}", expanded=False):
+                    st.write(exchange['answer'])
 
 if __name__ == "__main__":
     main()