Update services/chart_analysis.py

services/chart_analysis.py

@@ -9,7 +9,7 @@ class ChartAnalysisService:
     def __init__(self, claude_service):
         self.claude_service = claude_service
 
-    def analyze_multiple_charts(self, image_data_list, patterns, indicators, comparison_type="Individual Analysis"):
+    def analyze_multiple_charts(self, image_data_list, patterns, indicators, comparison_type="Individual Analysis", expertise_level="Novice"):
         """Analyze multiple charts and provide comprehensive analysis"""
         try:
             results = []
@@ -20,7 +20,13 @@ class ChartAnalysisService:
                 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)
+                analysis = self.analyze_single_chart(
+                    image_data,
+                    patterns,
+                    indicators,
+                    chart_type,
+                    expertise_level
+                )
                 if analysis:
                     individual_analyses.append(analysis)
 
@@ -28,7 +34,8 @@ class ChartAnalysisService:
             if len(individual_analyses) > 1 and comparison_type != "Individual Analysis":
                 comparison_prompt = self.create_comparison_prompt(
                     individual_analyses,
-                    comparison_type
+                    comparison_type,
+                    expertise_level
                 )
                 
                 # Combine all images for the comparison analysis
@@ -52,10 +59,10 @@ class ChartAnalysisService:
             st.error(f"Error in multiple chart analysis: {str(e)}")
             return None
 
-    def analyze_single_chart(self, image_data, patterns, indicators, chart_type=None):
+    def analyze_single_chart(self, image_data, patterns, indicators, chart_type=None, expertise_level="Novice"):
         """Analyze a single chart"""
         try:
-            prompt = create_analysis_prompt(patterns, indicators)
+            prompt = create_analysis_prompt(patterns, indicators, expertise_level)
             analysis = self.claude_service.analyze_image(image_data, prompt)
             
             if analysis:
@@ -91,10 +98,28 @@ class ChartAnalysisService:
             st.error(f"Error handling follow-up question: {str(e)}")
             return None
 
-    def create_comparison_prompt(self, individual_analyses, comparison_type):
+    def create_comparison_prompt(self, individual_analyses, comparison_type, expertise_level="Novice"):
         """Create a prompt for comparing multiple charts"""
+        expertise_adjustments = {
+            "Novice": {
+                "depth": "Focus on basic relationships and clear patterns.",
+                "language": "Use simple terms and explain technical concepts."
+            },
+            "Intermediate": {
+                "depth": "Include both basic and advanced correlations.",
+                "language": "Balance technical terms with clear explanations."
+            },
+            "Expert": {
+                "depth": "Provide sophisticated multi-chart analysis.",
+                "language": "Use full technical terminology and advanced concepts."
+            }
+        }
+        
+        adj = expertise_adjustments[expertise_level]
+        
         if comparison_type == "Correlated Analysis":
-            prompt = """Analyze these charts together and provide insights on their relationships:
+            prompt = f"""Analyze these charts together and provide insights on their relationships.
+            {adj['depth']} {adj['language']}
             
             1. Correlation Analysis
             - Identify any price correlations
@@ -113,8 +138,10 @@ class ChartAnalysisService:
             
             4. Summary
             Provide a clear summary of the key relationships and trading implications."""
+            
         else:  # Market Trend Analysis
-            prompt = """Analyze these charts to identify broader market patterns:
+            prompt = f"""Analyze these charts to identify broader market patterns.
+            {adj['depth']} {adj['language']}
             
             1. Market Trend Analysis
             - Identify common trend directions
@@ -138,5 +165,4 @@ class ChartAnalysisService:
 
         # Add disclaimer
         prompt += "\n\nIMPORTANT: Clearly mark this as AI-generated analysis for informational purposes only."
-        return prompt
-        
+        return prompt