Fix language support

download.py

@@ -130,7 +130,8 @@ def caption_generator(tweet_url, language="Autodetect", model_size=model_size):
   try:
     print(f"Starting whisper transcribe with {meta['id']}.mp3")
     transcribe_whisper_result = transcribe(audio, translate_action=False, language=language, override_model_size=model_size)
-    translate_whisper_result = transcribe(audio, translate_action=True, language=language, override_model_size=model_size)
+    detected_language = LANGUAGES[transcribe_whisper_result["language"]]
+    translate_whisper_result = transcribe(audio, translate_action=True, language=detected_language, override_model_size=model_size)
     srt = get_srt(transcribe_whisper_result["segments"])
     en_srt = get_srt(translate_whisper_result["segments"])
 
@@ -233,7 +234,9 @@ def transcribe(audio, translate_action=True, language='Autodetect', override_mod
   output = model.transcribe(audio, verbose=True, **props)
 
   output['segments'] = output['segments']
-  print(f'Finished transcribe from {output["language"]}', output["text"])
+  output['requested_language'] = language.lower()
+
+  print(f'Finished transcribe from {LANGUAGES[output["language"]].capitalize()}', output["text"])
   return output