Update core.py

core.py

@@ -5,6 +5,10 @@ import whisper
 
 import datetime
 
+import os
+import gradio as gr
+from pytube import YouTube
+
 transformers.utils.move_cache()
 
 # ====================================
@@ -75,4 +79,22 @@ def text_summarize(text_input, lang):
     elif lang == 'Vietnamese':
         return text_summarize_vi(text_input)
     else:
-        return ""
+        return ""
+
+def load_video_url(url):
+    current_dir = os.getcwd()
+    
+    try: 
+        yt = YouTube(url) 
+    except: 
+        print("Connection Error")
+        raise gr.Error("Connection Error")
+    try:
+        highest_audio = yt.streams.filter(progressive=False).get_highest_resolution().itag
+        file_url = os.path.join(current_dir, "audio", "temp.mp4")
+        yt.streams.get_by_itag(highest_audio).download(output_path=os.path.join(current_dir, "audio"), filename = "temp.mp4") 
+    except : 
+        print("Download video error") 
+        raise gr.Error("Download video error")
+
+    return file_url