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()