Upload app.py

app.py

@@ -23,7 +23,7 @@ def load_answer(question):
         AIMessage(content=assistant_answer.content)
     )
 
-    return assistant_answer.content
+    return assistant_answer
 
 
 def get_text():
@@ -39,7 +39,9 @@ submit = st.button("Generate")
 
 if submit:
 
-    response = load_answer(user_input)
+    answer = load_answer(user_input)
+    response = answer.content
     st.subheader("Answer:")
 
     st.write(response, key=1)
+    st.write(response, key=2)