Update app_chat.py

app_chat.py

@@ -55,13 +55,23 @@ def app_chat(agent_config):
                 st.write(chat_respone)
                # st.warning("The agent's response is None. Please try again. Generate an image of a flying uncormn.")
             elif isinstance(response, Image.Image):
+                agent_config.image = response
+                chat_respone = handle_submission_chat(user_message, "No context . Created an image.")
+                st.write(chat_respone)
                 st.image(response)
             elif isinstance(response, AudioSegment):
+                agent_config.audio = response
+                chat_respone = handle_submission_chat(user_message, "No context . Created audio file.")
+                st.write(chat_respone)
                 st.audio(response)
-            elif isinstance(response, int):
+            elif isinstance(response, int): 
+                chat_respone = handle_submission_chat(user_message, response)
+                st.write(chat_respone)
                 st.markdown(response)
             elif isinstance(response, str):
-                if "emojified_text" in response:
+                if "emojified_text" in response: 
+                    chat_respone = handle_submission_chat(user_message, "No context . Created the text with emojies.")
+                    st.write(chat_respone)
                     st.markdown(f"{response['emojified_text']}")
                 else:
                     chat_respone = handle_submission_chat(user_message, response)