Update app.py

app.py

@@ -6,6 +6,7 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from llama_index.core import Settings
 import os
 import base64
+import time
 
 # Load environment variables
 load_dotenv()
@@ -71,6 +72,11 @@ def handle_query(query):
     else:
         return "Sorry, I couldn't find an answer."
 
+def streamer(text):
+    for i in text:
+        yield i
+        time.sleep(0.01)
+
 
 # Streamlit app initialization
 st.title("Chat with your PDF 🦜📄")
@@ -106,7 +112,7 @@ if user_prompt and uploaded_file:
     # Trigger assistant's response retrieval and update UI
     with st.spinner("Thinking..."):
         response = handle_query(user_prompt)
-        with st.chat_message("user", avatar="robot.png"):
-            st.write(response)
-        st.session_state.messages.append({'role': 'assistant', "content": response})
+    with st.chat_message("user", avatar="robot.png"):
+        st.write_stream(streamer(response))
+    st.session_state.messages.append({'role': 'assistant', "content": response})