from os import pipe
import gradio as gr
from functions.punctuation import punctuate

title = "sponsoredBye - never listen to sponsors again"
description = "Sponsored sections in videos are annoying and take up a lot of time. Improve your YouTube watching experience, by filling in the youtube url and figure out what segments to skip."


def pipeline(video_url):
    video_id = video_url.split("?v=")[-1]
    punctuated_text = punctuate(video_id)
    return punctuated_text


# print(pipeline("VL5M5ZihJK4"))
demo = gr.Interface(
    fn=pipeline,
    title=title,
    description=description,
    inputs="text",
    outputs="text",
    examples=[
        "https://www.youtube.com/watch?v=VL5M5ZihJK4",
        "https://www.youtube.com/watch?v=VL5M5ZihJK4",
    ],
)
demo.launch(share=True)