Update app.py

app.py

@@ -1,18 +1,13 @@
 import os
 import streamlit as st
-import random
 import time
+import re
 from langchain.chat_models import ChatOpenAI
 from langchain.chains import ConversationalRetrievalChain
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationSummaryBufferMemory
 from langchain.vectorstores import FAISS
 from langchain.embeddings import OpenAIEmbeddings
-import re
-
-# Function to split text into sentences
-def split_into_sentences(text):
-    return re.split(r'(?<=[.!?]) +', text)
 
 # Set up the OpenAI API key
 os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY")
@@ -115,21 +110,15 @@ if prompt := st.chat_input("What are you looking to learn?"):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # Assistant response generator with sentence streaming
+    # Assistant response generation with a streaming effect
     with st.chat_message("assistant"):
         response = qa_chain({"question": prompt})
         response_text = response["answer"]
 
-        # Split response into sentences and display one by one
-        sentences = split_into_sentences(response_text)
-        for sentence in sentences:
-            st.markdown(sentence)
-            time.sleep(1)  # Delay between sentences (adjust as needed)
+        # 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
 
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response_text})
-
-# Optional: Add a button to clear the chat history
-if st.button("Clear Chat History"):
-    st.session_state.messages.clear()
-    st.experimental_rerun()
+    st.session_state.messages.append({"role": "assistant", "content": response_text})