Update services/chart_analysis.py

services/chart_analysis.py

@@ -80,13 +80,40 @@ class ChartAnalysisService:
     def handle_follow_up_question(self, question, previous_analysis, image_data=None):
         """Handle follow-up questions about the analysis"""
         try:
+            # Get previous conversation context from session state
+            if 'conversation_context' not in st.session_state:
+                st.session_state.conversation_context = []
+            
+            # Add previous analysis to context if not already present
+            if previous_analysis and not st.session_state.conversation_context:
+                st.session_state.conversation_context.append({
+                    'role': 'assistant',
+                    'content': previous_analysis
+                })
+            
+            # Add current question to context
+            st.session_state.conversation_context.append({
+                'role': 'user',
+                'content': question
+            })
+            
+            # Create a conversation-aware prompt
+            context = "\n".join([f"{'Q' if msg['role'] == 'user' else 'A'}: {msg['content']}" 
+                               for msg in st.session_state.conversation_context])
+            
             response = self.claude_service.continue_analysis(
                 question=question,
-                previous_analysis=previous_analysis,
+                previous_analysis=context,  # Use full conversation context
                 image_data=image_data
             )
             
             if response:
+                # Add response to context
+                st.session_state.conversation_context.append({
+                    'role': 'assistant',
+                    'content': response
+                })
+                
                 return {
                     'timestamp': datetime.now().isoformat(),
                     'question': question,