Update app.py

app.py

@@ -71,6 +71,8 @@ if st.button("Submit"):
         translated = model.generate(**inputs)
         st.session_state.translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
         st.write(f"Translated text: {st.session_state.translated_text}")
+    else:
+        st.error("Please enter text and select a target language.")
 
 # Listen to Translated Audio button
 if st.button("Listen to Translated Audio"):
@@ -87,6 +89,8 @@ if st.button("Listen to Translated Audio"):
 if st.button("Reset"):
     st.session_state.translated_text = ""
     st.session_state.audio_path = ""
+    st.text_area("Enter text in English", value="", key="text_input")
+    st.selectbox("Select target language", ["fr", "zh", "it", "ur", "hi"], key="target_lang")
     st.experimental_rerun()  # Reload the app to reset the inputs
 
 # Display current state of translated text and audio
@@ -94,5 +98,3 @@ if st.session_state.translated_text:
     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")
-
-