Fix button

app.py

@@ -99,6 +99,11 @@ with open("style.css") as css:
 if "messages" not in st.session_state:
     st.session_state["messages"] = []
 
+def clear_chat_history():
+    st.session_state["messages"] = []
+
+st.button('Clear Chat', on_click=clear_chat_history)
+
 def last_role_is_user():
     return len(st.session_state["messages"]) > 0 and st.session_state["messages"][-1]["role"] == "user"
     
@@ -248,22 +253,13 @@ 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)
-    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)