add option to translate

app.py

@@ -22,6 +22,9 @@ uploaded_file = st.file_uploader("Upload Audio File", type=["mp3", "wav", "mov"]
 use_free_tier = st.checkbox("Free Tier (Max 2 minutes)")
 api_token = st.text_input("API Token (Unlimited)")
 
+# Should we translate to english? 
+translate = st.checkbox("Would you like a translation to english?")
+
 # Model selection
 model_size = st.selectbox("Model Size", ("tiny", "base", "small", "medium"))
 
@@ -45,6 +48,12 @@ def transcribe_to_subtitle(audio_bytes, model_name):
   except Exception as e:
      return {"error": f"Error during transcription: {str(e)}"}
   
+  if translate:
+    # Get text from speech for subtitles from audio file
+    translation = model.transcribe(audio_bytes, verbose=True, task = 'translate')
+    english_output = translation.text
+    st.markdown(english_output, unsafe_allow_html=True)
+    
   captions = pysrt.open("audio.srt")
   for caption in captions:
      print(caption.start)