Update app.py

app.py

@@ -80,9 +80,9 @@ def launch_bot():
             else:
                 with st.spinner("Thinking..."):
                     response = generate_response(prompt)
-                    st.write(response)
-        #message = {"role": "assistant", "content": response}
-        #st.session_state.messages.append(message)
+                    #st.write(response)
+        message = {"role": "assistant", "content": response}
+        st.session_state.messages.append(message)
     
 if __name__ == "__main__":
     launch_bot()