Update app.py

app.py

@@ -44,22 +44,24 @@ def main():
 
         if user_text and (st.session_state.get("last_text", "") != user_text or st.button("Translate", key="translate_btn")):
             st.session_state.last_text = user_text
-            with st.spinner("Translating...") as spinner:  # Hidden spinner logic
-                start_time = time.time()
-                try:
-                    # Use English as buffer for translation
-                    if source_lang != "English" and target_lang != "English":
-                        temp_translation = translation.translate(user_text, source_lang, "English")
-                        translated_text = translation.translate(temp_translation, "English", target_lang)
-                    else:
-                        translated_text = translation.translate(user_text, source_lang, target_lang)
-                    st.session_state.translated_text = translated_text
-                    st.session_state.translation_time = time.time() - start_time
-                except Exception as e:
-                    st.session_state.translated_text = user_text
-                    st.session_state.translation_time = time.time() - start_time
-                    st.warning(f"Translation error: {str(e)}. Using input as fallback.")
-                spinner.empty()  # Hide spinner after processing
+            spinner = st.empty()  # Create spinner placeholder
+            spinner.info("Translating...")  # Show spinner
+            start_time = time.time()
+            try:
+                # Use English as buffer for translation
+                if source_lang != "English" and target_lang != "English":
+                    temp_translation = translation.translate(user_text, source_lang, "English")
+                    translated_text = translation.translate(temp_translation, "English", target_lang)
+                else:
+                    translated_text = translation.translate(user_text, source_lang, target_lang)
+                st.session_state.translated_text = translated_text
+                st.session_state.translation_time = time.time() - start_time
+            except Exception as e:
+                st.session_state.translated_text = user_text
+                st.session_state.translation_time = time.time() - start_time
+                st.warning(f"Translation error: {str(e)}. Using input as fallback.")
+            finally:
+                spinner.empty() 
 
     # Output Section
     with col_output:
@@ -81,7 +83,7 @@ def main():
                     audio = audio_processor.text_to_speech(st.session_state.translated_text, target_lang)
                     if audio and audio.getbuffer().nbytes > 0:
                         st.audio(audio, format="audio/mp3", start_time=0)
-                        st.success("Audio playing.")  # Only success message after playback starts
+                        st.success("Audio playing.")
                     else:
                         audio_fallback = audio_processor.text_to_speech(st.session_state.translated_text, "English")
                         if audio_fallback and audio_fallback.getbuffer().nbytes > 0: