3

app.py

@@ -38,8 +38,6 @@ if user_input:
     st.session_state.past.append(user_input)
     st.session_state.generated.append(output)
 
-if st.session_state["generated"]:
-
-    for i in range(len(st.session_state["generated"]) - 1, -1, -1):
-        message(st.session_state["generated"][i], key=str(i))
-        message(st.session_state["past"][i], is_user=True, key=str(i) + "_user")
+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")