Upload app.py

app.py

@@ -19,9 +19,12 @@ def load_answer(question):
 
     assistant_answer = chat(st.session_state.sessionMessages)
 
-    st.session_state.sessionMessages.append(
-        AIMessage(content=assistant_answer.content)
-    )
+    # st.session_state.sessionMessages.append(
+    #     AIMessage(content=assistant_answer.content)
+    # )
+
+    st.session_state.sessionMessages.append(assistant_answer)
+    assistant_answer
 
     return assistant_answer
 
@@ -44,4 +47,3 @@ if submit:
     st.subheader("Answer:")
 
     st.write(response, key=1)
-    st.write(answer, key=2)