Update app.py

app.py

@@ -52,13 +52,35 @@ model_info = {
 # Random dog images for error message
 random_dog_images = ["broken_llama3.jpeg"]
 
+def reset_conversation():
+    '''
+    Resets Conversation
+    '''
+    st.session_state.conversation = []
+    st.session_state.messages = []
+    return None
+
+st.sidebar.button('Reset Chat', on_click=reset_conversation) #Reset button
+    
 def main():
     st.header('Multi-Models')
 
+    # Initialize client
+    client = initialize_client()
+
     # Sidebar for model selection and temperature
     selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
     temperature = st.sidebar.slider('Select a temperature value', 0.0, 1.0, DEFAULT_TEMPERATURE)
 
+    if "prev_option" not in st.session_state:
+        st.session_state.prev_option = selected_model
+
+    if st.session_state.prev_option != selected_model:
+        st.session_state.messages = []
+        # st.write(f"Changed to {selected_model}")
+        st.session_state.prev_option = selected_model
+        reset_conversation()
+
     st.markdown(f'_powered_ by ***:violet[{selected_model}]***')
 
     # Display model info and logo
@@ -76,9 +98,6 @@ def main():
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
-    # Initialize client
-    client = initialize_client()
-
     # Chat input and response
     if prompt := st.chat_input("Type message here..."):
         process_user_input(client, prompt, selected_model, temperature)