Update app.py

app.py

@@ -1,12 +1,47 @@
+from util import load_model
+from util import pipeline
+
 import gradio as gr
-from util import update
+cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented2'
+
+
+def get_model(cp):
+    checkpoint = cp
+    tokenizer, model = load_model(checkpoint)
+    return tokenizer, model
+
+tokenizer, model = get_model(cp_aug)
+
+
+def generate_summary(url):
+    results = pipeline(url, model, tokenizer)
+    summary = "\n".join(results)
+    return summary
+
+def generate_summary_and_video(url):
+    summary = generate_summary(url)
+    summary_html = summary.replace("\n", "<br>")
+    try:
+        video_id = url.split("v=")[1].split("&")[0]
+        iframe = f'<iframe width="300" height="200" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="accelerometer; autoplay; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
+        return f"{iframe}<br><br>Những ý chính trong video:<br><br>{summary_html}"
+    except IndexError:
+        return f"**Summary:**\n{summary}\n\nInvalid YouTube URL for video display."
+
+css = """
+.output-html {
+    font-size: 40px;
+}
+"""
+
+demo = gr.Interface(
+    fn=generate_summary_and_video,
+    inputs=gr.Textbox(lines=2, placeholder="Enter URL..."),
+    outputs=gr.HTML(label="Results"),
+    title="Summarizer",
+    description="Enter the URL to display the YouTube video and summarize the content.",
+    css=css
+)
 
-with gr.Blocks() as demo:
-    gr.Markdown("Start typing below and then click **Run** to see the output.")
-    with gr.Row():
-        inp = gr.Textbox(placeholder="What is your name?")
-        out = gr.Textbox()
-    btn = gr.Button("Run")
-    btn.click(fn=update, inputs=inp, outputs=out)
 
-demo.launch()
+demo.launch(share=True)