Update app.py

app.py

@@ -64,10 +64,9 @@ st.write('You selected', option)
 
 prompt = st.chat_input("Say something to our #CodeWars bot...")
 context = [] # the context stores a conversation history, you can use this to make the model more context aware
-while True:
-    if(prompt):
-        st.write(f"{datetime.datetime.now()} :red[{option}:] ", prompt)
-        context = qachain({"query": prompt})
-        
-        st.write(f"{datetime.datetime.now()} :red[Bot:] ", context['result'])
-
+if(prompt):
+    with st.chat_message(option):
+         st.write(f"{datetime.datetime.now()} :red[{option}:] ", prompt)
+    context = qachain({"query": prompt})
+    
+    st.write(f"{datetime.datetime.now()}", context['result'])