Update app.py

app.py

@@ -1,6 +1,60 @@
 import gradio as gr
-import os
+import os,random
 from basic_pitch.inference import predict_and_save
+from urllib.parse import quote
+os.system('chmod +x ./pget')
+
+
+
+
+
+def downloadaud(query):
+    filename=str(random.randint(0,9999999))
+    # Step 1: Search for videos with the given query
+    search_url = f"https://draw-8fj.begin.app/api/search/{quote(query)}"
+    search_response = requests.get(search_url).json()
+    print('1=>', search_response)
+    # Step 2: Find the first video with duration less than 10 minutes and extract its ID
+    video_id = None
+    #search_response = sorted(search_response, key=lambda x: x["views"], reverse=True)
+    for item in search_response:
+        duration = item.get("duration_raw")
+        if duration and len(duration.split(':'))< 3 and int(duration.split(':')[0])<10:
+            video_id = item.get("id", {}).get("videoId")
+            break
+    print('1-r',video_id)
+    # If no video with duration less than 10 minutes was found, return None
+    if not video_id:
+        return None
+
+    # Step 3: Get the formats for the video and find the URL for the best audio-only format
+    formats_url = f"https://draw-8fj.begin.app/api/info/{video_id}"
+    formats = requests.get(formats_url)
+    if formats.ok!=True:
+        formats_url = f"https://draw-8fj-staging.begin.app/api/info/{video_id}"
+        formats = requests.get(formats_url) 
+    formats_response=formats.json()
+    print(formats_response["formats"])
+    best_audio_format = None
+    for fmt in formats_response.get("formats", []):
+        if fmt.get("hasVideo") is False and fmt.get("hasAudio") is True and fmt.get("container") == "mp4":
+            if not best_audio_format or fmt.get("audioBitrate") > best_audio_format.get("audioBitrate"):
+                best_audio_format = fmt
+    print(best_audio_format)
+    # If no suitable audio format was found, return None
+    if not best_audio_format:
+        return None
+
+    aurl=best_audio_format["url"]
+    print(aurl)
+    command = f'./pget -o {filename}.mp4 -p 4 "{aurl}" '
+    os.system(command)
+    os.system(f'ffmpeg -i {filename}.mp4 -q:a 0 -map a {filename}.mp3')
+    os.system(f'mv {filename}.mp3 /tmp/{filename}.mp3')
+    res= predict(f'/tmp/{filename}.mp3')
+    return res
+
+
 def predict(ina):
     predict_and_save(
     [ina],
@@ -15,8 +69,8 @@ def predict(ina):
     os.system(f'ffmpeg -i "{ina}.wav" -f mp3 -acodec libmp3lame -y "{ina}.mp3"')
     
     return f'{ina}.mp3',f'{ina}.mid'
-aud=gr.Audio(label="mp3/ogg/wav/flac/m4a 尽量不要叫容易重名的文件名",type='filepath')
+aud=gr.Textbox(label="yt query")
 oa=gr.Audio(label="预览midi")
 pf=gr.File(label="下载midi")
-demo = gr.Interface(fn=predict, inputs=[aud], outputs=[oa,pf])
+demo = gr.Interface(fn=downloadaud, inputs=[aud], outputs=[oa,pf])
 demo.launch()