Update app.py

app.py

@@ -73,6 +73,12 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
+# Streamed response generator
+def response_generator(response):
+    for word in response.split():
+        yield word + " "
+        time.sleep(0.05)  # Adjust this delay to control typing speed
+
 # Accept user input
 if prompt := st.chat_input("What are you looking to learn?"):
     # Add user message to chat history
@@ -81,15 +87,14 @@ if prompt := st.chat_input("What are you looking to learn?"):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # Assistant response generation with a streaming effect
+    # Assistant response generation with streaming effect
     with st.chat_message("assistant"):
         response = qa_chain({"question": prompt})
         response_text = response["answer"]
 
-        # Stream the response character by character
-        for char in response_text:
-            st.markdown(char, unsafe_allow_html=True)
-            time.sleep(0.01)  # Adjust this delay to control typing speed
+        # Stream the response word by word
+        for word in response_generator(response_text):
+            st.markdown(word, unsafe_allow_html=True)
 
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response_text})
+    st.session_state.messages.append({"role": "assistant", "content": response_text})