Suno API 快速开始

import requests
import time

class SunoAPI:
    def __init__(self, api_key):
        self.api_key = api_key
        self.base_url = 'https://api.kie.ai/api/v1'
        self.headers = {
            'Authorization': f'Bearer {api_key}',
            'Content-Type': 'application/json'
        }
    
    def generate_music(self, prompt, **options):
        data = {
            'prompt': prompt,
            'customMode': options.get('customMode', False),
            'instrumental': options.get('instrumental', False),
            'model': options.get('model', 'V3_5'),
            'callBackUrl': options.get('callBackUrl', 'https://your-app.com/callback')
        }
        
        if options.get('style'):
            data['style'] = options['style']
        if options.get('title'):
            data['title'] = options['title']
        if options.get('negativeTags'):
            data['negativeTags'] = options['negativeTags']
        
        response = requests.post(f'{self.base_url}/generate', 
                               headers=self.headers, json=data)
        result = response.json()
        
        if not response.ok or result.get('code') != 200:
            raise Exception(f"生成失败: {result.get('msg', '未知错误')}")
        
        return result['data']['taskId']
    
    def extend_music(self, audio_id, **options):
        data = {
            'audioId': audio_id,
            'defaultParamFlag': options.get('defaultParamFlag', False),
            'model': options.get('model', 'V3_5'),
            'callBackUrl': options.get('callBackUrl', 'https://your-app.com/callback')
        }
        
        if options.get('prompt'):
            data['prompt'] = options['prompt']
        if options.get('style'):
            data['style'] = options['style']
        if options.get('title'):
            data['title'] = options['title']
        if options.get('continueAt'):
            data['continueAt'] = options['continueAt']
        
        response = requests.post(f'{self.base_url}/generate/extend', 
                               headers=self.headers, json=data)
        result = response.json()
        
        if not response.ok or result.get('code') != 200:
            raise Exception(f"延长失败: {result.get('msg', '未知错误')}")
        
        return result['data']['taskId']
    
    def generate_lyrics(self, prompt, callback_url):
        data = {
            'prompt': prompt,
            'callBackUrl': callback_url
        }
        
        response = requests.post(f'{self.base_url}/lyrics', 
                               headers=self.headers, json=data)
        result = response.json()
        
        if not response.ok or result.get('code') != 200:
            raise Exception(f"歌词生成失败: {result.get('msg', '未知错误')}")
        
        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':
                print("所有音轨生成成功！")
                return status['response']
            elif status['status'] == 'FIRST_SUCCESS':
                print("第一个音轨生成完成！")
                return status['response']
            elif status['status'] == 'TEXT_SUCCESS':
                print("歌词/文本生成成功！")
                return status['response']
            elif status['status'] == 'PENDING':
                print("任务等待处理中...")
            elif status['status'] == 'CREATE_TASK_FAILED':
                error_msg = status.get('errorMessage', '创建任务失败')
                print(f"错误信息: {error_msg}")
                raise Exception(error_msg)
            elif status['status'] == 'GENERATE_AUDIO_FAILED':
                error_msg = status.get('errorMessage', '音频生成失败')
                print(f"错误信息: {error_msg}")
                raise Exception(error_msg)
            elif status['status'] == 'CALLBACK_EXCEPTION':
                error_msg = status.get('errorMessage', '回调过程中出错')
                print(f"错误信息: {error_msg}")
                raise Exception(error_msg)
            elif status['status'] == 'SENSITIVE_WORD_ERROR':
                error_msg = status.get('errorMessage', '内容因敏感词被过滤')
                print(f"错误信息: {error_msg}")
                raise Exception(error_msg)
            else:
                print(f"未知状态: {status['status']}")
                if status.get('errorMessage'):
                    print(f"错误信息: {status['errorMessage']}")
            
            time.sleep(10)  # 等待10秒
        
        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}'})
        result = response.json()
        
        if not response.ok or result.get('code') != 200:
            raise Exception(f"查询状态失败: {result.get('msg', '未知错误')}")
        
        return result['data']

# 使用示例
def main():
    api = SunoAPI('YOUR_API_KEY')
    
    try:
        # 生成带歌词的音乐
        print('开始生成音乐...')
        task_id = api.generate_music(
            '一首关于童年回忆的怀旧民谣',
            customMode=True,
            instrumental=False,
            model='V4_5',
            style='民谣, 原声吉他, 怀旧',
            title='童年梦想'
        )
        
        # 等待完成
        print(f'任务ID: {task_id}。等待完成...')
        result = api.wait_for_completion(task_id)
        
        print('音乐生成成功！')
        print('生成的曲目：')
        for i, track in enumerate(result['response']['sunoData']):
            print(f"曲目 {i + 1}:")
            print(f"  标题: {track['title']}")
            print(f"  音频URL: {track['audioUrl']}")
            print(f"  时长: {track['duration']}秒")
            print(f"  标签: {track['tags']}")
        
        # 延长第一个曲目
        first_track = result['response']['sunoData'][0]
        print('\n延长第一个曲目...')
        extend_task_id = api.extend_music(
            first_track['id'],
            defaultParamFlag=True,
            prompt='继续一个充满希望的副歌',
            style='民谣, 振奋',
            title='童年梦想延长版',
            continueAt=60,
            model='V4_5'
        )
        
        extend_result = api.wait_for_completion(extend_task_id)
        print('音乐延长成功！')
        print(f"延长曲目URL: {extend_result['response']['sunoData'][0]['audioUrl']}")
        
    except Exception as error:
        print(f'错误: {error}')

if __name__ == '__main__':
    main()