Update components/chat.py

components/chat.py

@@ -15,6 +15,24 @@ def ensure_embeddings_initialized():
             return False
     return True
 
+def format_session_date(session_id: str) -> str:
+    """Format session ID into readable date."""
+    try:
+        # Handle different possible date formats
+        if '_' in session_id:
+            date_part = session_id.split('_')[1]
+        else:
+            date_part = session_id
+            
+        if len(date_part) == 8:  # Format: YYYYMMDD
+            return datetime.strptime(date_part, '%Y%m%d').strftime('%B %d, %Y')
+        elif len(date_part) == 14:  # Format: YYYYMMDD_HHMMSS
+            return datetime.strptime(date_part, '%Y%m%d%H%M%S').strftime('%B %d, %Y %I:%M %p')
+        else:
+            return date_part  # Return original if format unknown
+    except Exception as e:
+        return f"Session: {session_id}"  # Fallback display
+
 def clean_ai_response(content):
     """Clean up AI response content and remove technical artifacts."""
     content = str(content)
@@ -203,7 +221,7 @@ def display_chat_interface():
                 selected_session = st.selectbox(
                     "Select a previous conversation:",
                     options=[s['session_id'] for s in sessions],
-                    format_func=lambda x: f"Chat from {datetime.strptime(x.split('_')[1], '%Y%m%d_%H%M%S').strftime('%B %d, %Y %I:%M %p')}"
+                    format_func=lambda x: f"Chat from {format_session_date(x)}"
                 )
                 
                 if selected_session and st.button("Load Selected Chat"):
@@ -230,10 +248,7 @@ def display_chat_interface():
         
         # Display current session info if available
         if 'current_session_id' in st.session_state:
-            session_date = datetime.strptime(
-                st.session_state.current_session_id.split('_')[1], 
-                '%Y%m%d_%H%M%S'
-            ).strftime('%B %d, %Y %I:%M %p')
+            session_date = format_session_date(st.session_state.current_session_id)
             st.markdown(f"""
                 <div class="session-info">
                     📅 Current Session: {session_date}