Update app.py

app.py

@@ -19,7 +19,24 @@ def generate_summary(url):
     summary = "\n".join(results)
     return summary
 
-demo = gr.Interface(
+
+def get_youtube_video(url):
+    try:
+        video_id = url.split("v=")[1].split("&")[0]
+        iframe = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="accelerometer; autoplay; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
+        return iframe
+    except IndexError:
+        return "Invalid YouTube URL"
+
+video_interface = gr.Interface(
+    fn=get_youtube_video,
+    inputs=gr.Textbox(lines=2, placeholder="Enter YouTube URL..."),
+    outputs=gr.HTML(label="YouTube Video"),
+    title="YouTube Video Display",
+    description="Enter the YouTube URL to display the video."
+)
+
+summary_interface = gr.Interface(
     fn=generate_summary,
     inputs=gr.Textbox(lines=2, placeholder="Enter your URL..."),
     outputs=gr.Textbox(label="Generated Text"),
@@ -27,6 +44,9 @@ demo = gr.Interface(
     description="Enter the URL to summarize and click 'Submit' to generate the summary."
 )
 
+demo = gr.TabbedInterface([summary_interface, video_interface], ["Summary Generator", "YouTube Video Info"])
+
+
 demo.launch(share=True)