Update app.py

app.py

@@ -25,8 +25,8 @@ if "formatted_chat_history" not in st.session_state:
 if "just_chat_history" not in st.session_state:
     st.session_state.just_chat_history = []
     
-for chat_index in range(0,len(st.session_state.context_chat_history)):
-    assert len(st.session_state.context_chat_history) == len(st.session_state.formatted_chat_history) == len(st.session_state.just_chat_history)
+for chat_index in range(0,len(st.session_state.formatted_chat_history)):
+    assert len(st.session_state.formatted_chat_history) == len(st.session_state.just_chat_history)
     for col, chat_history, sources_text in zip(st.columns(2, vertical_alignment="top"), [st.session_state.just_chat_history, st.session_state.formatted_chat_history], ["Current Model", "Formatted Text"]):
         chat = chat_history[chat_index]
         with col.chat_message("user"):