main.py

"""
Главный скрипт для запуска суммаризации
"""
from summarizer import AudioSummarizer


def main():
    # Параметры модели
    summarizer = AudioSummarizer(
        beta=[0.2, 0.25, 0.3, -0.15],  # pitch_std, volume_std, pause_duration, tempo_std
        lambd=0.5,  # вес аномалии
        delta=0.1,  # сила позиционного эффекта
        theta=0.6,  # коэффициент подавления избыточности
        alpha=0.7,  # баланс IDF/POS в сходстве
        L=2  # радиус локальной окрестности
    )

    # Запуск суммаризации
    result = summarizer.summarize(
        audio_path="segment_002.wav",  # замените на путь к вашему аудиофайлу
        k=6
    )

    print("\n" + "=" * 60)
    print("ИТОГОВЫЙ РЕЗУЛЬТАТ")
    print("=" * 60)
    print("Реферат:", " ".join(result["summary"]))


if __name__ == "__main__":
    main()