Update app.py

app.py

@@ -27,8 +27,62 @@ import mutagen
 from mutagen.mp3 import MP3
 from mutagen.wave import WAVE
 
+
+import time
+import base64
+import gradio as gr
+from sentence_transformers import SentenceTransformer
+
+import httpx
+import json
+
+from utils import get_tags_for_prompts, get_mubert_tags_embeddings, get_pat
+
+minilm = SentenceTransformer('all-MiniLM-L6-v2')
+mubert_tags_embeddings = get_mubert_tags_embeddings(minilm)
+
+
+def get_track_by_tags(tags, pat, duration, maxit=20, loop=False):
+    if loop:
+        mode = "loop"
+    else:
+        mode = "track"
+    r = httpx.post('https://api-b2b.mubert.com/v2/RecordTrackTTM',
+                   json={
+                       "method": "RecordTrackTTM",
+                       "params": {
+                           "pat": pat,
+                           "duration": duration,
+                           "tags": tags,
+                           "mode": mode
+                       }
+                   })
+
+    rdata = json.loads(r.text)
+    assert rdata['status'] == 1, rdata['error']['text']
+    trackurl = rdata['data']['tasks'][0]['download_link']
+
+    print('Generating track ', end='')
+    for i in range(maxit):
+        r = httpx.get(trackurl)
+        if r.status_code == 200:
+            return trackurl
+        time.sleep(1)
+
+
+def generate_track_by_prompt(prompt):
+    try:
+        pat = get_pat("[email protected]")
+        _, tags = get_tags_for_prompts(minilm, mubert_tags_embeddings, [prompt, ])[0]
+        result = get_track_by_tags(tags, pat, int(30), loop=False)
+        print(result)
+        return result 
+    except Exception as e:
+        return str(e)
+
+
 img_to_text = gr.Blocks.load(name="spaces/fffiloni/CLIP-Interrogator-2")
-text_to_music = gr.Interface.load("spaces/fffiloni/text-2-music") 
+#text_to_music = gr.Interface.load("spaces/fffiloni/text-2-music") 
 
 language_translation_model = hub.Module(name='baidu_translate')
 language_recognition_model = hub.Module(name='baidu_language_recognition')
@@ -123,7 +177,8 @@ def get_music(prompt, musicAI_indx, duration):
         wavfile.export(mp3file_name, format="mp3")
         return spec, mp3file_name
     else: 
-        result = text_to_music(prompt, fn_index=0)
+        #result = text_to_music(prompt, fn_index=0)
+        result = generate_track_by_prompt(prompt)
         
         print(f"""—————
         NEW RESULTS