Update app.py

app.py

@@ -91,14 +91,9 @@ if not isinstance(st.session_state.messages, list):
 if not all(isinstance(msg, dict) for msg in st.session_state.messages):
     st.session_state.messages = []
 
-# Display chat messages from history on app rerun
+# Display chat messages from history on app rerun, excluding system messages
 for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
-
-# Display chat messages from history on app rerun
-for message in st.session_state.messages:
-    if message["role"] != "system":  # Skip system messages
+    if message["role"] != "system":  # Skip displaying system messages
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
@@ -108,8 +103,10 @@ if prompt := st.chat_input("😉 Ask any question or feel free to use the exampl
     # Display user message in chat message container
     st.chat_message("user").markdown(prompt)
 
-    # Add user message to chat history
+    # Add a system message to the chat history, but don't display it
     st.session_state.messages.append({"role": "system", "content": f"You are a helpful assistant. Year now is {current_year}"})
+    
+    # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
     # API Call