Update TextGen/suno.py

TextGen/suno.py

@@ -27,6 +27,11 @@ def get_audio_information(audio_ids):
     return response.json()
 
 
+def generate_lyrics(payload):
+    url = f"{base_url}/api/generate_lyrics"
+    response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
+    return response.json()
+    
 def get_quota_information():
     url = f"{base_url}/api/get_limit"
     response = requests.get(url)
@@ -50,22 +55,3 @@ def generate_lyrics(prompt):
     response = requests.post(url, json=payload)
     print(response)
     return response.json()
-
-if __name__ == '__main__':
-    data = generate_audio_by_prompt({
-        "prompt": "A popular heavy metal song about war, sung by a deep-voiced male singer, slowly and melodiously. The lyrics depict the sorrow of people after the war.",
-        "make_instrumental": False,
-        "wait_audio": False
-    })
-
-    ids = f"{data[0]['id']},{data[1]['id']}"
-    print(f"ids: {ids}")
-
-    for _ in range(60):
-        data = get_audio_information(ids)
-        if data[0]["status"] == 'streaming':
-            print(f"{data[0]['id']} ==> {data[0]['audio_url']}")
-            print(f"{data[1]['id']} ==> {data[1]['audio_url']}")
-            break
-        # sleep 5s
-        time.sleep(5)