Sasha commited on
Commit
9fb93a9
·
1 Parent(s): e98099c

style: optimize transcribe_confirm prompt and skip fully transcribed audio

Browse files
Files changed (1) hide show
  1. local_worker/vod_backfiller.py +18 -14
local_worker/vod_backfiller.py CHANGED
@@ -460,20 +460,24 @@ if __name__ == "__main__":
460
  print("\n-> Синхронизация чата успешно завершена!")
461
 
462
  # 5. Transcribe Audio
463
- transcribe_confirm = input("\nХотите запустить скачивание и транскрибацию аудиодорожки (Whisper)? (y/n): ").strip().lower()
464
- if transcribe_confirm == 'y':
465
- # Init Whisper
466
- print(f"\n[Whisper] Загрузка модели {WHISPER_MODEL_SIZE}...")
467
- try:
468
- model = WhisperModel(
469
- WHISPER_MODEL_SIZE,
470
- device=WHISPER_DEVICE,
471
- compute_type=WHISPER_COMPUTE_TYPE,
472
- cpu_threads=WHISPER_CPU_THREADS
473
- )
474
- transcribe_vod_audio(vod_id, created_at, model, start_offset=voice_start_offset)
475
- except Exception as e:
476
- print(f"[Whisper Error] Не удалось инициализировать Whisper: {e}")
 
 
 
 
477
 
478
  # 5. Mark Completed on Server
479
  try:
 
460
  print("\n-> Синхронизация чата успешно завершена!")
461
 
462
  # 5. Transcribe Audio
463
+ if voice_start_offset >= length_seconds:
464
+ print(f"\n[Sync] Голосовая дорожка уже полностью распознана (секунда {voice_start_offset} из {length_seconds}). Транскрибация не требуется.")
465
+ else:
466
+ prompt_str = f"\nХотите запустить скачивание и транскрибацию аудиодорожки (Whisper)? (y/n) [Продолжить с {voice_start_offset}s]: " if voice_start_offset > 0 else "\nХотите запустить скачивание и транскрибацию аудиодорожки (Whisper)? (y/n): "
467
+ transcribe_confirm = input(prompt_str).strip().lower()
468
+ if transcribe_confirm == 'y':
469
+ # Init Whisper
470
+ print(f"\n[Whisper] Загрузка модели {WHISPER_MODEL_SIZE}...")
471
+ try:
472
+ model = WhisperModel(
473
+ WHISPER_MODEL_SIZE,
474
+ device=WHISPER_DEVICE,
475
+ compute_type=WHISPER_COMPUTE_TYPE,
476
+ cpu_threads=WHISPER_CPU_THREADS
477
+ )
478
+ transcribe_vod_audio(vod_id, created_at, model, start_offset=voice_start_offset)
479
+ except Exception as e:
480
+ print(f"[Whisper Error] Не удалось инициализировать Whisper: {e}")
481
 
482
  # 5. Mark Completed on Server
483
  try: