Update app.py

app.py

@@ -3,6 +3,7 @@ import openai
 from kokoro import KPipeline
 import soundfile as sf
 import io
+import time  # To simulate some delay for spinner visibility
 
 # Streamlit App UI Setup
 st.title("Text-to-Speech Translator with Kokoro")
@@ -191,40 +192,46 @@ def generate_audio(text, lang_code, voice, speed):
 
 # Generate and display the audio file
 if st.button('Generate Audio'):
-
-    # Display spinner during audio generation
-    with st.spinner('Generating 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)
-        
-        # Display Audio player for the original language
-        st.audio(audio_buffer, format='audio/wav')
-
-        # Optional: Save the generated audio file for download (Original Text)
+    # Display custom spinner while generating audio
+    spinner_placeholder = st.empty()  # Create a placeholder for the spinner
+    spinner_placeholder.markdown('<div class="lds-ellipsis"><div></div><div></div><div></div><div></div></div>', unsafe_allow_html=True)
+
+    # Simulate a delay (can be removed once audio generation is complete)
+    time.sleep(1)  # Simulating audio generation time (Remove this line in production)
+
+    # Generate audio for the original text
+    audio_buffer = generate_audio(input_text, lang_code, voice, speed)
+    
+    # Remove the spinner once audio generation is complete
+    spinner_placeholder.empty()
+
+    # Display Audio player for the original language
+    st.audio(audio_buffer, format='audio/wav')
+
+    # Optional: Save the generated audio file for download (Original Text)
+    st.download_button(
+        label="Download Audio (Original Text)",
+        data=audio_buffer,
+        file_name="generated_speech_original.wav",
+        mime="audio/wav"
+    )
+
+    # Check if OpenAI API Key is provided for translation and English audio generation
+    if openai_api_key:
+        # Translate the input text to English using OpenAI
+        translated_text = translate_to_english(openai_api_key, input_text, lang_code)
+
+        # Generate audio for the translated English text
+        translated_audio_buffer = generate_audio(translated_text, 'a', voice, speed)
+
+        # Display Audio for the translated text
+        st.write(f"Translated Text: {translated_text}")
+        st.audio(translated_audio_buffer, format='audio/wav')
+
+        # Optional: Save the generated audio file for download (Translated Text)
         st.download_button(
-            label="Download Audio (Original Text)",
-            data=audio_buffer,
-            file_name="generated_speech_original.wav",
+            label="Download Audio (Translated to English)",
+            data=translated_audio_buffer,
+            file_name="generated_speech_translated.wav",
             mime="audio/wav"
         )
-
-        # Check if OpenAI API Key is provided for translation and English audio generation
-        if openai_api_key:
-            # Translate the input text to English using OpenAI
-            translated_text = translate_to_english(openai_api_key, input_text, lang_code)
-
-            # Generate audio for the translated English text
-            translated_audio_buffer = generate_audio(translated_text, 'a', voice, speed)
-
-            # Display Audio for the translated text
-            st.write(f"Translated Text: {translated_text}")
-            st.audio(translated_audio_buffer, format='audio/wav')
-
-            # Optional: Save the generated audio file for download (Translated Text)
-            st.download_button(
-                label="Download Audio (Translated to English)",
-                data=translated_audio_buffer,
-                file_name="generated_speech_translated.wav",
-                mime="audio/wav"
-            )