4

app.py

@@ -38,6 +38,6 @@ if user_input:
     st.session_state.past.append(user_input)
     st.session_state.generated.append(output)
 
-for i in reversed(range(len(st.session_state["generated"]))):
-    message(st.session_state["generated"][i], key=str(i))
-    message(st.session_state["past"][i], is_user=True, key=str(i) + "_user")
+if st.session_state["generated"]:
+    for i in range(len(st.session_state["generated"]) - 1, -1, -1):
+        message(st.session_state["generated"][i], "bot")