calderbuild
752adfc7bd
- Add base_resp status code checks to all MiniMax API agents - Surface clear error messages (e.g. "insufficient balance") instead of NoneType errors - Fix duration select value parsing (18-3/30-3/30-5 format) - Add inline SVG favicon, use relative API paths - Mount frontend static files in FastAPI
75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
"""Script Agent: refines the Director's outline into production-ready prompts."""
|
|
import json
|
|
import requests
|
|
from backend.config import MINIMAX_BASE_URL, HEADERS
|
|
|
|
|
|
SYSTEM_PROMPT = """You are the Scriptwriter of an AI film production team. You take the Director's shot outline and refine each shot into production-ready specifications.
|
|
|
|
For each shot, enhance:
|
|
1. video_prompt: a highly detailed, specific prompt optimized for AI video generation (Hailuo). Include: subject, action, camera movement, lighting, color palette, style. Be vivid and specific. Max 200 words.
|
|
2. narration: polished voiceover text, timed to fit the shot duration. Keep it concise and impactful.
|
|
3. subtitle: clean subtitle text (shorter than narration if needed)
|
|
|
|
Respond ONLY with valid JSON. Format:
|
|
{
|
|
"shots": [
|
|
{
|
|
"shot_number": 1,
|
|
"duration": 10,
|
|
"video_prompt": "Detailed AI video generation prompt...",
|
|
"narration": "Polished voiceover text...",
|
|
"subtitle": "Clean subtitle text...",
|
|
"mood": "inspiring"
|
|
}
|
|
]
|
|
}"""
|
|
|
|
|
|
def run(outline: dict) -> dict:
|
|
"""Refine Director's outline into production-ready shot specifications."""
|
|
user_message = (
|
|
f"Video title: {outline['title']}\n"
|
|
f"Visual style: {outline['style']}\n"
|
|
f"Total duration: {outline['total_duration']}s\n\n"
|
|
f"Director's outline:\n{json.dumps(outline['shots'], indent=2)}\n\n"
|
|
f"Refine each shot with detailed video prompts optimized for AI video generation."
|
|
)
|
|
|
|
url = f"{MINIMAX_BASE_URL}/text/chatcompletion_v2"
|
|
payload = {
|
|
"model": "MiniMax-M1",
|
|
"messages": [
|
|
{"role": "system", "content": SYSTEM_PROMPT},
|
|
{"role": "user", "content": user_message},
|
|
],
|
|
"temperature": 0.7,
|
|
"max_tokens": 3000,
|
|
}
|
|
|
|
resp = requests.post(url, headers=HEADERS, json=payload)
|
|
resp.raise_for_status()
|
|
data = resp.json()
|
|
|
|
base_resp = data.get("base_resp", {})
|
|
if base_resp.get("status_code", 0) != 0:
|
|
raise RuntimeError(f"MiniMax API error: {base_resp.get('status_msg', 'unknown')} (code {base_resp.get('status_code')})")
|
|
|
|
content = data.get("choices", [{}])[0].get("message", {}).get("content", "")
|
|
if not content:
|
|
raise RuntimeError("MiniMax API returned empty response")
|
|
content = content.strip()
|
|
if content.startswith("```"):
|
|
content = content.split("\n", 1)[1] if "\n" in content else content[3:]
|
|
if content.endswith("```"):
|
|
content = content[:-3]
|
|
content = content.strip()
|
|
|
|
result = json.loads(content)
|
|
|
|
# Carry over metadata from outline
|
|
result["title"] = outline["title"]
|
|
result["style"] = outline["style"]
|
|
result["total_duration"] = outline["total_duration"]
|
|
return result
|