Update app.py

app.py

@@ -23,10 +23,13 @@ def ask_question(question):
         else:
             return "Error: Failed to retrieve an answer."
 
-def animate_typing(text):
-    for char in text:
-        st.write_stream("Answer: " + text[:text.index(char)+1])
-        time.sleep(0.05)
+def typewriter(text: str, speed: int):
+    tokens = text.split()
+    container = st.empty()
+    for index in range(len(tokens) + 1):
+        curr_full_text = " ".join(tokens[:index])
+        container.markdown(curr_full_text)
+        time.sleep(1 / speed)
 
 def main():
     st.title("Question Answering System")
@@ -39,7 +42,7 @@ def main():
             answer_placeholder.write("Thinking...")
             answer = ask_question(question)
             answer_placeholder.empty()
-            animate_typing(answer)
+            typewriter("Answer: " + answer, speed=10)
         else:
             st.write("Please enter a question.")