Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import logging
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+import time 
 
 # Set the logger to display only CRITICAL messages
 logging.basicConfig(level=logging.CRITICAL)
@@ -32,6 +33,18 @@ if st.button("Submit"):
     if user_input:
         # Generate text based on the input
         generated_text = generate_text(user_input)
-        st.write(generated_text)
+        
+        # Initialize or reset the progress state
+        if 'text_progress' not in st.session_state:
+            st.session_state['text_progress'] = ""
+        
+        # Simulate typing effect
+        for i in range(len(generated_text)):
+            st.session_state['text_progress'] = generated_text[:i+1]
+            st.write(st.session_state['text_progress'])
+            time.sleep(0.05)  # Adjust as needed for the desired speed of the "typing" effect
+            
+        # Reset progress for next input
+        del st.session_state['text_progress']
     else:
         st.write("Please enter a prompt.")