example_usage.py - master - dayekb/vibecoding_whisper

example_usage.py
78 строк3 KB
dayekb
upload files
09 окт 2025, 06:01
09 окт 2025, 06:01f90517a
#!/usr/bin/env python3
"""
Пример использования расшифровщика аудио
"""

import os
import sys
import whisper
from audio_transcriber import AudioTranscriber

def example_basic_transcription():
    """Пример базовой транскрипции"""
    print("=== Пример базовой транскрипции ===")

    # Создаем тестовый аудио файл (если он существует)
    test_audio = "audio_files/test_sample.wav"

    if not os.path.exists(test_audio):
        print(f"Тестовый файл {test_audio} не найден.")
        print("Поместите аудио файл в папку audio_files и запустите пример снова.")
        return

    # Создаем транскриптор
    transcriber = AudioTranscriber()

    # Загружаем модель
    if transcriber.load_model("base"):
        # Выполняем транскрипцию
        result = transcriber.transcribe_audio(test_audio, language="ru")
        if result:
            print(f"Результат: {result[:200]}...")  # Первые 200 символов

def example_different_models():
    """Пример сравнения разных моделей"""
    print("\n=== Пример сравнения моделей ===")

    test_audio = "audio_files/test_sample.wav"

    if not os.path.exists(test_audio):
        print(f"Тестовый файл {test_audio} не найден.")
        return

    models_to_test = ["tiny", "base", "small"]

    for model_name in models_to_test:
        print(f"\nТестируем модель: {model_name}")
        transcriber = AudioTranscriber()

        if transcriber.load_model(model_name):
            result = transcriber.transcribe_audio(test_audio, language="ru")
            if result:
                print(f"Результат ({model_name}): {result[:100]}...")

def show_available_models():
    """Показать доступные модели"""
    print("\n=== Доступные модели Whisper ===")
    models = whisper.available_models()
    for i, model in enumerate(models, 1):
        print(f"{i}. {model}")

def main():
    print("Примеры использования расшифровщика аудио")
    print("=" * 50)

    # Показываем доступные модели
    show_available_models()

    # Запускаем примеры
    example_basic_transcription()
    example_different_models()

    print("\n" + "=" * 50)
    print("Примеры завершены!")
    print("Для реального использования запустите:")
    print("python audio_transcriber.py путь_к_вашему_файлу.mp3")

if __name__ == "__main__":
    main()