Update app.py

app.py

@@ -132,8 +132,8 @@ if "messages" not in st.session_state:
 
 # 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.chat_message(message[0]):
+        st.markdown(message[1])
 
 # Accept user input
 if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):