import os
import json
import argparse

from src.transcript_generator import TranscriptGenerator


def main():
    # Create a parser
    parser = argparse.ArgumentParser(description='Transcript Generator for Videos')

    # Add the arguments
    parser.add_argument('--job', type=str, help='The name of the job', required=True)

    # Parse the arguments
    args = parser.parse_args()

    # Load the job config
    with open(f'jobs/{args.job}.json', 'r') as f:
        job_config = json.load(f)

    # Check the hf_token
    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(
        name=job_config['name'],
        language_code=job_config['language_code']
    )

    transcript_generator.generate(job_config['source'])

    if __name__ == "__main__":
        main()