Update app.py

app.py

@@ -27,6 +27,9 @@ if "messages" not in st.session_state:
 if "auth" not in st.session_state:
     st.session_state.auth = False
 
+if "chart" not in st.session_state:
+    st.session_state.chart = None
+
 if st.session_state.auth:
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
@@ -44,6 +47,11 @@ if st.session_state.auth:
             typed_response += chunk
             placeholder.markdown(typed_response)
             time.sleep(0.01) 
+        
+        if st.session_state.chart:
+            st.chat_message("assistant").plotly_chart(
+                st.session_state.chart
+            )
 
         st.session_state.messages.append({"role": "assistant", "content": typed_response})
 else: