Add button

app.py

@@ -71,8 +71,6 @@ EXAMPLE_PROMPTS = [
     "Code a python function that can run merge sort on a list.",
     "Give me the character profile of a gumdrop obsessed knight in JSON.",
     "Write a rap battle between Alan Turing and Claude Shannon.",
-    "Write a sarcastic joke about spoiled milk to make my day.",
-    
 ]
 
 TITLE = "DBRX Instruct"
@@ -250,11 +248,22 @@ def handle_user_input(user_input):
         st.session_state["messages"].append({"role": "assistant", "content": response, "warning": stream_warning,"error": stream_error})
         print(st.session_state["messages"])
 
+def clear_chat_history():
+    st.session_state["messages"] = []
+
 if prompt := st.chat_input("Type a message!", max_chars=1000):
     handle_user_input(prompt)
 
+
+
 with st.sidebar:
     with st.container():
         st.title("Examples")
         for prompt in EXAMPLE_PROMPTS:
-            st.button(prompt, args=(prompt,), on_click=handle_user_input)
+            st.button(prompt, args=(prompt,), on_click=handle_user_input)
+    for _ in range(5):
+        st.write("")
+    col1, col2, col3 = st.columns([1, 2, 1])
+    with col2:
+        st.button('Clear Chat History', on_click=clear_chat_history)
+