Update app.py

app.py

@@ -52,21 +52,7 @@ def reset_conversation():
     '''
     st.session_state.conversation = []
     st.session_state.messages = []
-    return None
-
-def load_conversation_history():
-    history_file = "conversation_history.pickle"
-    if os.path.exists(history_file):
-        with open(history_file, "rb") as f:
-            conversation_history = pickle.load(f)
-    else:
-        conversation_history = []
-    return conversation_history
-
-def save_conversation_history(conversation_history):
-    history_file = "conversation_history.pickle"
-    with open(history_file, "wb") as f:
-        pickle.dump(conversation_history, f)
+    st.experimental_rerun()  # Add this line to rerun the app after reset
 
 models = [key for key in model_links.keys()]
 selected_model = st.sidebar.selectbox("Select Model", models)
@@ -86,14 +72,13 @@ if st.session_state.prev_option != selected_model:
     st.session_state.messages = []
     st.session_state.prev_option = selected_model
 
-reset_conversation()
+# Load the conversation history from the file
+if "messages" not in st.session_state:
+    st.session_state.messages = load_conversation_history()
 
 repo_id = model_links[selected_model]
 st.subheader(f'{selected_model}')
 
-# Load the conversation history from the file
-st.session_state.messages = load_conversation_history()
-
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])