Refactor conversation selection

gui/streamlit_chat_app.py

@@ -105,7 +105,7 @@ def exit_handler():
 for idx, conversation in enumerate(st.session_state.conversations):
  if st.sidebar.button(f"Conversation {idx + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{idx}"):
  st.session_state['selected_conversation'] = idx
- st.session_state['current_conversation'] = st.session_state.conversations[idx]
+ st.session_state['current_conversation'] = conversation
 
 if st.session_state['selected_conversation'] is not None:
  conversation_to_display = st.session_state.conversations[st.session_state['selected_conversation']]
@@ -115,4 +115,4 @@ def exit_handler():
 if conversation_to_display['generated_responses']:
  for i in range(len(conversation_to_display['generated_responses']) - 1, -1, -1):
  message(conversation_to_display["generated_responses"][i], key=f"display_generated_{i}")
- message(conversation_to_display['user_inputs'][i], is_user=True, key=f"display_user_{i}")
+ message(conversation_to_display['user_inputs'][i], is_user=True, key=f"display_user_{i}")