Create services/chart_analysis.py

services/chart_analysis.py

@@ -0,0 +1,67 @@
+# src/services/chart_analysis.py
+
+import streamlit as st
+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
+
+    def analyze_chart(self, image_data, patterns, indicators):
+        """Analyze chart and return results"""
+        try:
+            # First detect chart type
+            chart_type = self.claude_service.detect_chart_type(image_data)
+            st.info(f"Detected chart type: {chart_type}")
+
+            # Generate analysis prompt
+            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
+                }
+            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
+            )
+            
+            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