from huggingface_hub import login
from diarization import startDiarization
import ffmpeg
import gradio as gr
import os


def prepareInput(input_file):
    output_file = "input.wav"
    ffmpeg.input(input_file).audio.output(
        output_file, format="wav").run()
    return startDiarization(output_file)


video_interface = gr.Interface(
    fn=prepareInput,
    inputs=gr.Video(type="file"),
    outputs="text",
    title="Get Diarization"
)

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