transcriber.py

import click
from app import application
from app import __version__, __app_name__
from datetime import datetime


@click.group()
def cli():
    pass


def print_version(ctx, param, value):
    if not value or ctx.resilient_parsing:
        return
    click.echo(f"{__app_name__} v{__version__}")
    ctx.exit()


def print_help(ctx, param, value):
    if not value or ctx.resilient_parsing:
        return
    print(ctx.get_help())
    ctx.exit()


@click.command()
@click.argument('source', nargs=1)
@click.argument('loc', nargs=1)
@click.option('-m', '--model', type=click.Choice(['tiny', 'tiny.en', 'base', 'base.en', 'small', 'small.en', 'medium', 'medium.en', 'large-v2']), default='tiny.en',
              help='Options for transcription model'
              )
@click.option('-t', '--title', type=str,
              help="Supply transcribed file title in 'quotes', title is mandatory in case of audio files")
@click.option('-d', '--date', type=str, help="Supply the event date in format 'yyyy-mm-dd'")
@click.option('-T', '--tags', type=str, help="Supply the tags for the transcript in 'quotes' and separated by commas")
@click.option('-s', '--speakers', type=str, help="Supply the speakers for the transcript in 'quotes' and separated by "
                                                 "commas")
@click.option('-c', '--category', type=str, help="Supply the category for the transcript in 'quotes' and separated by "
                                                 "commas")
@click.option('-v', '--version', is_flag=True, callback=print_version,
              expose_value=False, is_eager=True, help="Show the application's version and exit.")
@click.option('-C', '--chapters', is_flag=True, default=False,
              help="Supply this flag if you want to generate chapters for the transcript")
@click.option('-h', '--help', is_flag=True, callback=print_help, expose_value=False, is_eager=True,
              help="Show the application's help and exit.")
@click.option('-p', '--PR', is_flag=True, default=False, help="Supply this flag if you want to generate a payload")
def add(
        source: str,
        loc: str,
        model: str,
        title: str,
        date: str,
        tags: str,
        speakers: str,
        category: str,
        chapters: bool,
        pr: bool
) -> None:
    """Supply a YouTube video id and directory for transcription. \n
       Note: The https links need to be wrapped in quotes when running the command on zsh
    """
    created_files = []
    try:
        username = application.get_username()
        loc = loc.strip("/")
        event_date = None
        if date:
            try:
                event_date = datetime.strptime(date, '%Y-%m-%d').date()
            except:
                print("Supplied date is invalid")
                return

        source_type = application.check_source_type(source=source)
        if source_type is None:
            print("Invalid source")
            return
        filename = application.process_source(source=source, title=title, event_date=event_date, tags=tags,
                                              category=category, speakers=speakers, loc=loc, model=model,
                                              username=username, chapters=chapters, pr=pr,
                                              source_type=source_type, created_files=created_files)
        if filename:
            """ INITIALIZE GIT AND OPEN A PR"""
            print("Transcription complete")
        print("Cleaning up...")
        application.clean_up(created_files)
    except Exception as e:
        print(e)
        print("Cleaning up...")
        application.clean_up(created_files)