Update app.py

app.py

@@ -108,6 +108,18 @@ if st.session_state.messages[-1]["role"] == "user" and "image" in st.session_sta
                 "content": f"System: you received an image and a question: the question is: " + st.session_state.messages[-1]["content"].lower() +  f"and that is the {Imagecaptioned} write an better answer",
             }
             st.session_state.messages.append(message)
+            with st.chat_message("assistant"):
+               with st.spinner("Thinking..."):
+                 response = generate_llama2_response(prompt)
+                 placeholder = st.empty()
+                 full_response = ""
+                 for item in response:
+                   full_response += item
+                   placeholder.markdown(full_response)
+                 placeholder.markdown(full_response)
+            message = {"role": "assistant", "content": full_response}
+            st.session_state.messages.append(message)
+            
 
 # Generate a new response if the last message is not from the assistant
 if st.session_state.messages[-1]["role"] != "assistant" and not "image" in st.session_state.messages[-1]["content"].lower():