TextGen/suno.py

import time
import requests
import os

base_url = os.environ["VERCEL_API"]

def custom_generate_audio(payload):
 url = f"{base_url}/api/custom_generate"
 response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
 return response.json()


def extend_audio(payload):
 url = f"{base_url}/api/extend_audio"
 response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
 return response.json()

def generate_audio_by_prompt(payload):
 url = f"{base_url}/api/generate"
 response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
 return response.json()


def get_audio_information(audio_ids):
 url = f"{base_url}/api/get?ids={audio_ids}"
 response = requests.get(url)
 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)
 return response.json()

def get_clip(clip_id):
 url = f"{base_url}/api/clip?id={clip_id}"
 response = requests.get(url)
 return response.json()

def generate_whole_song(clip_id):
 payloyd = {"clip_id": clip_id}
 url = f"{base_url}/api/concat"
 response = requests.post(url, json=payload)
 return response.json()
 
def generate_lyrics(prompt):
 payload = {"prompt": prompt}
 url = f"{base_url}/api/generate_lyrics"
 print(url)
 response = requests.post(url, json=payload)
 print(response)
 return response.json()