import gradio as gr
from run import get_model, detect_video

model = get_model()

def greet(video):
    print(video, type(video))
    pred = detect_video(video_path=video, model=model)
    if pred > 0.5:
        string = f"Fake: {pred*100:.2f}%"
    else:
        string = f"Real: {(1-pred)*100:.2f}%"
    print(string)
    return string

with gr.Blocks() as demo:
    gr.Markdown("# Fake Video Detector")
    with gr.Tabs():
        with gr.TabItem("Video Detect"):
            with gr.Row():
                video_input = gr.Video()
                detect_output = gr.Textbox("Output")
            video_button = gr.Button("detect")

    video_button.click(greet, inputs=video_input, outputs=detect_output)

demo.launch()