Update app.py

app.py

@@ -147,6 +147,10 @@ if prompt := st.chat_input("How can I help you today?"):
         message_history = "\n".join(list(get_message_history())[-3:])
         result = qa_chain(prompt)
         output = [result['result']]
+        for item in output:
+            full_response += item
+            message_placeholder.markdown(full_response + "▌")
+            message_placeholder.markdown(full_response)
 
 def generate_pdf():
     page_number = int(result['source_documents'][0].metadata['page'])
@@ -184,10 +188,7 @@ def generate_audio():
 st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
 st.session_state['audio'] = ''
 
-for item in output:
-    full_response += item
-    message_placeholder.markdown(full_response + "▌")
-    message_placeholder.markdown(full_response)
+
 
 
 if "reference" not in st.session_state: