Update app.py

app.py

@@ -70,6 +70,10 @@ if st.session_state.selected_model != model_option:
     st.session_state.messages = []
     st.session_state.selected_model = model_option
 
+# Add a "Clear Chat" button
+if st.button("Clear Chat"):
+    st.session_state.messages = []
+    
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
     avatar = "🤖" if message["role"] == "assistant" else "🕺"