Update pages/previous_chats.py

pages/previous_chats.py

@@ -3,6 +3,21 @@
 import streamlit as st
 from datetime import datetime
 
+def format_timestamp(timestamp_str):
+    """Format timestamp for display, handling both ISO and custom formats"""
+    try:
+        # Try parsing as ISO format
+        dt = datetime.fromisoformat(timestamp_str)
+        return dt.strftime('%Y-%m-%d %H:%M')
+    except ValueError:
+        # If not ISO format, try parsing the custom format
+        try:
+            dt = datetime.strptime(timestamp_str, "%Y%m%d_%H%M%S")
+            return dt.strftime('%Y-%m-%d %H:%M')
+        except ValueError:
+            # If all parsing fails, return the original string
+            return timestamp_str
+
 def show_previous_chats_tab(storage_manager):
     """Display previous chats tab"""
     st.title("📚 Previous Analysis Sessions")
@@ -16,7 +31,10 @@ def show_previous_chats_tab(storage_manager):
         
     # Display chats
     for chat in chats:
-        with st.expander(f"Session from {datetime.fromisoformat(chat['timestamp']).strftime('%Y-%m-%d %H:%M')}"):
+        # Use formatted timestamp or fall back to timestamp
+        display_time = format_timestamp(chat.get('formatted_timestamp', chat.get('timestamp', 'No date')))
+        
+        with st.expander(f"Session from {display_time}"):
             # Show associated images if any
             if 'images' in chat:
                 cols = st.columns(min(len(chat['images']), 3))
@@ -24,39 +42,11 @@ def show_previous_chats_tab(storage_manager):
                     cols[idx % 3].image(img_data, caption=f"Chart {idx + 1}")
             
             # Show analysis history
-            for entry in chat.get('chat_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("---")
-            
-            # Add option to continue analysis
-            if st.button("Continue This Analysis", key=f"continue_{chat['id']}"):
-                # Load this chat's context into current session
-                st.session_state.chat_history = chat.get('chat_history', [])
-                st.session_state.current_analysis = chat.get('chat_history', [])[-1]['analysis']
-                if 'images' in chat:
-                    st.session_state.current_images = chat['images']
-                st.success("Context loaded! Switch to Chart Analysis tab to continue.")
-
-            # Add export option
-            if st.button("Export Analysis", key=f"export_{chat['id']}"):
-                # Create markdown export
-                export_text = f"# Analysis Session {chat['timestamp']}\n\n"
-                for entry in chat.get('chat_history', []):
-                    if entry.get('analysis_type'):
-                        export_text += f"\n## {entry['analysis_type']}\n"
-                    if entry.get('question'):
-                        export_text += f"\n### Question: {entry['question']}\n"
-                    export_text += f"\n{entry.get('analysis', '')}\n"
-                
-                # Provide download button
-                st.download_button(
-                    "Download Analysis",
-                    export_text,
-                    file_name=f"analysis_{chat['id']}.md",
-                    mime="text/markdown"
-                )
+                st.markdown("---")