'commit'
This commit is contained in:
256
dsLightRag/Routes/SunoRoute.py
Normal file
256
dsLightRag/Routes/SunoRoute.py
Normal file
@@ -0,0 +1,256 @@
|
||||
import logging
|
||||
import json
|
||||
import time
|
||||
import datetime
|
||||
import requests
|
||||
from typing import Optional
|
||||
|
||||
import fastapi
|
||||
from fastapi import APIRouter, HTTPException, Query
|
||||
from pydantic import BaseModel
|
||||
|
||||
from Routes.suno_music_generator import SunoMusicGenerator
|
||||
from Config import Config
|
||||
|
||||
# 创建路由路由器
|
||||
router = APIRouter(prefix="/api/suno", tags=["音乐"])
|
||||
|
||||
# 配置日志
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# 获取API密钥
|
||||
AK = Config.GPTNB_API_KEY
|
||||
|
||||
# 请求模型
|
||||
class MusicGenerateRequest(BaseModel):
|
||||
prompt: str
|
||||
make_instrumental: Optional[bool] = True
|
||||
|
||||
# 初始化音乐生成器
|
||||
music_generator = SunoMusicGenerator(AK)
|
||||
|
||||
# 任务状态存储(实际应用中可使用数据库)
|
||||
music_tasks = {}
|
||||
|
||||
|
||||
@router.post("/prompt_input")
|
||||
async def prompt_input(request: MusicGenerateRequest):
|
||||
"""
|
||||
生成音乐任务接口
|
||||
:param request: 包含提示词和是否纯音乐的请求体
|
||||
:return: 任务ID和状态信息
|
||||
"""
|
||||
prompt = request.prompt
|
||||
make_instrumental = request.make_instrumental
|
||||
|
||||
if not prompt:
|
||||
raise HTTPException(status_code=400, detail="缺少提示词参数")
|
||||
|
||||
try:
|
||||
logger.info(f"开始生成音乐任务,提示词: {prompt}")
|
||||
|
||||
# 调用音乐生成器生成音乐
|
||||
# 注意:我们只执行生成请求,不等待结果,因为这是一个异步过程
|
||||
# 构建JSON请求体
|
||||
request_json = {
|
||||
"gpt_description_prompt": prompt,
|
||||
"mv": "chirp-v3-5",
|
||||
"prompt": "",
|
||||
"make_instrumental": make_instrumental
|
||||
}
|
||||
|
||||
# 设置请求头
|
||||
headers = {
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": f"Bearer {AK}"
|
||||
}
|
||||
|
||||
# 执行生成请求
|
||||
response = requests.post(
|
||||
SunoMusicGenerator.GENERATE_URL,
|
||||
headers=headers,
|
||||
json=request_json,
|
||||
timeout=30
|
||||
)
|
||||
response.raise_for_status()
|
||||
|
||||
# 解析响应
|
||||
generate_json = response.json()
|
||||
logger.info(f"音乐生成响应: {generate_json}")
|
||||
|
||||
# 提取任务ID
|
||||
task_id = None
|
||||
if "id" in generate_json:
|
||||
task_id = generate_json["id"]
|
||||
elif "task_id" in generate_json:
|
||||
task_id = generate_json["task_id"]
|
||||
elif "clip_id" in generate_json:
|
||||
task_id = generate_json["clip_id"]
|
||||
|
||||
if task_id is None:
|
||||
raise HTTPException(status_code=500, detail="无法从响应中提取任务ID")
|
||||
|
||||
# 存储任务信息
|
||||
music_tasks[task_id] = {
|
||||
"status": "processing",
|
||||
"prompt": prompt,
|
||||
"create_time": datetime.datetime.now().isoformat(),
|
||||
"response": generate_json
|
||||
}
|
||||
|
||||
logger.info(f"音乐生成任务已提交,任务ID: {task_id}")
|
||||
|
||||
return {
|
||||
"code": 200,
|
||||
"message": "音乐生成任务已提交",
|
||||
"data": {
|
||||
"task_id": task_id,
|
||||
"status": "processing"
|
||||
}
|
||||
}
|
||||
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"解析音乐生成响应失败: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"解析音乐生成响应失败: {str(e)}")
|
||||
except Exception as e:
|
||||
logger.error(f"音乐生成过程中发生错误: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"音乐生成过程中发生错误: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/check_task_status")
|
||||
async def check_task_status(task_id: str = Query(..., description="音乐生成任务ID")):
|
||||
"""
|
||||
检查音乐生成任务状态接口
|
||||
:param task_id: 音乐生成任务ID
|
||||
:return: 任务状态信息
|
||||
"""
|
||||
if not task_id:
|
||||
raise HTTPException(status_code=400, detail="缺少任务ID参数")
|
||||
|
||||
# 检查任务是否存在
|
||||
if task_id not in music_tasks:
|
||||
raise HTTPException(status_code=404, detail=f"任务ID不存在: {task_id}")
|
||||
|
||||
try:
|
||||
logger.info(f"检查任务状态,任务ID: {task_id}")
|
||||
|
||||
# 获取任务信息
|
||||
task_info = music_tasks[task_id]
|
||||
generate_json = task_info["response"]
|
||||
|
||||
# 构建查询URL
|
||||
url_builder = [SunoMusicGenerator.FEED_URL, "?"]
|
||||
|
||||
# 尝试从生成响应中获取clips的ID
|
||||
clip_ids = []
|
||||
if "clips" in generate_json:
|
||||
clips_array = generate_json["clips"]
|
||||
for clip in clips_array:
|
||||
if "id" in clip:
|
||||
clip_ids.append(clip["id"])
|
||||
|
||||
# 添加ids参数
|
||||
if clip_ids:
|
||||
ids_param = ",".join(clip_ids)
|
||||
url_builder.append(f"ids={ids_param}")
|
||||
logger.info(f"使用clips ID查询: {ids_param}")
|
||||
else:
|
||||
url_builder.append(f"ids={task_id}")
|
||||
logger.info(f"使用任务ID查询: {task_id}")
|
||||
|
||||
url = "".join(url_builder)
|
||||
logger.info(f"查询URL: {url}")
|
||||
|
||||
# 设置请求头
|
||||
headers = {
|
||||
"Authorization": f"Bearer {AK}",
|
||||
"Accept": "application/json"
|
||||
}
|
||||
|
||||
# 执行查询请求
|
||||
response = requests.get(url, headers=headers, timeout=30)
|
||||
response.raise_for_status()
|
||||
|
||||
# 解析查询响应
|
||||
json_response = response.json()
|
||||
clips = json_response.get("clips", [])
|
||||
|
||||
if clips:
|
||||
# 遍历所有返回的音乐片段
|
||||
for clip in clips:
|
||||
clip_id = clip.get("id")
|
||||
status = clip.get("status")
|
||||
title = clip.get("title")
|
||||
audio_url = clip.get("audio_url")
|
||||
|
||||
logger.info(f"查询结果:")
|
||||
logger.info(f"ID: {clip_id}")
|
||||
logger.info(f"标题: {title}")
|
||||
logger.info(f"状态: {status}")
|
||||
|
||||
# 更新任务状态
|
||||
task_info["status"] = status
|
||||
task_info["last_check_time"] = datetime.datetime.now().isoformat()
|
||||
task_info["title"] = title
|
||||
|
||||
if status == "complete":
|
||||
# 确保audio_url字段存在
|
||||
if audio_url:
|
||||
# 移除URL中可能存在的反引号
|
||||
audio_url = audio_url.replace("`", "").strip()
|
||||
task_info["audio_url"] = audio_url
|
||||
logger.info("音乐生成已完成!")
|
||||
logger.info(f"音频URL: {audio_url}")
|
||||
|
||||
# 下载音频文件
|
||||
file_name = f"suno_music_{int(time.time())}.mp3"
|
||||
save_path = music_generator.base_path / file_name
|
||||
if music_generator.download_audio(audio_url, str(save_path)):
|
||||
task_info["local_path"] = str(save_path)
|
||||
else:
|
||||
logger.warning("音乐生成已完成,但未找到音频URL!")
|
||||
|
||||
elif status == "failed":
|
||||
logger.error("音乐生成失败!")
|
||||
task_info["error_message"] = clip.get("error_message", "未知错误")
|
||||
|
||||
# 更新任务存储
|
||||
music_tasks[task_id] = task_info
|
||||
|
||||
return {
|
||||
"code": 200,
|
||||
"message": "查询成功",
|
||||
"data": {
|
||||
"task_id": task_id,
|
||||
"status": status,
|
||||
"title": title,
|
||||
"audio_url": audio_url if status == "complete" else None,
|
||||
"local_path": task_info.get("local_path") if status == "complete" else None,
|
||||
"error_message": task_info.get("error_message") if status == "failed" else None
|
||||
}
|
||||
}
|
||||
else:
|
||||
logger.info("未找到音乐片段")
|
||||
return {
|
||||
"code": 200,
|
||||
"message": "查询成功",
|
||||
"data": {
|
||||
"task_id": task_id,
|
||||
"status": "processing",
|
||||
"title": None,
|
||||
"audio_url": None,
|
||||
"local_path": None,
|
||||
"error_message": None
|
||||
}
|
||||
}
|
||||
|
||||
except fastapi.Requests.exceptions.RequestException as e:
|
||||
logger.error(f"查询任务状态失败: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"查询任务状态失败: {str(e)}")
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"解析查询响应失败: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"解析查询响应失败: {str(e)}")
|
||||
except Exception as e:
|
||||
logger.error(f"查询任务状态过程中发生错误: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"查询任务状态过程中发生错误: {str(e)}")
|
||||
|
Reference in New Issue
Block a user