import gradio as gr
from summarize import Summarizer

interface = gr.Interface(fn = Summarizer,
                        inputs = [gr.inputs.Textbox(lines=2, 
                                                    placeholder="Enter your link...",
                                                    label='YouTube Video Link'),
                                  gr.inputs.Radio(["mT5", "BART"], type="value", label='Model')],
                        outputs = [gr.outputs.Textbox(
                                                      label="Summary")],
                         
                        title = "Youtube Summarizer",
                        examples = [
                                   ['https://www.youtube.com/watch?v=OaeYUm06in0&list=PLHgX2IExbFouJoqEr8JMF5MbZSbyC91-L&t=5761s', 'BART'],
                                   ['https://youtu.be/8hOzsFETm4I?list=PLHgX2IExbFosAKDJ6yuVjnQLRhoj_yc3g', 'BART'],
                                   ['https://youtu.be/DpVt1WDTD9k?list=PLHgX2IExbFosAKDJ6yuVjnQLRhoj_yc3g', 'BART'],
                                   ['https://youtu.be/3bSVKNKb_PY?list=PLHgX2IExbFosAKDJ6yuVjnQLRhoj_yc3g', 'BART'],
                                   ['https://youtu.be/SGazDb8o-to?list=PLHgX2IExbFosAKDJ6yuVjnQLRhoj_yc3g', 'mT5']
                                   ],
                        enable_queue=True)

interface.launch(debug=True)