Update app.py

app.py

@@ -18,7 +18,7 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Custom CSS for a more modern look
+# Custom CSS for a modern look
 st.markdown("""
     <style>
     .stApp {
@@ -37,6 +37,9 @@ st.markdown("""
         padding: 0.5rem 1rem;
         border: none;
     }
+    .stButton > button:hover {
+        background-color: #553C9A;
+    }
     .chat-message {
         padding: 1.5rem;
         border-radius: 10px;
@@ -53,9 +56,213 @@ st.markdown("""
         margin-right: 20%;
         border-left: 4px solid #9F7AEA;
     }
+    .element-card {
+        background-color: white;
+        padding: 1rem;
+        border-radius: 8px;
+        margin-bottom: 0.5rem;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.05);
+    }
     </style>
 """, unsafe_allow_html=True)
 
+def create_chart_visualization(chart_data: Dict[str, Any]):
+    """Create a professional astrological chart visualization"""
+    fig = go.Figure()
+    
+    # Define colors for zodiac signs (matching traditional colors)
+    zodiac_colors = {
+        'Aries': '#e8fde8',      # Light green
+        'Taurus': '#fff9e6',     # Light yellow
+        'Gemini': '#ffe6e6',     # Light pink
+        'Cancer': '#e6f3ff',     # Light blue
+        'Leo': '#e8fde8',        # Light green
+        'Virgo': '#fff9e6',      # Light yellow
+        'Libra': '#ffe6e6',      # Light pink
+        'Scorpio': '#e6f3ff',    # Light blue
+        'Sagittarius': '#e8fde8', # Light green
+        'Capricorn': '#fff9e6',   # Light yellow
+        'Aquarius': '#ffe6e6',    # Light pink
+        'Pisces': '#e6f3ff'      # Light blue
+    }
+    
+    # Create outer wheel segments (zodiac signs)
+    for i, (sign, color) in enumerate(zodiac_colors.items()):
+        start_angle = i * 30
+        end_angle = (i + 1) * 30
+        
+        # Create arc for each sign
+        theta = np.linspace(start_angle, end_angle, 31)
+        r_outer = 1.2
+        r_inner = 1.0
+        
+        x_outer = r_outer * np.cos(np.radians(theta))
+        y_outer = r_outer * np.sin(np.radians(theta))
+        x_inner = r_inner * np.cos(np.radians(theta))
+        y_inner = r_inner * np.sin(np.radians(theta))
+        
+        # Combine points to form the segment
+        x = np.concatenate([x_outer, x_inner[::-1], [x_outer[0]]])
+        y = np.concatenate([y_outer, y_inner[::-1], [y_outer[0]]])
+        
+        fig.add_trace(go.Scatter(
+            x=x, y=y,
+            fill='toself',
+            fillcolor=color,
+            line=dict(color='rgba(70, 70, 70, 0.5)', width=1),
+            hoverinfo='skip',
+            showlegend=False
+        ))
+    
+    # Add degree markings
+    for degree in range(0, 360, 5):
+        r_start = 1.1
+        r_end = 1.15 if degree % 30 == 0 else 1.13
+        angle = np.radians(degree)
+        
+        fig.add_trace(go.Scatter(
+            x=[r_start * np.cos(angle), r_end * np.cos(angle)],
+            y=[r_start * np.sin(angle), r_end * np.sin(angle)],
+            mode='lines',
+            line=dict(color='rgba(70, 70, 70, 0.5)', width=1),
+            hoverinfo='skip',
+            showlegend=False
+        ))
+        
+        if degree % 30 == 0:
+            # Add zodiac symbols at sign boundaries
+            r_symbol = 1.25
+            x = r_symbol * np.cos(angle)
+            y = r_symbol * np.sin(angle)
+            zodiac_symbols = ['♈', '♉', '♊', '♋', '♌', '♍', '♎', '♏', '♐', '♑', '♒', '♓']
+            symbol = zodiac_symbols[degree // 30]
+            
+            fig.add_annotation(
+                x=x, y=y,
+                text=symbol,
+                showarrow=False,
+                font=dict(size=20),
+                textangle=degree
+            )
+    
+    # Add house lines
+    if 'houses' in chart_data and 'cusps' in chart_data['houses']:
+        for angle in chart_data['houses']['cusps']:
+            rad = np.radians(angle)
+            fig.add_trace(go.Scatter(
+                x=[0, np.cos(rad)],
+                y=[0, np.sin(rad)],
+                mode='lines',
+                line=dict(color='rgba(70, 70, 70, 0.7)', width=1, dash='dot'),
+                hoverinfo='skip',
+                showlegend=False
+            ))
+    
+    # Add aspect grid in the center
+    if 'aspects' in chart_data:
+        for aspect in chart_data['aspects']:
+            planet1_pos = chart_data['planets'][aspect['planet1']]['degrees']
+            planet2_pos = chart_data['planets'][aspect['planet2']]['degrees']
+            
+            # Define aspect line colors
+            aspect_colors = {
+                'Conjunction': 'blue',
+                'Sextile': 'green',
+                'Square': 'red',
+                'Trine': 'purple',
+                'Opposition': 'orange'
+            }
+            
+            rad1 = np.radians(planet1_pos)
+            rad2 = np.radians(planet2_pos)
+            
+            # Draw aspect line
+            r_aspect = 0.4  # Size of aspect grid
+            fig.add_trace(go.Scatter(
+                x=[r_aspect * np.cos(rad1), r_aspect * np.cos(rad2)],
+                y=[r_aspect * np.sin(rad1), r_aspect * np.sin(rad2)],
+                mode='lines',
+                line=dict(
+                    color=aspect_colors.get(aspect['aspect'], 'gray'),
+                    width=1
+                ),
+                hoverinfo='text',
+                hovertext=f"{aspect['planet1']} {aspect['aspect']} {aspect['planet2']} ({aspect['orb']}°)",
+                showlegend=False
+            ))
+    
+    # Add planets with proper symbols
+    planet_symbols = {
+        'Sun': '☉',
+        'Moon': '☽',
+        'Mercury': '☿',
+        'Venus': '♀',
+        'Mars': '♂',
+        'Jupiter': '♃',
+        'Saturn': '♄',
+        'Uranus': '⛢',
+        'Neptune': '♆',
+        'Pluto': '♇',
+        'North Node': '☊',
+        'Chiron': '⚷'
+    }
+    
+    for planet, data in chart_data['planets'].items():
+        if 'error' not in data:
+            angle = np.radians(data['degrees'])
+            r = 0.8  # Planet placement radius
+            
+            symbol = planet_symbols.get(planet, planet[:2])
+            
+            # Add planet symbol
+            fig.add_annotation(
+                x=r * np.cos(angle),
+                y=r * np.sin(angle),
+                text=symbol,
+                showarrow=False,
+                font=dict(size=16),
+                hovertext=f"{planet} in {data['sign']} ({data['degrees']:.1f}°)",
+                hoverlabel=dict(bgcolor='white')
+            )
+            
+            # Add degree marker
+            deg_text = f"{data['degrees']:.0f}°"
+            r_deg = 0.9
+            fig.add_annotation(
+                x=r_deg * np.cos(angle),
+                y=r_deg * np.sin(angle),
+                text=deg_text,
+                showarrow=False,
+                font=dict(size=8),
+                textangle=data['degrees']
+            )
+    
+    # Update layout
+    fig.update_layout(
+        showlegend=False,
+        xaxis=dict(
+            range=[-1.3, 1.3],
+            showgrid=False,
+            zeroline=False,
+            showticklabels=False
+        ),
+        yaxis=dict(
+            range=[-1.3, 1.3],
+            showgrid=False,
+            zeroline=False,
+            showticklabels=False,
+            scaleanchor="x",
+            scaleratio=1
+        ),
+        plot_bgcolor='rgba(0,0,0,0)',
+        paper_bgcolor='rgba(0,0,0,0)',
+        margin=dict(l=20, r=20, t=20, b=20),
+        width=800,
+        height=800
+    )
+    
+    return fig
+
 def init_session_state():
     """Initialize session state variables"""
     if 'birth_chart' not in st.session_state:
@@ -94,6 +301,51 @@ def get_location_details(location_name: str) -> Dict[str, Any]:
     except Exception as e:
         return {'error': f"Error finding location: {str(e)}"}
 
+def display_chart_details(chart_data: Dict[str, Any]):
+    """Display detailed chart information"""
+    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("### Aspects")
+        if 'aspects' in chart_data:
+            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)
+    
+    with col3:
+        st.markdown("### Elements & Modalities")
+        if 'patterns' in chart_data:
+            st.markdown("#### Element Balance")
+            element_data = pd.DataFrame(
+                chart_data['patterns']['elements'], 
+                index=['Count']
+            ).T
+            st.bar_chart(element_data)
+            
+            st.markdown("#### Modality Balance")
+            modality_data = pd.DataFrame(
+                chart_data['patterns']['modalities'], 
+                index=['Count']
+            ).T
+            st.bar_chart(modality_data)
+
 def main():
     st.title("🌟 AI Astrology Assistant")
     
@@ -125,9 +377,12 @@ def main():
                 placeholder="City, Country"
             )
         
-        submit = st.form_submit_button("Calculate Birth Chart")
+        calculate_button = st.form_submit_button(
+            "Calculate Birth Chart",
+            use_container_width=True
+        )
         
-        if submit:
+        if calculate_button:
             if not birth_place:
                 st.error("Please enter your birth place.")
             else:
@@ -160,49 +415,68 @@ def main():
 
     # Display chart and analysis if available
     if st.session_state.birth_chart and 'error' not in st.session_state.birth_chart:
-        tab1, tab2, tab3 = st.tabs(["Chart Visualization", "Technical Details", "AI Analysis"])
+        tab1, tab2, tab3 = st.tabs(["Birth Chart", "Technical Details", "AI Analysis"])
         
         with tab1:
-            st.plotly_chart(
-                create_chart_visualization(st.session_state.birth_chart),
-                use_container_width=True
-            )
+            st.markdown("### Your Birth Chart")
+            chart_col1, chart_col2 = st.columns([2, 1])
+            
+            with chart_col1:
+                st.plotly_chart(
+                    create_chart_visualization(st.session_state.birth_chart),
+                    use_container_width=True
+                )
+            
+            with chart_col2:
+                st.markdown("### Quick Overview")
+                st.markdown("#### Rising Sign (Ascendant)")
+                if 'houses' in st.session_state.birth_chart:
+                    rising_sign = st.session_state.birth_chart['houses']['ascendant']['sign']
+                    st.markdown(f"Your Ascendant is in **{rising_sign}**")
+                
+                st.markdown("#### Sun Sign")
+                sun_data = st.session_state.birth_chart['planets']['Sun']
+                st.markdown(f"Your Sun is in **{sun_data['sign']}**")
+                
+                st.markdown("#### Moon Sign")
+                moon_data = st.session_state.birth_chart['planets']['Moon']
+                st.markdown(f"Your Moon is in **{moon_data['sign']}**")
         
         with tab2:
-            st.subheader("Planetary Positions")
-            for planet, data in st.session_state.birth_chart['planets'].items():
-                if 'error' not in data:
-                    st.markdown(f"""
-                        <div style='background-color: white; padding: 1rem; border-radius: 8px; margin-bottom: 0.5rem;'>
-                            <strong>{planet}</strong>: {data['degrees']:.1f}° {data['sign']} in House {data.get('house', 'Unknown')}
-                            {' 🔄' if data.get('retrograde', False) else ''}
-                        </div>
-                    """, unsafe_allow_html=True)
-            
+            display_chart_details(st.session_state.birth_chart)
+        
         with tab3:
             if st.session_state.initial_analysis:
+                st.markdown("### 🌟 Initial Chart Analysis")
                 st.markdown(st.session_state.initial_analysis)
         
         # Chat interface
+        st.markdown("---")
         st.markdown("### 💬 Ask about your chart")
         
         # Suggested questions
         questions = [
-            "What are the main themes in my chart?",
-            "What are my career strengths?",
-            "How does my chart influence relationships?",
-            "What are my biggest challenges?",
-            "What are my spiritual or creative gifts?"
+            "",
+            "What are the main themes in my birth chart?",
+            "What are my career strengths and potential paths?",
+            "How does my chart influence my relationships?",
+            "What are my biggest challenges and growth opportunities?",
+            "What are my spiritual or creative gifts?",
+            "How can I best work with my current planetary placements?",
+            "What areas of life are most emphasized in my chart?",
+            "How do my Sun, Moon, and Rising signs work together?",
+            "What are the most significant aspects in my chart?"
         ]
         
         selected_question = st.selectbox(
             "Choose a question or type your own below:",
-            [""] + questions
+            questions,
+            label_visibility="collapsed"
         )
         
         user_question = st.text_input(
             "Your question:",
-            value=selected_question,
+            value=selected_question if selected_question != "" else "",
             key="user_input"
         )