Update app.py

app.py

@@ -27,7 +27,7 @@ def render_chat_history(chat_history):
     for message in chat_history:
         if(message["role"]!= "system"):
             with st.chat_message(message["role"]):
-                st.write(message["content"])
+                st.markdown(message["content"])
 
 #Application 
 with st.container():
@@ -46,17 +46,18 @@ with st.container():
             },
             {"role": "assistant", "content": "Hello, how can I help you today?"},
         ] #Initialize chat history
-        message_container = st.empty()
-
+        
     if 'model' not in st.session_state:
         st.session_state.model = model
+
+    render_chat_history(st.session_state.chat_history)
     
     #Set up input text field
     input_text = st.chat_input(placeholder="Here you can chat with our hotel booking model.")
 
     if input_text:
-        #with st.chat_message("user"):
-            #st.write(input_text)
+        with st.chat_message("user"):
+            st.markdown(input_text)
         st.session_state.chat_history.append({"role" : "user", "content" : input_text}) #append message to chat history
         
         #chat_response = demo_chat.demo_chain(input_text=input_text, memory=st.session_state.memory, model= chat_model)
@@ -66,10 +67,7 @@ with st.container():
         outputs = model.generate(tokenized_chat, max_new_tokens=128) 
         first_answer = tokenizer.decode(outputs[0][tokenized_chat.shape[1]:],skip_special_tokens=True)
         
-        #with st.chat_message("assistant"):
-            #st.write(first_answer)
+        with st.chat_message("assistant"):
+            st.markdown(first_answer)
         st.session_state.chat_history.append({"role": "assistant", "content": first_answer})    
     st.markdown('</div>', unsafe_allow_html=True)
-
-with message_container:
-    render_chat_history(st.session_state.chat_history)