Translate done

download.py

@@ -191,10 +191,9 @@ def caption_generator(social_media_url,uid, language="Autodetect", model_size=mo
       if detected_language != "en":
         print(f"Transcribe successful! Starting translation to English")
         translate_whisper_result = transcribe(audio, translate_action=True, language=detected_language, override_model_size=model_size)
-        print(f"Trfan!, writing files")
         en_vtt_path = tempdir / f"en.vtt"
         with open(en_vtt_path.resolve(), "w", encoding="utf-8") as en_vtt:
-          write_vtt(transcribe_whisper_result["segments"], file=en_vtt)
+          write_vtt(translate_whisper_result["segments"], file=en_vtt)
         print(f"Finished translation to English, preparing subtitle files")
         whisper_result_captions.append(
           {
@@ -206,7 +205,8 @@ def caption_generator(social_media_url,uid, language="Autodetect", model_size=mo
     except Exception as e:
       print(f"Could not transcribe file: {e}")
       raise
-
+    print(f"Finished processing {uid} file, returning results")
+    print(whisper_result_captions)
     return 'success', whisper_result_captions
 
   # Run whisper with translation task enabled (and save to different srt file)