Update app.py

app.py

@@ -1,82 +1,126 @@
 import streamlit as st
 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
+import plotly.graph_objects as go
+import pandas as pd
+import anthropic
+from typing import Dict, Any
+import json
 
-load_dotenv()
+# Page configuration
+st.set_page_config(
+    page_title="AI Astrology Assistant",
+    page_icon="🌟",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
 
-def get_location_details(location_name: str) -> dict:
+# Custom CSS for a more modern look
+st.markdown("""
+    <style>
+    .stApp {
+        background: linear-gradient(135deg, #f5f7ff 0%, #f8f0ff 100%);
+    }
+    .main {
+        background-color: transparent;
+    }
+    .stTextInput > div > div > input {
+        background-color: rgba(255, 255, 255, 0.8);
+    }
+    .stButton > button {
+        background-color: #6B46C1;
+        color: white;
+        border-radius: 8px;
+        padding: 0.5rem 1rem;
+        border: none;
+    }
+    .stButton > button:hover {
+        background-color: #553C9A;
+    }
+    .chat-message {
+        padding: 1rem;
+        border-radius: 8px;
+        margin-bottom: 0.5rem;
+    }
+    .user-message {
+        background-color: rgba(107, 70, 193, 0.1);
+        margin-left: 20%;
+    }
+    .assistant-message {
+        background-color: white;
+        margin-right: 20%;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+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 'location_details' not in st.session_state:
+        st.session_state.location_details = None
+    if 'chart_analysis' not in st.session_state:
+        st.session_state.chart_analysis = None
+
+def get_location_details(location_name: str) -> Dict[str, Any]:
+    """Get coordinates and timezone for a location"""
     try:
         geolocator = Nominatim(user_agent="ai_astrology_app")
         tf = TimezoneFinder()
+        
         location = geolocator.geocode(location_name)
         if location:
             latitude, longitude = location.latitude, location.longitude
-            timezone_str = tf.timezone_at(lng=longitude, lat=latitude)
-            if not timezone_str:
-                timezone_str = 'UTC'
-            return {'latitude': latitude, 'longitude': longitude, 'timezone': timezone_str, 'error': None}
+            timezone_str = tf.timezone_at(lat=latitude, lng=longitude) or '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)}'}
     except Exception as e:
-        return {'error': f'Unexpected error: {str(e)}'}
-
-def calculate_chart(user_data: dict):
-    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")
-
-    with st.form("user_info_form"):
-        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("Fetching location details..."):
-                    location_details = get_location_details(birth_place)
-
-                if location_details.get('error'):
-                    st.error(location_details['error'])
-                else:
-                    user_data = {
-                        'name': name,
-                        'birth_date': birth_date,
-                        'birth_time': birth_time,
-                        'latitude': location_details['latitude'],
-                        'longitude': location_details['longitude'],
-                        'timezone': location_details['timezone']
-                    }
-
-                    with st.spinner("Calculating birth chart..."):
-                        birth_chart = calculate_chart(user_data)
-
-                    if 'error' in birth_chart:
-                        st.error(f"Error: {birth_chart['error']}")
-                    else:
-                        st.success("Birth chart calculated successfully!")
-                        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']}")
+        return {'error': f"Error finding location: {str(e)}"}
 
-if __name__ == "__main__":
-    main()
+def create_chart_visualization(chart_data: Dict[str, Any]):
+    """Create an interactive circular chart visualization"""
+    # Create circular chart layout
+    fig = go.Figure()
+    
+    # Add zodiac ring
+    zodiac_positions = list(range(0, 360, 30))
+    zodiac_signs = ['♈', '♉', '♊', '♋', '♌', '♍', '♎', '♏', '♐', '♑', '♒', '♓']
+    
+    for pos, sign in zip(zodiac_positions, zodiac_signs):
+        fig.add_annotation(
+            x=1.1 * np.cos(np.radians(pos)),
+            y=1.1 * np.sin(np.radians(pos)),
+            text=sign,
+            showarrow=False,
+            font=dict(size=20)
+        )
+    
+    # Add planets
+    for planet, data in chart_data['planets'].items():
+        if 'error' not in data:
+            angle = np.radians(data['longitude'])
+            r = 0.8  # Radius for planets
+            fig.add_annotation(
+                x=r * np.cos(angle),
+                y=r * np.sin(angle),
+                text=planet,
+                showarrow=True,
+                arrowhead=2
+            )
+    
+    # Update layout
+    fig.update_layout(
+        showlegend=False,
+        xaxis=dict(range=[-1.2, 1.2], showgrid=False, zeroline=False, showticklabels=False),
+        yaxis=dict(range=[-1.2, 1.2], showgrid=False, zeroline=False,