Add application file

app.py

@@ -1,11 +1,24 @@
 import gradio as gr
+from pytube import YouTube
 
-def check_link(link):
+def get_youtube_details(link):
+    try:
+        yt = YouTube(link)
+        
+        # Get the thumbnail URL
+        thumbnail_url = yt.thumbnail_url
+        
+        # Get captions (if available)
+        caption_text = "No captions available."
+        if yt.captions:
+            # Choose the first caption track available
+            caption = yt.captions.get_by_language_code('en')
+            if caption:
+                caption_text = caption.generate_srt_captions()
+        
+        return caption_text, thumbnail_url
+    except Exception as e:
+        return str(e), None
 
-    if "clickbait" in link.lower():
-        return "Yes"
-    else:
-        return "No"
-
-demo = gr.Interface(fn=check_link, inputs="text", outputs="text")
+demo = gr.Interface(fn=get_youtube_details, inputs="text", outputs=["text", "image"])
 demo.launch()