Update app.py

app.py

@@ -1,11 +1,20 @@
 import gradio as gr
 import whisper
 #from pytube import YouTube
-import OS
+import os
 from googleapiclient.discovery import build
+
 api_key = os.environ['api_key']
 youtube_api = build('youtube', 'v3', developerKey=api_key)
 
+#Get title and thumbnail
+def get_vid_details(youtube_link):
+    youtube_id = get_video_id(youtube_link)
+    return response['items'][0]['snippet']['thumbnails']['high']['url'], youtube_id['items'][0]['snippet']['title']
+
+def get_video_id(url):
+    parsed_url = urlparse(url)
+    return parse_qs(parsed_url.query)['v'][0]
 
 loaded_model = whisper.load_model("medium")
 current_size = 'medium'
@@ -61,7 +70,8 @@ with block:
           
           # Events
           btn.click(inference, inputs=[link], outputs=[text])
-          link.change(populate_metadata, inputs=[link], outputs=[img, title])
+          #link.change(populate_metadata, inputs=[link], outputs=[img, title])
+          link.change(get_vid_details, inputs=[link], outputs=[img, title])
           sz.change(change_model, inputs=[sz], outputs=[])
 
 block.launch(debug=True)