app.py

app.py

@@ -11,11 +11,6 @@ st.caption("Chat with an Ai Assistant on your Pathology Queries")
 with st.sidebar:
     OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
 
-# Clear chat button
-if st.sidebar.button("Clear Chat"):
-    st.session_state.messages = []
-    st.rerun()
-
 if OPENAI_API_KEY:
     client = OpenAI(api_key=OPENAI_API_KEY)
 else:
@@ -28,6 +23,11 @@ ASSISTANT_ID = "asst_Iu6cCU4CyugmAD2yPgktyivH"
 if "messages" not in st.session_state:
     st.session_state["messages"] = []
 
+# Clear chat button above chat input
+if st.button("Clear Chat", use_container_width=True):
+    st.session_state.messages = []
+    st.rerun()
+
 # Display chat history
 for message in st.session_state.messages:
     role, content = message["role"], message["content"]