Update app.py

app.py

@@ -1,5 +1,6 @@
 from datetime import datetime, time
 import streamlit as st
+import numpy as np
 
 
 st.title("Streamlit App at Hugging Face Spaces!")
@@ -44,3 +45,21 @@ with st.sidebar:
         value=datetime(2020, 1, 1, 9, 30),
         format="MM/DD/YY - hh:mm")
     st.write("Start time:", start_time)
+
+
+if "chat_messages" not in st.session_state:
+    st.session_state.chat_messages = []
+
+prompt = st.chat_input("Say something")
+if prompt:
+    st.session_state.chat_messages.append({"type": "user", "message": prompt})
+    st.session_state.chat_messages.append({"type": "bot", "message": "Hello!", "chart": np.random.randn(30, 3)})
+
+for message in st.session_state.chat_messages[::-1]:
+    if message["type"] == "user":
+        with st.chat_message("You"):
+            st.write(message["message"])
+    else:
+        with st.chat_message("Bot"):
+            st.write(message["message"])
+            st.line_chart(message["chart"])