import os

from src.transcript_generator import TranscriptGenerator


def main():
    # input could be a YouTube link or a local file
    input_sources = ["https://www.youtube.com/watch?v=BVVo1BaL6JA"]

    hf_token = os.environ.get("HF_TOKEN")
    if not hf_token:
        hf_token = input("Please enter your HF_TOKEN: ")
    os.environ["HF_TOKEN"] = hf_token

    transcript_generator = TranscriptGenerator()
    transcript_generator.generate(input_sources)


if __name__ == "__main__":
    main()