Update app.py

app.py

@@ -3,6 +3,7 @@ import openai
 from kokoro import KPipeline
 import soundfile as sf
 import io
+import time
 
 # Streamlit App UI Setup
 st.title("Text-to-Speech Translator with Kokoro")
@@ -115,21 +116,29 @@ def translate_to_english(api_key, text, lang_code):
 # Generate Audio function
 def generate_audio(text, lang_code, voice, speed):
     generator = pipeline(text, voice=voice, speed=speed, split_pattern=r'\n+')
+    audio_data = None
     for i, (gs, ps, audio) in enumerate(generator):
         audio_data = audio
-        # Save audio to in-memory buffer
-        buffer = io.BytesIO()
-        # Explicitly specify format as WAV
-        sf.write(buffer, audio_data, 24000, format='WAV')  # Add 'format="WAV"'
-        buffer.seek(0)
-        return buffer
+        time.sleep(1)  # Simulate processing delay for the spinner (this can be removed or adjusted)
+    # Save audio to in-memory buffer
+    buffer = io.BytesIO()
+    # Explicitly specify format as WAV
+    sf.write(buffer, audio_data, 24000, format='WAV')  # Add 'format="WAV"'
+    buffer.seek(0)
+    return buffer
 
 # Generate and display the audio file
 if st.button('Generate Audio'):
 
-    # Generate audio for the original text
-    st.write("Generating speech for the original text...")
-    audio_buffer = generate_audio(input_text, lang_code, voice, speed)
+    # Show the spinner with percentage count
+    with st.spinner("Generating audio... 0%"):
+        for i in range(1, 101, 10):
+            time.sleep(0.5)  # Simulate work being done, you can adjust timing here
+            st.spinner(f"Generating audio... {i}%")
+        
+        # Generate audio for the original text after the spinner
+        st.write("Generating speech for the original text...")
+        audio_buffer = generate_audio(input_text, lang_code, voice, speed)
     
     # Display Audio player for the original language
     st.audio(audio_buffer, format='audio/wav')
@@ -160,4 +169,4 @@ if st.button('Generate Audio'):
             data=translated_audio_buffer,
             file_name="generated_speech_translated.wav",
             mime="audio/wav"
-        )
+        )