app.py

from huggingface_hub import login
from diarization import start_diarization
from transcribe import start_transcribe
import ffmpeg
import gradio as gr


def prepare_input(input_file, progress=gr.Progress()):
    output_file = "input.wav"
    progress(0.2, desc="Preparing video")
    ffmpeg.input(input_file).audio.output(
        output_file, format="wav").run()
    progress(0.4, desc="Acquiring diarization")
    start_diarization(output_file)
    progress(0.6, desc="Transcribing audio")
    return start_transcribe(progress)


output_files = gr.outputs.File(
    label="Transcribe"), gr.outputs.File(label="Subtitle")

video_interface = gr.Interface(
    fn=prepare_input,
    inputs=gr.Video(type="file"),
    outputs=output_files,
    title="Test 2"
)

if __name__ == "__main__":
    video_interface.launch()