This commit is contained in:
2025-08-21 14:16:13 +08:00
parent 42825a7d9b
commit 2918f9597d
6 changed files with 452 additions and 90 deletions

View 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)}")

View File

@@ -20,6 +20,7 @@ from Routes.TeachingModel.api.DocumentController import router as document_route
from Routes.TeachingModel.api.TeachingModelController import router as teaching_model_router
from Routes.QA import router as qa_router
from Routes.JiMengRoute import router as jimeng_router
from Routes.SunoRoute import router as suno_router
from Util.LightRagUtil import *
from contextlib import asynccontextmanager
@@ -44,6 +45,8 @@ async def lifespan(_: FastAPI):
finally:
# 应用关闭时销毁连接池
await close_postgres_pool(pool)
app = FastAPI(lifespan=lifespan)
# 挂载静态文件目录
@@ -56,8 +59,9 @@ app.include_router(rag_router) # LightRAG路由
app.include_router(knowledge_router) # 知识图谱路由
app.include_router(oss_router) # 阿里云OSS路由
app.include_router(llm_router) # 大模型路由
app.include_router(qa_router) # 答疑路由
app.include_router(jimeng_router) # 即梦路由
app.include_router(qa_router) # 答疑路由
app.include_router(jimeng_router) # 即梦路由
app.include_router(suno_router) # Suno路由
# Teaching Model 相关路由
# 登录相关(不用登录)
@@ -75,6 +79,5 @@ app.include_router(teaching_model_router, prefix="/api/teaching/model", tags=["t
# 教学答疑
app.include_router(teaching_model_router, prefix="/api/teaching/model", tags=["teacher_model"])
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8100)

View File

@@ -134,7 +134,7 @@
<option value="安静学习">安静学习</option>
<option value="欢快活泼">欢快活泼</option>
<option value="庄重严肃">庄重严肃</option>
<option value="温馨感">温馨感</option>
<option value="温馨感">温馨感</option>
</select>
<div class="form-helper">示例:「夏日海边」「深夜治愈」「热血励志」</div>
</div>
@@ -199,6 +199,7 @@
<div class="form-actions">
<button type="submit" class="btn btn-primary"><i class="fas fa-magic"></i> 生成音乐提示词</button>
<button type="button" id="randomSampleBtn" class="btn btn-info"><i class="fas fa-random"></i> 随机示例</button>
<button type="button" id="resetForm" class="btn btn-secondary"><i class="fas fa-undo"></i> 重置</button>
</div>
</form>
@@ -220,93 +221,195 @@
</div>
<script>
document.addEventListener('DOMContentLoaded', function() {
// 表单提交事件
const form = document.getElementById('musicPromptForm');
const resultContainer = document.getElementById('resultContainer');
const resultContent = document.getElementById('resultContent');
const resetFormBtn = document.getElementById('resetForm');
const copyResultBtn = document.getElementById('copyResult');
const generateMusicBtn = document.getElementById('generateMusic');
form.addEventListener('submit', function(e) {
e.preventDefault();
// 获取表单数据
const style = document.getElementById('style').value;
const mood = document.getElementById('mood').value;
const vocal = document.getElementById('vocal').value;
const tempo = document.getElementById('tempo').value;
const instruments = document.getElementById('instruments').value;
const theme = document.getElementById('theme').value;
const teachingSubject = document.getElementById('teachingSubject').value;
// 生成提示词
let prompt = `${style} ${mood}${vocal}${tempo},配${instruments},主题围绕'${theme}'`;
if (teachingSubject) {
prompt += `,适合${teachingSubject}教学使用`;
}
prompt += `,时长 2 分钟。`;
// 显示结果
resultContent.textContent = prompt;
resultContainer.style.display = 'block';
// 滚动到结果区域
resultContainer.scrollIntoView({ behavior: 'smooth' });
});
// 重置表单
resetFormBtn.addEventListener('click', function() {
form.reset();
resultContainer.style.display = 'none';
});
// 复制结果
copyResultBtn.addEventListener('click', function() {
const textToCopy = resultContent.textContent;
navigator.clipboard.writeText(textToCopy).then(function() {
// 显示复制成功提示
const originalText = copyResultBtn.innerHTML;
copyResultBtn.innerHTML = '<i class="fas fa-check"></i> 已复制';
setTimeout(function() {
copyResultBtn.innerHTML = originalText;
}, 2000);
});
});
// 生成音乐
generateMusicBtn.addEventListener('click', function() {
alert('音乐生成功能已触发,正在处理您的请求...');
// 这里可以添加实际的音乐生成逻辑
});
// 添加页面加载动画
const featureCards = document.querySelectorAll('.feature-card');
function checkElements() {
const triggerBottom = window.innerHeight * 0.8;
featureCards.forEach(card => {
const cardTop = card.getBoundingClientRect().top;
if (cardTop < triggerBottom) {
card.style.opacity = '1';
card.style.transform = 'translateY(0)';
document.addEventListener('DOMContentLoaded', function() {
// 音乐示例数据
const musicSamples = [
{
style: "华语流行",
mood: "欢快活泼",
vocal: "甜美女声",
tempo: "4/4 拍 120 BPM",
instruments: "吉他+贝斯+鼓",
theme: "快乐学习",
teachingSubject: "英语"
},
{
style: "古风",
mood: "国风江湖",
vocal: "清澈男声",
tempo: "4/4 拍 80 BPM",
instruments: "笛子+古筝",
theme: "诗词歌赋",
teachingSubject: "语文"
},
{
style: "Classical",
mood: "安静学习",
vocal: "钢琴弦乐",
tempo: "3/4 拍 90 BPM",
instruments: "钢琴弦乐",
theme: "数学的旋律",
teachingSubject: "数学"
},
{
style: "民谣",
mood: "温馨感动",
vocal: "温柔女声",
tempo: "4/4 拍 100 BPM",
instruments: "木吉他主奏",
theme: "自然科学的奥秘",
teachingSubject: "科学"
},
{
style: "City Pop",
mood: "夏日海边",
vocal: "中性嗓音",
tempo: "4/4 拍 120 BPM",
instruments: "电子合成器",
theme: "阳光沙滩海浪",
teachingSubject: "地理"
},
{
style: "电子 R&B",
mood: "深夜治愈",
vocal: "低沉男声",
tempo: "6/8 拍 85 BPM 慢摇",
instruments: "808 鼓点+萨克斯",
theme: "历史的回声",
teachingSubject: "历史"
},
{
style: "摇滚",
mood: "热血励志",
vocal: "高亢女声",
tempo: "4/4 拍 140 BPM",
instruments: "吉他+贝斯+鼓",
theme: "运动精神",
teachingSubject: "体育"
},
{
style: "爵士",
mood: "庄重严肃",
vocal: "沙哑烟嗓",
tempo: "4/4 拍 100 BPM",
instruments: "钢琴弦乐",
theme: "艺术欣赏",
teachingSubject: "美术"
}
];
// 获取DOM元素
const form = document.getElementById('musicPromptForm');
const resultContainer = document.getElementById('resultContainer');
const resultContent = document.getElementById('resultContent');
const resetFormBtn = document.getElementById('resetForm');
const copyResultBtn = document.getElementById('copyResult');
const generateMusicBtn = document.getElementById('generateMusic');
const randomSampleBtn = document.getElementById('randomSampleBtn');
// 随机示例按钮点击事件
randomSampleBtn.addEventListener('click', function() {
// 随机选择一个示例
const randomIndex = Math.floor(Math.random() * musicSamples.length);
const sample = musicSamples[randomIndex];
// 填充表单
document.getElementById('style').value = sample.style;
document.getElementById('mood').value = sample.mood;
document.getElementById('vocal').value = sample.vocal;
document.getElementById('tempo').value = sample.tempo;
document.getElementById('instruments').value = sample.instruments;
document.getElementById('theme').value = sample.theme;
document.getElementById('teachingSubject').value = sample.teachingSubject;
// 隐藏结果区域
resultContainer.style.display = 'none';
});
}
// 初始化样式
featureCards.forEach(card => {
card.style.opacity = '0';
card.style.transform = 'translateY(20px)';
card.style.transition = 'opacity 0.5s ease, transform 0.5s ease, delay 0.1s';
// 表单提交事件
form.addEventListener('submit', function(e) {
e.preventDefault();
// 获取表单数据
const style = document.getElementById('style').value;
const mood = document.getElementById('mood').value;
const vocal = document.getElementById('vocal').value;
const tempo = document.getElementById('tempo').value;
const instruments = document.getElementById('instruments').value;
const theme = document.getElementById('theme').value;
const teachingSubject = document.getElementById('teachingSubject').value;
// 生成提示词
let prompt = `${style} ${mood}${vocal}${tempo},配${instruments},主题围绕'${theme}'`;
if (teachingSubject) {
prompt += `,适合${teachingSubject}教学使用`;
}
prompt += `,时长 2 分钟。`;
// 显示结果
resultContent.textContent = prompt;
resultContainer.style.display = 'block';
});
// 重置表单按钮点击事件
resetFormBtn.addEventListener('click', function() {
form.reset();
resultContainer.style.display = 'none';
});
// 复制结果按钮点击事件
copyResultBtn.addEventListener('click', function() {
const textToCopy = resultContent.textContent;
navigator.clipboard.writeText(textToCopy).then(function() {
alert('提示词已复制到剪贴板!');
}).catch(function(err) {
alert('复制失败: ' + err);
});
});
// 生成音乐按钮点击事件
generateMusicBtn.addEventListener('click', function() {
const prompt = resultContent.textContent;
if (!prompt) {
alert('请先生成音乐提示词');
return;
}
// 显示加载状态
generateMusicBtn.disabled = true;
generateMusicBtn.innerHTML = '<i class="fas fa-spinner fa-spin"></i> 提交中...';
// 调用后端API生成音乐任务
fetch('/api/suno/prompt_input', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify({
prompt: prompt,
is_instrumental: false // 默认为带 vocals 的音乐
})
})
.then(response => response.json())
.then(data => {
generateMusicBtn.disabled = false;
generateMusicBtn.innerHTML = '<i class="fas fa-play"></i> 生成音乐';
if (data.task_id) {
// 跳转到进度页面
window.location.href = `/static/Suno/music_progress.html?task_id=${data.task_id}`;
} else {
alert('生成音乐任务失败: ' + (data.error || '未知错误'));
}
})
.catch(error => {
generateMusicBtn.disabled = false;
generateMusicBtn.innerHTML = '<i class="fas fa-play"></i> 生成音乐';
alert('请求失败: ' + error.message);
});
});
});
// 检查可见性
checkElements();
window.addEventListener('scroll', checkElements);
});
</script>
</script>
</body>
</html>