Update services/chart_analysis.py

services/chart_analysis.py

@@ -5,63 +5,111 @@ from datetime import datetime
 from utils.file_handlers import save_chat_history
 from utils.prompts import create_analysis_prompt
 
-class ChartAnalysisService:
-    def __init__(self, claude_service):
-        self.claude_service = claude_service
+# services/chart_analysis.py
 
-    def analyze_chart(self, image_data, patterns, indicators):
-        """Analyze chart and return results"""
+class ChartAnalysisService:
+    def analyze_multiple_charts(self, image_data_list, patterns, indicators, comparison_type="Individual Analysis"):
+        """Analyze multiple charts and provide comprehensive analysis"""
         try:
-            # First detect chart type
-            chart_type = self.claude_service.detect_chart_type(image_data)
-            st.info(f"Detected chart type: {chart_type}")
+            results = []
+            individual_analyses = []
+
+            # First get individual analyses
+            for idx, image_data in enumerate(image_data_list):
+                chart_type = self.claude_service.detect_chart_type(image_data)
+                st.info(f"Chart {idx + 1} detected as: {chart_type}")
+                
+                analysis = self.analyze_single_chart(image_data, patterns, indicators, chart_type)
+                if analysis:
+                    individual_analyses.append(analysis)
+
+            # If we have multiple charts, perform comparison analysis
+            if len(individual_analyses) > 1 and comparison_type != "Individual Analysis":
+                comparison_prompt = self.create_comparison_prompt(
+                    individual_analyses,
+                    comparison_type,
+                    patterns,
+                    indicators
+                )
+                
+                # Combine all images for the comparison analysis
+                comparative_analysis = self.claude_service.analyze_multiple_images(
+                    image_data_list,
+                    comparison_prompt
+                )
+                
+                if comparative_analysis:
+                    results.append({
+                        'timestamp': datetime.now().isoformat(),
+                        'analysis_type': comparison_type,
+                        'analysis': comparative_analysis,
+                        'num_charts': len(image_data_list)
+                    })
+
+            # Add individual analyses
+            results.extend(individual_analyses)
+            return results
+        except Exception as e:
+            st.error(f"Error in multiple chart analysis: {str(e)}")
+            return None
 
-            # Generate analysis prompt
+    def analyze_single_chart(self, image_data, patterns, indicators, chart_type=None):
+        """Analyze a single chart"""
+        try:
             prompt = create_analysis_prompt(patterns, indicators)
-            
-            # Get analysis from Claude
             analysis = self.claude_service.analyze_image(image_data, prompt)
             
             if analysis:
                 return {
                     'timestamp': datetime.now().isoformat(),
                     'chart_type': chart_type,
-                    'analysis': analysis
+                    'analysis': analysis,
+                    'analysis_type': 'Individual'
                 }
             return None
         except Exception as e:
             st.error(f"Error in chart analysis: {str(e)}")
             return None
 
-    def handle_follow_up_question(self, question, previous_analysis, image_data):
-        """Handle follow-up questions about the analysis"""
-        try:
-            response = self.claude_service.continue_analysis(
-                question,
-                previous_analysis,
-                image_data
-            )
+    def create_comparison_prompt(self, individual_analyses, comparison_type, patterns, indicators):
+        """Create a prompt for comparing multiple charts"""
+        if comparison_type == "Correlated Analysis":
+            return """Analyze these charts together and provide insights on their relationships:
+            1. Correlation Analysis
+            - Identify any price correlations
+            - Note common patterns or divergences
+            - Highlight synchronized movements
             
-            if response:
-                return {
-                    'timestamp': datetime.now().isoformat(),
-                    'question': question,
-                    'analysis': response
-                }
-            return None
-        except Exception as e:
-            st.error(f"Error handling follow-up question: {str(e)}")
-            return None
-
-    def extract_stock_info(self, analysis_text):
-        """Extract stock name and other metadata from analysis"""
-        stock_name = "Unknown"
-        try:
-            if "analyzing" in analysis_text.lower():
-                words = analysis_text.split()
-                for i, word in enumerate(words):
-                    if word.lower() == "analyzing":
-                        stock_name = words[i + 1].strip("(),.:")
-        except:
-            pass
-        return stock_name
+            2. Relative Strength Comparison
+            - Compare relative performance
+            - Identify leader/laggard relationships
+            - Note any rotation patterns
+            
+            3. Market Implications
+            - What these relationships suggest
+            - Potential trading opportunities
+            - Risk considerations
+            
+            4. Summary
+            Provide a clear summary of the key relationships and trading implications.
+            """
+        else:  # Market Trend Analysis
+            return """Analyze these charts to identify broader market patterns:
+            1. Market Trend Analysis
+            - Identify common trend directions
+            - Note sector/market momentum
+            - Highlight any broader pattern formations
+            
+            2. Market Condition Assessment
+            - Overall market health
+            - Risk levels
+            - Potential market turning points
+            
+            3. Trading Implications
+            - Sector rotation opportunities
+            - Risk management considerations
+            - Timeframe recommendations
+            
+            4. Summary
+            Provide a clear summary of the market condition and trading implications.
+            """