Spaces:
Sleeping
Sleeping
Sasha commited on
Commit ·
9fb93a9
1
Parent(s): e98099c
style: optimize transcribe_confirm prompt and skip fully transcribed audio
Browse files- 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 |
-
|
| 464 |
-
|
| 465 |
-
|
| 466 |
-
|
| 467 |
-
|
| 468 |
-
|
| 469 |
-
|
| 470 |
-
|
| 471 |
-
|
| 472 |
-
|
| 473 |
-
|
| 474 |
-
|
| 475 |
-
|
| 476 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|