Suno API 快速开始

import requests
import time

class SunoAPI:
    def __init__(self, api_key):
        self.api_key = api_key
        self.base_url = 'https://api.sunoapi.org/api/v1'
        self.headers = {
            'Authorization': f'Bearer {api_key}',
            'Content-Type': 'application/json'
        }
    
    def generate_music(self, **options):
        response = requests.post(f'{self.base_url}/generate', 
                               headers=self.headers, json=options)
        result = response.json()
        
        if result['code'] != 200:
            raise Exception(f"生成失败: {result['msg']}")
        
        return result['data']['taskId']
    
    def generate_lyrics(self, prompt):
        response = requests.post(f'{self.base_url}/lyrics',
                               headers=self.headers,
                               json={
                                   'prompt': prompt,
                                   'callBackUrl': 'https://your-server.com/lyrics-callback'
                               })
        result = response.json()
        return result['data']['taskId']
    
    def wait_for_completion(self, task_id, max_wait_time=600):
        start_time = time.time()
        
        while time.time() - start_time < max_wait_time:
            status = self.get_task_status(task_id)
            
            if status['status'] == 'SUCCESS':
                return status['response']
            elif status['status'] == 'FAILED':
                raise Exception(f"生成失败: {status.get('errorMessage')}")
            
            time.sleep(30)  # 等待30秒
        
        raise Exception('生成超时')
    
    def get_task_status(self, task_id):
        response = requests.get(f'{self.base_url}/generate/record-info?taskId={task_id}',
                              headers={'Authorization': f'Bearer {self.api_key}'})
        return response.json()['data']
    
    def extend_music(self, audio_id, **options):
        response = requests.post(f'{self.base_url}/generate/extend',
                               headers=self.headers,
                               json={'audioId': audio_id, **options})
        return response.json()['data']['taskId']
    
    def separate_vocals(self, task_id, audio_id):
        response = requests.post(f'{self.base_url}/vocal-removal/generate',
                               headers=self.headers,
                               json={
                                   'taskId': task_id,
                                   'audioId': audio_id,
                                   'callBackUrl': 'https://your-server.com/vocal-callback'
                               })
        return response.json()['data']['taskId']
    
    def get_remaining_credits(self):
        response = requests.get(f'{self.base_url}/get-credits',
                              headers={'Authorization': f'Bearer {self.api_key}'})
        return response.json()['data']['credits']

# 使用示例
def main():
    api = SunoAPI('YOUR_API_KEY')
    
    try:
        # 检查剩余积分
        credits = api.get_remaining_credits()
        print(f'剩余积分: {credits}')
        
        # 首先生成歌词
        print('正在生成歌词...')
        lyrics_task_id = api.generate_lyrics(
            '一首关于冒险和发现的歌曲，振奋人心且鼓舞人心'
        )
        
        lyrics_result = api.wait_for_completion(lyrics_task_id)
        print('歌词已生成:', lyrics_result['data'][0]['text'])
        
        # 使用自定义参数生成音乐
        print('正在生成音乐...')
        music_task_id = api.generate_music(
            prompt=lyrics_result['data'][0]['text'],
            customMode=True,
            style='民谣流行',
            title='冒险之歌',
            instrumental=False,
            model='V4_5',
            callBackUrl='https://your-server.com/music-callback'
        )
        
        # 等待完成
        music_result = api.wait_for_completion(music_task_id)
        print('音乐生成成功！')
        
        for i, track in enumerate(music_result['data']):
            print(f'音轨 {i + 1}:')
            print(f'  标题: {track["title"]}')
            print(f'  时长: {track["duration"]}秒')
            print(f'  音频URL: {track["audio_url"]}')
        
        # 扩展第一个音轨
        original_track = music_result['data'][0]
        print('正在扩展音乐...')
        extend_task_id = api.extend_music(
            original_track['id'],
            defaultParamFlag=True,
            prompt='继续添加美妙的器乐尾奏',
            continueAt=original_track['duration'] - 30,
            model='V4_5'
        )
        
        extended_result = api.wait_for_completion(extend_task_id)
        print('扩展版本已创建:', extended_result['data'][0]['audio_url'])
        
        # 分离人声
        print('正在分离人声...')
        separation_task_id = api.separate_vocals(music_task_id, original_track['id'])
        separation_result = api.wait_for_completion(separation_task_id)
        
        print('人声分离完成:')
        vocal_info = separation_result['vocal_removal_info']
        print(f'  伴奏: {vocal_info["instrumental_url"]}')
        print(f'  纯人声: {vocal_info["vocal_url"]}')
        
    except Exception as error:
        print(f'错误: {error}')

if __name__ == '__main__':
    main()