Update app.py

app.py

@@ -70,6 +70,7 @@ if st.button("Submit"):
         inputs = tokenizer(text_input, return_tensors="pt")
         translated = model.generate(**inputs)
         st.session_state.translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
+        st.session_state.audio_path = ""  # Clear previous audio path
         st.write(f"Translated text: {st.session_state.translated_text}")
     else:
         st.error("Please enter text and select a target language.")
@@ -94,7 +95,7 @@ if st.button("Reset"):
     st.experimental_rerun()  # Reload the app to reset the inputs
 
 # Display current state of translated text and audio
-if st.session_state.translated_text:
+if st.session_state.translated_text and not st.session_state.audio_path:
     st.write(f"Translated text: {st.session_state.translated_text}")
-if st.session_state.audio_path:
-    st.audio(st.session_state.audio_path, format="audio/wav")
+elif st.session_state.audio_path:
+    st.audio(st.session_state.audio_path, format="audio/wav")