Update app.py

app.py

@@ -54,11 +54,11 @@ booking process.
 #Check if chat history exists in this session
 if 'chat_history' not in st.session_state:
     st.session_state.chat_history = [
-        #{
-            #"role": "system",
-            #"content": system_content,
-        #},
-        {"role": "assistant", "content": "Hello, how can I help you today?"},
+        {
+            "role": "system",
+            "content": system_content,
+        },
+        {"role": "ai", "content": "Hello, how can I help you today?"},
     ] #Initialize chat history
     
 # if 'model' not in st.session_state:
@@ -72,12 +72,12 @@ if input_text := st.chat_input(placeholder="Here you can chat with our hotel boo
     
     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
+    st.session_state.chat_history.append({"role" : "human", "content" : input_text}) #append message to chat history
 
     with st.spinner("Generating response..."):
-        first_answer = demo_chat.demo_chain(input_text = input_text, history = st.session_state.chat_history)
+        first_answer = demo_chat.demo_chain(input_text = input_text, history = st.session_state.memory)
         
         with st.chat_message("assistant"):
             st.markdown(first_answer)
-        st.session_state.chat_history.append({"role": "assistant", "content": first_answer})    
+        st.session_state.chat_history.append({"role": "ai", "content": first_answer})    
 st.markdown('</div>', unsafe_allow_html=True)