Update app_multi.py

app_multi.py

@@ -8,6 +8,9 @@ import os
 import time
 import random
 
+from search import get_youtube, download_random
+import soundfile
+
 import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image, ImageDraw, ImageFont
@@ -38,6 +41,38 @@ from infer_pack.models import (
 )
 from vc_infer_pipeline import VC
 
+# music search
+
+def auto_search(name):
+    save_music_path = '/tmp/downloaded'
+    if not os.path.exists(save_music_path):
+        os.makedirs(save_music_path)
+
+    config = {'logfilepath': 'musicdl.log', save_music_path: save_music_path, 'search_size_per_source': 5,
+              'proxies': {}}
+    save_path = os.path.join(save_music_path, name + '.mp3')
+    # youtube
+    get_youtube(name, os.path.join(save_music_path, name))
+    # task1 = threading.Thread(
+    #     target=get_youtube,
+    #     args=(name, os.path.join(save_music_path, name))
+    # )
+    # task1.start()
+    # task2 = threading.Thread(
+    #     target=download_random,
+    #     args=(name, config, save_path)
+    # )
+    # task2.start()
+    # task1.join(timeout=20)
+    # task2.join(timeout=10)
+
+    if not os.path.exists(save_path):
+        return "Not Found", None
+    signal, sampling_rate = soundfile.read(save_path, dtype=np.int16)
+    # signal, sampling_rate = open_audio(save_path)
+
+    return (sampling_rate, signal)
+
 # SadTalker
 
 import os, sys