mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-11-09 20:40:36 +00:00
7e1f814b2c
* fix(skills): validate music audio before opening output files * fix(skills): reject non-MP3 music responses before writing --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
import argparse
|
|
import json
|
|
import os
|
|
|
|
import requests
|
|
|
|
MINIMAX_DEFAULT_HOST = "https://api.minimaxi.com"
|
|
|
|
|
|
def _check_base_resp(payload: dict) -> None:
|
|
base = payload.get("base_resp") or {}
|
|
if base.get("status_code", 0) != 0:
|
|
raise Exception(
|
|
f"MiniMax error {base.get('status_code')}: {base.get('status_msg')}"
|
|
)
|
|
|
|
|
|
def _has_mp3_header(audio: bytes) -> bool:
|
|
"""Check common MP3 signatures without decoding the audio stream."""
|
|
has_id3 = len(audio) >= 10 and audio.startswith(b"ID3")
|
|
has_mpeg_frame = len(audio) >= 4 and audio[0] == 0xFF and audio[1] & 0xE0 == 0xE0
|
|
return has_id3 or has_mpeg_frame
|
|
|
|
|
|
def generate_music(prompt_file: str, output_file: str) -> str:
|
|
"""Generate a song from a JSON spec via MiniMax /v1/music_generation.
|
|
|
|
Spec JSON: {"title": str, "prompt": str, "lyrics"?: str, "is_instrumental"?: bool}
|
|
- lyrics given -> use them (supports [Verse]/[Chorus] structure tags, \\n lines)
|
|
- is_instrumental true -> pure music, no lyrics needed
|
|
- otherwise -> lyrics_optimizer auto-writes lyrics from prompt
|
|
"""
|
|
with open(prompt_file, "r", encoding="utf-8") as f:
|
|
spec = json.load(f)
|
|
|
|
api_key = os.getenv("MINIMAX_API_KEY")
|
|
if not api_key:
|
|
return "MINIMAX_API_KEY is not set"
|
|
|
|
prompt = (spec.get("prompt") or "").strip()
|
|
if not prompt:
|
|
raise ValueError("`prompt` is required in the music spec")
|
|
lyrics = spec.get("lyrics") or None # treat empty string the same as absent
|
|
is_instrumental = bool(spec.get("is_instrumental", False))
|
|
|
|
body = {
|
|
"model": os.getenv("MINIMAX_MUSIC_MODEL", "music-2.6-free"),
|
|
"prompt": prompt,
|
|
"output_format": "hex",
|
|
"audio_setting": {"sample_rate": 44100, "bitrate": 256000, "format": "mp3"},
|
|
}
|
|
if lyrics:
|
|
body["lyrics"] = lyrics
|
|
elif is_instrumental:
|
|
body["is_instrumental"] = True
|
|
else:
|
|
body["lyrics_optimizer"] = True
|
|
|
|
host = os.getenv("MINIMAX_API_HOST", MINIMAX_DEFAULT_HOST).rstrip("/")
|
|
response = requests.post(
|
|
f"{host}/v1/music_generation",
|
|
headers={
|
|
"Authorization": f"Bearer {api_key}",
|
|
"Content-Type": "application/json",
|
|
},
|
|
json=body,
|
|
timeout=300,
|
|
)
|
|
response.raise_for_status()
|
|
payload = response.json()
|
|
_check_base_resp(payload)
|
|
audio_hex = (payload.get("data") or {}).get("audio")
|
|
if not audio_hex:
|
|
raise Exception("MiniMax returned no audio data")
|
|
# Decode before opening the output: malformed responses must not erase it.
|
|
audio = bytes.fromhex(audio_hex)
|
|
if not audio:
|
|
raise ValueError("MiniMax audio decoded to empty bytes")
|
|
# The request pins MP3; reject obvious non-audio responses before writing.
|
|
if not _has_mp3_header(audio):
|
|
raise ValueError("MiniMax audio does not have an MP3 header")
|
|
|
|
output_dir = os.path.dirname(output_file)
|
|
if output_dir:
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
with open(output_file, "wb") as f:
|
|
f.write(audio)
|
|
return f"Successfully generated music to {output_file}"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Generate music using MiniMax API")
|
|
parser.add_argument(
|
|
"--prompt-file",
|
|
required=True,
|
|
help="Absolute path to JSON spec file {title, prompt, lyrics?, is_instrumental?}",
|
|
)
|
|
parser.add_argument(
|
|
"--output-file", required=True, help="Output path for generated MP3"
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
print(generate_music(args.prompt_file, args.output_file))
|
|
except Exception as e:
|
|
print(f"Error while generating music: {e}")
|