fixes

view/app_chat.py

@@ -71,8 +71,13 @@ def app_chat(controller):
                 chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a dict")
                 st.write(chat_respone) 
                 st.json(response)
-            
-"""             elif isinstance(response, st.graphics_altair.AltairChart):
+            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_respone, "avatar": "🦖"})
+        st.session_state.messages.append({"role": "assistant", "content": response, "avatar": "🤖"})
+                  
+"""          elif isinstance(response, st.graphics_altair.AltairChart):
                 chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a st.graphics_altair.AltairChart")
                 st.write(chat_respone) 
                 st.altair_chart(response)
@@ -100,10 +105,5 @@ def app_chat(controller):
                 chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a VegaLiteChart")
                 st.write(chat_respone) 
                 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_respone, "avatar": "🦖"})
-        st.session_state.messages.append({"role": "assistant", "content": response, "avatar": "🤖"})
-    
+
+