Update app.py

app.py

@@ -288,15 +288,13 @@ def page1():
                 new_messages = st.session_state['chat_history_page1'][-2:]
                 for chat in new_messages:
                     try:
-                        # Diagnostic print: Display each chat entry
-                        st.text(f"Chat entry: {chat}")
-
-                        # Ensure tuple has two elements
-                        if len(chat) != 2:
+                        # Ensure the tuple has at least two elements
+                        if len(chat) < 2:
                             raise ValueError(f"Invalid chat tuple: {chat}")
-
-                        speaker, message = chat
-                        background_color = "#ffeecf" if chat[2] == "new" else "#ffeecf" if speaker == "User" else "#ffeecf"
+                
+                        speaker, message = chat[:2]
+                        status = chat[2] if len(chat) > 2 else "old"  # Default to "old" if the third element is missing
+                        background_color = "#ffeecf" if status == "new" else "#ffeecf" if speaker == "User" else "#ffeecf"
                         new_messages_placeholder.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{speaker}: {message}</div>", unsafe_allow_html=True)
                     except IndexError as e:
                         st.error(f"Error processing chat entry: {e}")