Update services/claude_service.py

services/claude_service.py

@@ -45,6 +45,37 @@ class ClaudeService:
             st.error(f"Error in image analysis: {str(e)}")
             return None
 
+    def analyze_multiple_images(self, image_data_list, prompt):
+        """Analyze multiple images together using Claude Vision"""
+        try:
+            # Create content list with prompt and all images
+            content = [{"type": "text", "text": prompt}]
+            
+            # Add all images to the content
+            for idx, image_data in enumerate(image_data_list):
+                encoded_image = base64.b64encode(image_data).decode('utf-8')
+                content.append({
+                    "type": "image",
+                    "source": {
+                        "type": "base64",
+                        "media_type": "image/jpeg",
+                        "data": encoded_image
+                    }
+                })
+            
+            message = self.client.messages.create(
+                model=self.model,
+                max_tokens=1500,  # Increased for multiple image analysis
+                messages=[{
+                    "role": "user",
+                    "content": content
+                }]
+            )
+            
+            return message.content[0].text
+        except Exception as e:
+            st.error(f"Error in multiple image analysis: {str(e)}")
+            return None
     def detect_chart_type(self, image_data):
         """Detect chart type from image"""
         prompt = """What type of financial chart is this?