Update app.py

app.py

@@ -52,6 +52,7 @@ def reset_conversation():
     '''
     st.session_state.conversation = []
     st.session_state.messages = []
+    save_conversation_history([])  
     return None
 
 def load_conversation_history():
@@ -86,6 +87,9 @@ if st.session_state.prev_option != selected_model:
     st.session_state.messages = []
     st.session_state.prev_option = selected_model
 
+if "conversation" not in st.session_state:
+    st.session_state.conversation = []
+
 reset_conversation()
 
 repo_id = model_links[selected_model]
@@ -122,4 +126,4 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, How can I help you today?"
         st.session_state.messages.append({"role": "assistant", "content": response})
         
         # Save the updated conversation history to the file
-        save_conversation_history(st.session_state.messages)
+        save_conversation_history(st.session_state.messages)