Update app.py

app.py

@@ -5,10 +5,9 @@ from geopy.geocoders import Nominatim
 from timezonefinder import TimezoneFinder
 from astro_core import ChartCalculator
 import plotly.graph_objects as go
+import numpy as np
 import pandas as pd
-import anthropic
 from typing import Dict, Any
-import json
 
 # Page configuration
 st.set_page_config(
@@ -53,6 +52,12 @@ st.markdown("""
         background-color: white;
         margin-right: 20%;
     }
+    .element-card {
+        background-color: white;
+        border-radius: 8px;
+        padding: 1rem;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+    }
     </style>
 """, unsafe_allow_html=True)
 
@@ -90,7 +95,6 @@ def get_location_details(location_name: str) -> Dict[str, Any]:
 
 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
@@ -122,5 +126,161 @@ def create_chart_visualization(chart_data: Dict[str, Any]):
     # 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,
+        xaxis=dict(
+            range=[-1.2, 1.2],
+            showgrid=False,
+            zeroline=False,
+            showticklabels=False
+        ),
+        yaxis=dict(
+            range=[-1.2, 1.2],
+            showgrid=False,
+            zeroline=False,
+            showticklabels=False
+        ),
+        plot_bgcolor='rgba(0,0,0,0)',
+        paper_bgcolor='rgba(0,0,0,0)',
+        height=600
+    )
+    
+    return fig
+
+def display_chart_analysis(chart_data: Dict[str, Any]):
+    """Display the chart analysis in a structured way"""
+    if 'error' in chart_data:
+        st.error(chart_data['error'])
+        return
+
+    # Create three columns for different aspects of the analysis
+    col1, col2, col3 = st.columns(3)
+
+    with col1:
+        st.markdown("### Planetary Positions")
+        for planet, data in chart_data['planets'].items():
+            if 'error' not in data:
+                st.markdown(f"""
+                    <div class="element-card">
+                        <strong>{planet}</strong><br>
+                        {data['sign']} ({data['degrees']:.1f}°)<br>
+                        House {data.get('house', 'Unknown')}<br>
+                        {'🔄 Retrograde' if data.get('retrograde', False) else ''}
+                    </div>
+                """, unsafe_allow_html=True)
+
+    with col2:
+        st.markdown("### Elements & Qualities")
+        element_data = pd.DataFrame(chart_data['patterns']['elements'], index=['Count']).T
+        st.bar_chart(element_data)
+        
+        quality_data = pd.DataFrame(chart_data['patterns']['modalities'], index=['Count']).T
+        st.bar_chart(quality_data)
+
+    with col3:
+        st.markdown("### Major Aspects")
+        for aspect in chart_data['aspects']:
+            st.markdown(f"""
+                <div class="element-card">
+                    {aspect['planet1']} {aspect['aspect']} {aspect['planet2']}<br>
+                    Orb: {aspect['orb']}° ({aspect['nature']})
+                </div>
+            """, unsafe_allow_html=True)
+
+def main():
+    st.title("🌟 AI Astrology Assistant")
+    
+    # Initialize session state
+    init_session_state()
+    
+    # Birth Information Form
+    with st.form("birth_info"):
+        st.subheader("Enter Birth Details")
+        
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            birth_date = st.date_input(
+                "Birth Date",
+                min_value=datetime(1900, 1, 1),
+                max_value=datetime.now()
+            )
+            birth_time = st.time_input("Birth Time")
+            
+        with col2:
+            birth_place = st.text_input(
+                "Birth Place",
+                placeholder="City, Country"
+            )
+        
+        submit = st.form_submit_button("Calculate Birth Chart")
+        
+        if submit:
+            if not birth_place:
+                st.error("Please enter your birth place.")
+            else:
+                with st.spinner("Looking up location and calculating chart..."):
+                    # Get location details
+                    location = get_location_details(birth_place)
+                    
+                    if location.get('error'):
+                        st.error(location['error'])
+                    else:
+                        # Calculate chart
+                        calculator = ChartCalculator()
+                        birth_datetime = datetime.combine(birth_date, birth_time)
+                        
+                        st.session_state.birth_chart = calculator.calculate_birth_chart(
+                            birth_datetime,
+                            location['latitude'],
+                            location['longitude']
+                        )
+                        
+                        if 'error' in st.session_state.birth_chart:
+                            st.error(st.session_state.birth_chart['error'])
+                        else:
+                            st.success("Birth chart calculated successfully!")
+    
+    # Display chart and analysis if available
+    if st.session_state.birth_chart and 'error' not in st.session_state.birth_chart:
+        # Tabs for different views
+        tab1, tab2 = st.tabs(["Chart Visualization", "Detailed Analysis"])
+        
+        with tab1:
+            st.plotly_chart(
+                create_chart_visualization(st.session_state.birth_chart),
+                use_container_width=True
+            )
+        
+        with tab2:
+            display_chart_analysis(st.session_state.birth_chart)
+        
+        # Chat interface
+        st.markdown("### 💬 Ask about your chart")
+        user_question = st.text_input("Enter your question about your birth chart")
+        
+        if user_question:
+            # Add user message to chat history
+            st.session_state.chat_history.append({
+                "role": "user",
+                "content": user_question
+            })
+            
+            # Generate response (implement AI response logic here)
+            response = "This is a placeholder response. Implement actual AI response logic."
+            
+            st.session_state.chat_history.append({
+                "role": "assistant",
+                "content": response
+            })
+        
+        # Display chat history
+        for message in st.session_state.chat_history:
+            st.markdown(
+                f"""<div class="chat-message {'user-message' if message['role'] == 'user' else 'assistant-message'}">
+                    <strong>{'You' if message['role'] == 'user' else '🔮 Assistant'}:</strong><br>
+                    {message['content']}
+                </div>""",
+                unsafe_allow_html=True
+            )
+
+if __name__ == "__main__":
+    main()