Update src/streamlit_app.py

src/streamlit_app.py

@@ -105,22 +105,23 @@ 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"])
-
+# 💡 Move this block up so it runs BEFORE chat messages are rendered
+# Handle sidebar early to ensure proper app flow
 with st.sidebar:
     instruction = st.selectbox(
         "What's the task you want this LLM to do?",
         ("math", "SQL", "python", "medical"),
     )
 
-    # Add clear history button
+    # Handle clear history BEFORE rendering messages
     if st.button("Clear History"):
         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."):