calderbuild
8c95f3da99
6 specialized AI agents collaborate through a structured SOP pipeline: Director (planning) -> Script (writing) -> Visual+Voice+Music (parallel generation) -> Editor (ffmpeg compositing) Backend: FastAPI with SSE streaming, MiniMax API integration (Hailuo video, Speech-2.6-HD TTS, Music-2.0, M1 LLM) Frontend: single-page app with real-time pipeline visualization
69 lines
2.4 KiB
Python
69 lines
2.4 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()
|
|
|
|
content = data.get("choices", [{}])[0].get("message", {}).get("content", "")
|
|
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
|