Update src/streamlit_app.py

src/streamlit_app.py

@@ -105,11 +105,6 @@ except KeyError as e:
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Display chat messages from history on app rerun
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
-
 with st.sidebar:
     instruction = st.selectbox(
         "What's the task you want this LLM to do?",
@@ -121,6 +116,10 @@ with st.sidebar:
         st.session_state.messages = []
         st.rerun()
 
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
 # Accept user input
 if prompt := st.chat_input("Enter a question based on a selected task."):