FilesExpand file tree

 main

/

test.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

80 lines (60 loc) · 2.71 KB

 main

/

test.py

Copy path

Top

File metadata and controls

• Code
• Blame

80 lines (60 loc) · 2.71 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

import json

import os

import time

import requests

from requests import get as rget

def test_generate_music():

data = {

"prompt": "[Verse]\nWake up in the morning, feeling brand new\nGonna shake off the worries, leave 'em in the rearview\nStep outside, feeling the warmth on my face\nThere's something 'bout the sunshine that puts me in my place\n\n[Verse 2]\nWalking down the street, got a spring in my step\nThe rhythm in my heart, it just won't forget\nEverywhere I go, people smiling at me\nThey can feel the joy, it's contagious, can't you see?\n\n[Chorus]\nI got sunshine in my pocket, happiness in my soul\nA skip in my stride, and I'm ready to go\nNothing gonna bring me down, gonna keep on shining bright\nI got sunshine in my pocket, this world feels so right",

"tags": "heartfelt anime",

"mv": "chirp-v3-0",

"title": "Sunshine in your Pocket",

"continue_clip_id": None,

"continue_at": None,

}

r = requests.post(

"http://127.0.0.1:8000/generate/description-mode", data=json.dumps(data)

)

resp = r.text

print(resp)

def test_generate_music_with_description():

data = {

"gpt_description_prompt": "A Blues song about a person who is feeling happy and optimistic about the future.",

"make_instrumental": False,

"mv": "chirp-v3-0",

}

r = requests.post("http://127.0.0.1:8000/generate", data=json.dumps(data))

resp = r.text

print(resp)

def test_generate_lyrics():

data = {"prompt": ""}

r = requests.post("http://127.0.0.1:8000/generate/lyrics/", data=json.dumps(data))

print(r.text)

def get_lyrics(lid):

r = requests.get(f"http://127.0.0.1:8000/lyrics/{lid}")

print(r.text)

def get_info(aid):

response = requests.get(f"http://127.0.0.1:8000/feed/{aid}")

data = json.loads(response.text)[0]

return data["audio_url"], data["metadata"]

def save_song(aid, output_path="output"):

start_time = time.time()

while True:

audio_url, metadata = get_info(aid)

if audio_url:

break

elif time.time() - start_time > 90:

raise TimeoutError("Failed to get audio_url within 90 seconds")

time.sleep(30)

response = rget(audio_url, allow_redirects=False, stream=True)

if response.status_code != 200:

raise Exception("Could not download song")

index = 0

while os.path.exists(os.path.join(output_path, f"suno_{index}.mp3")):

index += 1

path = os.path.join(output_path, f"suno_{index}.mp3")

with open(path, "wb") as output_file:

for chunk in response.iter_content(chunk_size=1024):

# If the chunk is not empty, write it to the file.

if chunk:

output_file.write(chunk)