Update storage/storage_manager.py

storage/storage_manager.py

@@ -19,15 +19,19 @@ class UserStorageManager:
         except Exception as e:
             st.error(f"Error creating directories: {str(e)}")
 
-    def save_chat(self, chat_data, images=None, filename=None):
-        """Save chat with associated images"""
+    def save_chat(self, chat_data, images=None, filename=None, claude_service=None):
+        """Save chat with associated images and smart naming"""
         try:
             timestamp = datetime.now().isoformat()
             formatted_timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-            chat_id = f"chat_{formatted_timestamp}"
             
-            if filename:
-                chat_id = filename.replace('.json', '')
+            # Generate smart name if available
+            chat_title = None
+            if claude_service and isinstance(chat_data, list):
+                from utils.smart_naming import generate_chat_name
+                chat_title = generate_chat_name(chat_data, claude_service)
+            
+            chat_id = filename or f"chat_{formatted_timestamp}"
             
             # Save chat data
             chat_file = self.paths["chats"] / f"{chat_id}.json"
@@ -37,36 +41,9 @@ class UserStorageManager:
                 'id': chat_id,
                 'timestamp': timestamp,
                 'formatted_timestamp': formatted_timestamp,
+                'title': chat_title or f"Trading Analysis {formatted_timestamp}",
                 'data': chat_data
             }
-            
-            with open(chat_file, "w") as f:
-                json.dump(chat_metadata, f)
-            
-            # Save associated images
-            if images:
-                image_dir = self.paths["images"] / chat_id
-                image_dir.mkdir(exist_ok=True)
-                
-                if isinstance(images, list):
-                    for idx, img_data in enumerate(images):
-                        img_file = image_dir / f"image_{idx}.png"
-                        with open(img_file, "wb") as f:
-                            f.write(img_data)
-                else:
-                    img_file = image_dir / "image_0.png"
-                    with open(img_file, "wb") as f:
-                        f.write(images)
-                
-                # Update chat data with image references
-                chat_metadata['image_paths'] = [str(p) for p in image_dir.glob("*.png")]
-                with open(chat_file, "w") as f:
-                    json.dump(chat_metadata, f)
-
-            return chat_id
-        except Exception as e:
-            st.error(f"Error saving chat: {str(e)}")
-            return None
 
     def get_all_chats(self):
         """Get all user's chats with metadata"""