Update services/claude_service.py

services/claude_service.py

@@ -84,7 +84,52 @@ class ClaudeService:
         
         result = self.analyze_image(image_data, prompt)
         return result.strip() if result else "Other"
-
+    
+    def generate_diagram(self, prompt):
+        """Generate an SVG diagram using Claude"""
+        try:
+            message = self.client.messages.create(
+                model=self.model,
+                max_tokens=1000,
+                messages=[{
+                    "role": "user",
+                    "content": f"""Please create an SVG diagram based on this request: {prompt}
+                    Make the diagram clean, professional, and educational.
+                    Use standard SVG elements and ensure the viewBox is properly set.
+                    Keep colors accessible and include clear labels."""
+                }]
+            )
+            
+            response = message.content[0].text
+            
+            # Extract SVG content if present
+            if '<svg' in response and '</svg>' in response:
+                svg_start = response.find('<svg')
+                svg_end = response.find('</svg>') + 6
+                return response[svg_start:svg_end]
+            return None
+            
+        except Exception as e:
+            st.error(f"Error generating diagram: {str(e)}")
+            return None
+            
+    def generate_educational_content(self, prompt):
+        """Generate educational content using Claude"""
+        try:
+            message = self.client.messages.create(
+                model=self.model,
+                max_tokens=1500,
+                messages=[{
+                    "role": "user",
+                    "content": prompt
+                }]
+            )
+            
+            return message.content[0].text
+            
+        except Exception as e:
+            st.error(f"Error generating educational content: {str(e)}")
+            return None
     def get_educational_content(self, concept):
         """Get educational content about trading concepts"""
         try: