import gradio as gr
from src.transcriber import transcriber

def main():
    with gr.Blocks(analytics_enabled=False, title='multilang-asr-transcriber') as demo:
        gr.Markdown('# multilang-asr-transcriber')
        gr.Markdown('### A multilingual automatic speech transcription tool using [faster-whisper](https://github.com/SYSTRAN/faster-whisper). Supports translation to english and user setting of max words per line.',)
        video_file = gr.File(file_types=["video"],type="filepath")
        max_words_per_line = gr.Number(value=6, label="Max words per line")
        task = gr.Dropdown(choices=["transcribe", "translate"], value="transcribe", label="Select Task")
        text_output = gr.Textbox(label="Text transcription")
        srt_file = gr.File(file_count="single", file_types=[".srt"], label="SRT file")
        gr.Interface(transcriber,
                    inputs=[video_file, max_words_per_line, task],
                    outputs=[text_output,srt_file],
                    allow_flagging="never",
                    analytics_enabled=False)
    demo.launch()

if __name__ == '__main__':
    main()