Update app.py

app.py

@@ -10,8 +10,11 @@ st.write('You selected', option)
 
 prompt = st.chat_input("Say something to our #CodeWars bot...")
 if prompt:
-    st.write(f"{datetime.datetime.now()} :red[{option}:] ", prompt)
-
-    response = requests.get("https://dummyjson.com/products/1").text
     
-    st.write(f"{datetime.datetime.now()} :red[Bot:] ", response)
+    with st.chat_message(option):
+         st.write(f"{datetime.datetime.now()} :red[{option}:] ", prompt)
+    
+    response = requests.get("https://dummyjson.com/products/1").text
+
+    with st.chat_message("Bot"):
+         st.write(f"{datetime.datetime.now()}", response)