Update app.py

app.py

@@ -10,6 +10,9 @@ import random
 from datetime import datetime
 from scipy.io.wavfile import write
 from pydub import AudioSegment
+import requests
+import re
+
 high_quality = True
 
 '''
@@ -36,8 +39,30 @@ def svc(audio_src, audio_ref, singing_check):
 
   return "output.wav"
 
+headers = {
+    "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4302.0 Safari/537.36"
+}
+
+pattern = r'//www\.bilibili\.com/video[^"]*'
+
+def find_first_appearance_with_neighborhood(text, pattern):
+    match = re.search(pattern, text)
+    
+    if match:
+        return match.group()
+    else:
+        return None
+
+def search_bilibili(keyword):
+    req = requests.get("https://search.bilibili.com/all?keyword={}&duration=1&tids=3&page=1".format(keyword), headers=headers).text
+
+    video_link = "https:" + find_first_appearance_with_neighborhood(req, pattern)
+
+    return video_link
+
+
 def youtube_downloader(
-    video_identifier,
+    song_name,
     start_time,
     end_time,
     is_full_song,
@@ -47,6 +72,7 @@ def youtube_downloader(
     quiet=False,
     force=True,
 ):
+    video_identifier = search_bilibili(song_name)
     if is_full_song:
         ydl_opts = {
             'noplaylist': True,
@@ -175,7 +201,7 @@ with app:
   gr.Markdown("### <center>🌊 更多精彩应用，敬请关注[滔滔AI](http://www.talktalkai.com)；滔滔AI，为爱滔滔！💕</center>")
   with gr.Row():
     with gr.Column():
-      ydl_url_input  = gr.Textbox(label="音乐视频网址(可直接填写相应的BV号)", value = "https://www.bilibili.com/video/BV...")
+      ydl_url_input  = gr.Textbox(label="通过歌曲名搜索")
       with gr.Row():
         start = gr.Number(value=0, label="起始时间 (秒)")
         end = gr.Number(value=15, label="结束时间 (秒)")