you-get 支持

app.py

@@ -3,9 +3,31 @@ import gradio as gr
 from pytube import YouTube
 import random
 import requests,json
+import subprocess
 
 
+def del_down_file():
+    command = f'rm -rf *.mp4'
+    subprocess.call(command, shell=True)
 
+def get_video(url):
+
+
+    # 下载视频
+    url = url
+    output_dir = '.'
+    command = f'you-get -o {output_dir} {url}'
+    subprocess.call(command, shell=True)
+    
+    mp4_files = []  # 用于存储所有找到的 mp4 文件名
+
+    # 遍历指定目录中的所有文件
+    for filename in os.listdir('.'):
+        # 检查文件是否以 '.mp4' 结尾
+        if filename.endswith('.mp4'):
+            # 如果是，将文件名添加到 mp4_files 列表中
+            mp4_files.append(filename)
+    return mp4_files[0]
 
 
 def create(prompt,openai_key):
@@ -66,11 +88,12 @@ def convert(res,openai_key):
 
 def get_audio(url):
     
-    yt = YouTube(url)
-    audio_file = f'{random.randint(10000,100000)}.mp4'
-    print(f'{url} {audio_file} start get audio ...')
-    yt.streams.filter(only_audio=True)[0].download(filename=audio_file)
-    print('aodio over ..')
+    # yt = YouTube(url)
+    # audio_file = f'{random.randint(10000,100000)}.mp4'
+    # print(f'{url} {audio_file} start get audio ...')
+    # yt.streams.filter(only_audio=True)[0].download(filename=audio_file)
+    # print('aodio over ..')
+    audio_file = get_video(url)
     return audio_file
 
 def get_transcript(url,openai_key):
@@ -89,6 +112,7 @@ def get_transcript(url,openai_key):
     }
     response = requests.post('https://api.gladia.io/audio/text/audio-transcription/', headers=headers, files=files)
     print(response.text)
+    del_down_file()
     return convert(response,openai_key)