Update app.py

app.py

@@ -21,9 +21,10 @@ if "messages" not in st.session_state:
     st.session_state["messages"] = [{"role": "system", "content": system_message},
 {"role": "assistant", "content": "How can I help you?"}]
 
-# Display all previous messages
+# Display all previous messages only when needed
 for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
+    if msg["role"] != "system":  # Skip system messages
+        st.chat_message(msg["role"]).write(msg["content"])
 
 
 # Input for new prompts