Update app.py

app.py

@@ -139,4 +139,5 @@ if user_prompt and (uploaded_file or video_url):
     with st.spinner("Thinking..."):
         response = handle_query(user_prompt)
     with st.chat_message("user", avatar="robot.png"):
-        st.write_stream(streamer(response))
+        st.write_stream(streamer(response))
+    st.session_state.messages.append({'role': 'assistant', "content": response})