import argparse

from subtitles.parser import parse_srt_to_df
from scoring.bert_ranker import BERTImportanceRanker
import config


def main():
    parser = argparse.ArgumentParser(
        description="Оценка важности фраз в субтитрах с помощью BERT.")
    parser.add_argument("srt_file", help="Путь к .srt файлу субтитров")
    parser.add_argument(
        "--output", help="Путь к выходному CSV", default=config.CSV_OUTPUT)
    args = parser.parse_args()

    df = parse_srt_to_df(args.srt_file)
    ranker = BERTImportanceRanker(config.MODEL_NAME)
    df_scored = ranker.apply_to_dataframe(df)
    df_scored.to_csv(args.output, index=False, encoding='utf-8')
    print(f"✅ Сохранено в {args.output}")


if __name__ == "__main__":
    main()