Create app.py

app.py

@@ -0,0 +1,118 @@
+# app.py
+"""
+Main Streamlit application file for the AI Astrology app.
+Supports both Western and Vedic astrology with AI interpretation.
+"""
+
+import streamlit as st
+from datetime import datetime
+import pytz
+from dotenv import load_dotenv
+from astro_core import ChartCalculator, BirthInfo
+from ai_interpreter import AstroAI
+
+# Load environment variables
+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 'astro_ai' not in st.session_state:
+        st.session_state.astro_ai = AstroAI()
+    if 'birth_time' not in st.session_state:
+        st.session_state.birth_time = None
+
+def main():
+    st.title("AI Astrology Assistant")
+    
+    # Initialize session state
+    init_session_state()
+    
+    # Sidebar for user input
+    st.sidebar.header("Birth Information")
+    
+    # System selection
+    system = st.sidebar.radio(
+        "Astrological System",
+        ['western', 'vedic'],
+        captions=['Tropical Zodiac', 'Sidereal Zodiac']
+    )
+    
+    # Birth information inputs
+    date = st.sidebar.date_input("Birth Date", datetime.now())
+    time = st.sidebar.time_input("Birth Time", datetime.now().time())
+    
+    # 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 instance
+    calculator = ChartCalculator()
+    
+    # Calculate button
+    if st.sidebar.button("Calculate Chart"):
+        with st.spinner(f"Calculating {system.title()} birth chart..."):
+            # Combine date and time
+            birth_datetime = datetime.combine(date, time)
+            st.session_state.birth_time = birth_datetime
+            
+            # Create birth info
+            birth_info = BirthInfo(
+                datetime=birth_datetime,
+                latitude=latitude,
+                longitude=longitude,
+                timezone=timezone,
+                system=system
+            )
+            
+            # Calculate chart
+            st.session_state.birth_chart = calculator.calculate_chart(birth_info)
+            
+            # Initialize AI with new chart
+            st.session_state.astro_ai.initialize_context(
+                st.session_state.birth_chart,
+                birth_datetime
+            )
+            
+            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 system-specific chart data
+            if system == 'western':
+                st.write("### Planetary Positions")
+                for planet, data in st.session_state.birth_chart['planets'].items():
+                    retrograde = " (R)" if data.get('retrograde') else ""
+                    st.write(f"{planet}: {data['degrees']:.2f}° {data['sign']} in House {data['house']}{retrograde}")
+            else:
+                st.write("### Vedic Positions")
+                for planet, data in st.session_state.birth_chart['planets'].items():
+                    st.write(f"{planet}: {data['degrees']:.2f}° in {data['rashi']}")
+                    st.write(f"House: {data['house']}, Nakshatra: {data['nakshatra']}")
+    
+    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 st.spinner("Generating interpretation..."):
+                    interpretation = st.session_state.astro_ai.get_interpretation(user_question)
+                    st.write("### Interpretation")
+                    st.write(interpretation)
+            
+            # Display previous interpretations
+            if st.session_state.astro_ai.context and st.session_state.astro_ai.context.discussion_history:
+                st.write("### Previous Interpretations")
+                for exchange in reversed(st.session_state.astro_ai.context.discussion_history[:-1]):  # Skip the last one as it's already shown
+                    with st.expander(f"Q: {exchange['question'][:50]}..."):
+                        st.write(exchange['answer'])
+
+if __name__ == "__main__":
+    main()