Update app_chat.py

app_chat.py

@@ -46,10 +46,12 @@ def app_chat(agent_config):
         # Handle submission with the selected inference URL
         #app_agent_config()
 
-        response = handle_submission(user_message, selected_tools, agent_config.url_endpoint, agent_config.document, agent_config.image)
+        response = handle_submission(user_message, selected_tools, agent_config.url_endpoint, agent_config.document, agent_config.image, agent_config.context)
     
         with st.chat_message("assistant"):
             if response is None:
+                chat_respone = handle_submission_chat(user_message, response)
+                st.write(chat_respone)
                 st.warning("The agent's response is None. Please try again. Generate an image of a flying horse.")
             elif isinstance(response, Image.Image):
                 st.image(response)
@@ -61,6 +63,8 @@ def app_chat(agent_config):
                 if "emojified_text" in response:
                     st.markdown(f"{response['emojified_text']}")
                 else:
+                    chat_respone = handle_submission_chat(user_message, response)
+                    st.write(chat_respone)
                     st.markdown(response)
             elif isinstance(response, list):
                 for item in response: