Update app.py

app.py

@@ -34,6 +34,15 @@ def init_session_state():
         st.session_state.current_conversation_id = None
     if 'analysis_results' not in st.session_state:
         st.session_state.analysis_results = []
+        
+def reset_session_state():
+    """Reset session state for a new chat"""
+    st.session_state.chat_history = []
+    st.session_state.current_image = None
+    st.session_state.current_analysis = None
+    st.session_state.current_images = []
+    st.session_state.analysis_results = []
+    st.session_state.conversation_context = []
 
 def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
     """Display chart analysis functionality"""
@@ -41,7 +50,26 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
     expertise_level = create_expertise_selector()
     
     # Create sidebar and get inputs
-    upload_option, uploaded_files, patterns, indicators, comparison_type = create_sidebar()
+    result = create_sidebar()
+    
+    # Check if we need to start a new chat
+    if result[0] == "new_chat":
+        # Save current chat if there's content
+        if st.session_state.chat_history:
+            storage_manager.save_chat(
+                st.session_state.chat_history,
+                st.session_state.current_images,
+                st.session_state.get('last_saved_chat'),
+                claude_service  # Pass claude_service for smart naming
+            )
+            st.success("Previous chat saved! Starting new chat...")
+        
+        # Reset session state
+        reset_session_state()
+        st.rerun()
+    
+    # Unpack the sidebar results
+    upload_option, uploaded_files, patterns, indicators, comparison_type = result
     
     # Main content area
     col1, col2 = st.columns([2, 1])
@@ -103,8 +131,13 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
                                 if response:
                                     st.session_state.chat_history.append(response)
                                     st.markdown(response['analysis'])
-                                    # Save to storage
-                                    storage_manager.save_chat(st.session_state.chat_history)
+                                    # Save to storage with smart naming
+                                    storage_manager.save_chat(
+                                        st.session_state.chat_history,
+                                        st.session_state.current_images,
+                                        None,
+                                        claude_service
+                                    )
             
             # Then show individual analyses
             individual_analyses = [r for r in results if r.get('analysis_type') == 'Individual']
@@ -126,8 +159,13 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
                             if response:
                                 st.session_state.chat_history.append(response)
                                 st.markdown(response['analysis'])
-                                # Save to storage
-                                storage_manager.save_chat(st.session_state.chat_history)
+                                # Save to storage with smart naming
+                                storage_manager.save_chat(
+                                    st.session_state.chat_history,
+                                    st.session_state.current_images,
+                                    None,
+                                    claude_service
+                                )
             
             # Risk warning at the bottom
             st.warning(
@@ -145,7 +183,8 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
             filename = storage_manager.save_chat(
                 st.session_state.chat_history,
                 st.session_state.current_images[0] if st.session_state.current_images else None,
-                f"{save_name}.json" if save_name else None
+                f"{save_name}.json" if save_name else None,
+                claude_service
             )
             if filename:
                 st.success(f"Chat saved as {filename}")