Update app.py

app.py

@@ -4,6 +4,7 @@ import pytz
 from geopy.geocoders import Nominatim
 from timezonefinder import TimezoneFinder
 from astro_core import ChartCalculator
+from ai_interpreter import AstroAI
 import plotly.graph_objects as go
 import numpy as np
 import pandas as pd
@@ -36,27 +37,21 @@ st.markdown("""
         padding: 0.5rem 1rem;
         border: none;
     }
-    .stButton > button:hover {
-        background-color: #553C9A;
-    }
     .chat-message {
-        padding: 1rem;
-        border-radius: 8px;
-        margin-bottom: 0.5rem;
+        padding: 1.5rem;
+        border-radius: 10px;
+        margin-bottom: 1rem;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.05);
     }
     .user-message {
         background-color: rgba(107, 70, 193, 0.1);
         margin-left: 20%;
+        border-left: 4px solid #6B46C1;
     }
     .assistant-message {
         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);
+        border-left: 4px solid #9F7AEA;
     }
     </style>
 """, unsafe_allow_html=True)
@@ -69,8 +64,14 @@ def init_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
+    if 'ai_interpreter' not in st.session_state:
+        try:
+            st.session_state.ai_interpreter = AstroAI()
+        except ValueError as e:
+            st.error(str(e))
+            st.session_state.ai_interpreter = None
+    if 'initial_analysis' not in st.session_state:
+        st.session_state.initial_analysis = None
 
 def get_location_details(location_name: str) -> Dict[str, Any]:
     """Get coordinates and timezone for a location"""
@@ -93,104 +94,17 @@ def get_location_details(location_name: str) -> Dict[str, Any]:
     except Exception as e:
         return {'error': f"Error finding location: {str(e)}"}
 
-def create_chart_visualization(chart_data: Dict[str, Any]):
-    """Create an interactive circular chart visualization"""
-    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,
-            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()
     
+    # Show API key status
+    if st.session_state.ai_interpreter is None:
+        st.warning("⚠️ Claude API key not found. Please add CLAUDE_API_KEY to your environment variables or Hugging Face secrets.")
+        return
+
     # Birth Information Form
     with st.form("birth_info"):
         st.subheader("Enter Birth Details")
@@ -237,12 +151,16 @@ def main():
                         if 'error' in st.session_state.birth_chart:
                             st.error(st.session_state.birth_chart['error'])
                         else:
-                            st.success("Birth chart calculated successfully!")
-    
+                            # Generate initial AI analysis
+                            with st.spinner("Generating astrological insights..."):
+                                st.session_state.initial_analysis = st.session_state.ai_interpreter.get_initial_analysis(
+                                    st.session_state.birth_chart
+                                )
+                            st.success("Birth chart calculated and analyzed 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"])
+        tab1, tab2, tab3 = st.tabs(["Chart Visualization", "Technical Details", "AI Analysis"])
         
         with tab1:
             st.plotly_chart(
@@ -251,11 +169,42 @@ def main():
             )
         
         with tab2:
-            display_chart_analysis(st.session_state.birth_chart)
+            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)
+            
+        with tab3:
+            if st.session_state.initial_analysis:
+                st.markdown(st.session_state.initial_analysis)
         
         # Chat interface
         st.markdown("### 💬 Ask about your chart")
-        user_question = st.text_input("Enter your question about your birth 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?"
+        ]
+        
+        selected_question = st.selectbox(
+            "Choose a question or type your own below:",
+            [""] + questions
+        )
+        
+        user_question = st.text_input(
+            "Your question:",
+            value=selected_question,
+            key="user_input"
+        )
         
         if user_question:
             # Add user message to chat history
@@ -264,8 +213,12 @@ def main():
                 "content": user_question
             })
             
-            # Generate response (implement AI response logic here)
-            response = "This is a placeholder response. Implement actual AI response logic."
+            # Get AI interpretation
+            with st.spinner("Generating interpretation..."):
+                response = st.session_state.ai_interpreter.get_interpretation(
+                    st.session_state.birth_chart,
+                    user_question
+                )
             
             st.session_state.chat_history.append({
                 "role": "assistant",
@@ -273,14 +226,16 @@ def main():
             })
         
         # 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 st.session_state.chat_history:
+            st.markdown("#### Previous Interpretations")
+            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()