Update pages/previous_chats.py

pages/previous_chats.py

@@ -31,8 +31,9 @@ def show_previous_chats_tab(storage_manager):
         
     # Display chats
     for chat in chats:
-        # Use formatted timestamp or fall back to timestamp
-        display_time = format_timestamp(chat.get('formatted_timestamp', chat.get('timestamp', 'No date')))
+        # Get the timestamp from metadata
+        timestamp = chat.get('timestamp') or chat.get('formatted_timestamp', 'No date')
+        display_time = format_timestamp(timestamp)
         
         with st.expander(f"Session from {display_time}"):
             # Show associated images if any
@@ -41,12 +42,45 @@ def show_previous_chats_tab(storage_manager):
                 for idx, img_data in enumerate(chat['images']):
                     cols[idx % 3].image(img_data, caption=f"Chart {idx + 1}")
             
+            # Get the chat history from the correct location in the data structure
+            chat_history = chat.get('data', [])
+            if isinstance(chat_history, list):
+                # Direct list of chat entries
+                entries = chat_history
+            else:
+                # Nested under 'chat_history' key
+                entries = chat_history.get('chat_history', [])
+            
             # Show analysis history
-            for entry in chat.get('data', {}).get('chat_history', []):
-                if entry.get('analysis_type') in ['Individual', 'Correlated Analysis']:
-                    st.markdown("**Initial Analysis:**")
-                elif entry.get('question'):
-                    st.markdown(f"**Follow-up Question:** {entry['question']}")
-                
-                st.markdown(entry.get('analysis', ''))
-                st.markdown("---")
+            for entry in entries:
+                if isinstance(entry, dict):  # Ensure entry is a dictionary
+                    analysis_type = entry.get('analysis_type', '')
+                    if analysis_type in ['Individual', 'Correlated Analysis']:
+                        st.markdown("**Initial Analysis:**")
+                    elif 'question' in entry:
+                        st.markdown(f"**Follow-up Question:** {entry['question']}")
+                    
+                    analysis = entry.get('analysis', '')
+                    if analysis:
+                        st.markdown(analysis)
+                        st.markdown("---")
+            
+            # Add buttons for chat actions
+            col1, col2 = st.columns(2)
+            with col1:
+                if st.button("Continue Analysis", key=f"continue_{chat.get('id', '')}"):
+                    # Load this chat into current session
+                    st.session_state.chat_history = entries
+                    if len(entries) > 0:
+                        st.session_state.current_analysis = entries[-1].get('analysis', '')
+                    if 'images' in chat:
+                        st.session_state.current_images = chat['images']
+                    st.success("Analysis loaded! Switch to Chart Analysis tab to continue.")
+            
+            with col2:
+                if st.button("Delete Chat", key=f"delete_{chat.get('id', '')}"):
+                    if storage_manager.delete_chat(chat.get('id', '')):
+                        st.success("Chat deleted successfully!")
+                        st.rerun()
+                    else:
+                        st.error("Failed to delete chat.")