Update services/chart_analysis.py

services/chart_analysis.py

@@ -28,9 +28,7 @@ class ChartAnalysisService:
             if len(individual_analyses) > 1 and comparison_type != "Individual Analysis":
                 comparison_prompt = self.create_comparison_prompt(
                     individual_analyses,
-                    comparison_type,
-                    patterns,
-                    indicators
+                    comparison_type
                 )
                 
                 # Combine all images for the comparison analysis
@@ -72,10 +70,32 @@ class ChartAnalysisService:
             st.error(f"Error in chart analysis: {str(e)}")
             return None
 
-    def create_comparison_prompt(self, individual_analyses, comparison_type, patterns, indicators):
+    def handle_follow_up_question(self, question, previous_analysis, image_data=None):
+        """Handle follow-up questions about the analysis"""
+        try:
+            response = self.claude_service.continue_analysis(
+                question=question,
+                previous_analysis=previous_analysis,
+                image_data=image_data
+            )
+            
+            if response:
+                return {
+                    'timestamp': datetime.now().isoformat(),
+                    'question': question,
+                    'analysis': response,
+                    'analysis_type': 'Follow-up'
+                }
+            return None
+        except Exception as e:
+            st.error(f"Error handling follow-up question: {str(e)}")
+            return None
+
+    def create_comparison_prompt(self, individual_analyses, comparison_type):
         """Create a prompt for comparing multiple charts"""
         if comparison_type == "Correlated Analysis":
-            return """Analyze these charts together and provide insights on their relationships:
+            prompt = """Analyze these charts together and provide insights on their relationships:
+            
             1. Correlation Analysis
             - Identify any price correlations
             - Note common patterns or divergences
@@ -92,10 +112,10 @@ class ChartAnalysisService:
             - Risk considerations
             
             4. Summary
-            Provide a clear summary of the key relationships and trading implications.
-            """
+            Provide a clear summary of the key relationships and trading implications."""
         else:  # Market Trend Analysis
-            return """Analyze these charts to identify broader market patterns:
+            prompt = """Analyze these charts to identify broader market patterns:
+            
             1. Market Trend Analysis
             - Identify common trend directions
             - Note sector/market momentum
@@ -112,5 +132,10 @@ class ChartAnalysisService:
             - Timeframe recommendations
             
             4. Summary
-            Provide a clear summary of the market condition and trading implications.
-            """
+            - Provide a clear summary of the market condition
+            - List key trading implications
+            - Note major risk factors"""
+
+        # Add disclaimer
+        prompt += "\n\nIMPORTANT: Clearly mark this as AI-generated analysis for informational purposes only."
+        return prompt