Update app_chat.py

app_chat.py

@@ -48,6 +48,8 @@ def app_chat(agent_config):
         # Handle submission with the selected inference URL
         #app_agent_config()
 
+        response = ""
+        chat_response = ""
         with st.spinner('Please stand by ...'):
 
             response = handle_submission(user_message, selected_tools, agent_config.url_endpoint, agent_config.document, agent_config.image, agent_config.context)
@@ -64,7 +66,7 @@ def app_chat(agent_config):
                 st.image(response)
             elif isinstance(response, AudioSegment):
                 agent_config.audio = response
-                chat_respone = handle_submission_chat(user_message, "No context . Created audio file.")
+                chat_respone = handle_submission_chat(user_message, "Agent Tool created audio file.")
                 st.write(chat_respone)
                 st.audio(response)
             elif isinstance(response, int): 
@@ -73,7 +75,7 @@ def app_chat(agent_config):
                 st.markdown(response)
             elif isinstance(response, str):
                 if "emojified_text" in response: 
-                    chat_respone = handle_submission_chat(user_message, "No context . Created the text with emojies.")
+                    chat_respone = handle_submission_chat(user_message, "Agent Tool created the text with emojies.")
                     st.write(chat_respone)
                     st.markdown(f"{response['emojified_text']}")
                 else:
@@ -81,30 +83,42 @@ def app_chat(agent_config):
                     st.write(chat_respone)
                     st.markdown(response)
             elif isinstance(response, list):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a list")
                 for item in response:
                     st.markdown(item)  # Assuming the list contains strings
             elif isinstance(response, pd.DataFrame):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a pd.DataFrame")
                 st.dataframe(response)
             elif isinstance(response, pd.Series):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a pd.Series")
                 st.table(response.iloc[0:10])
             elif isinstance(response, dict):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a dict")
                 st.json(response)
             elif isinstance(response, st.graphics_altair.AltairChart):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a st.graphics_altair.AltairChart")
                 st.altair_chart(response)
             elif isinstance(response, st.graphics_bokeh.BokehChart):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a st.graphics_bokeh.BokehChart")
                 st.bokeh_chart(response)
             elif isinstance(response, st.graphics_graphviz.GraphvizChart):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a st.graphics_graphviz.GraphvizChart")
                 st.graphviz_chart(response)
             elif isinstance(response, st.graphics_plotly.PlotlyChart):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a st.graphics_plotly.PlotlyChart")
                 st.plotly_chart(response)
             elif isinstance(response, st.graphics_pydeck.PydeckChart):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a  st.graphics_pydeck.PydeckChart")
                 st.pydeck_chart(response)
             elif isinstance(response, matplotlib.figure.Figure):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a matplotlib.figure.Figure")
                 st.pyplot(response)
             elif isinstance(response, streamlit.graphics_vega_lite.VegaLiteChart):
+                chat_respone = handle_submission_chat(user_message, "Agent Tool produced a VegaLiteChart")
                 st.vega_lite_chart(response)
             else:
                 st.warning("Unrecognized response type. Please try again. e.g. Generate an image of a flying horse.")
     
+        st.session_state.messages.append({"role": "assistant", "content": chat_response})
         st.session_state.messages.append({"role": "assistant", "content": response})