海螺AI(MiniMax)文生视频实战

import jsonimport timefrom typing import List, Dict, Optional  class HailuoVideoGenerator:    """海螺AI视频生成工具封装类（模拟API调用）"""     def __init__(self, api_key: str = "demo_key"):        self.api_key = api_key        self.aspect_ratios = {"9:16": (1080, 1920), "16:9": (1920, 1080), "1:1": (1080, 1080)}        self.generation_history = []     def build_prompt(        self,        subject: str,        action: str = "",        environment: str = "",        camera_movement: str = "",        style: str = "电影感画质, 柔和自然光"    ) -> str:        """构建结构化的视频生成提示词"""        parts = []        if subject:            parts.append(subject)        if action:            parts.append(action)        if environment:            parts.append(environment)        if camera_movement:            parts.append(camera_movement)        if style:            parts.append(style)        return ", ".join(parts)     def generate_video(        self,        prompt: str,        aspect_ratio: str = "9:16",        duration: int = 3,        motion_intensity: float = 0.5,        seed: Optional[int] = None,        reference_image: Optional[str] = None    ) -> Dict:        """生成视频（模拟API调用）"""        if aspect_ratio not in self.aspect_ratios:            raise ValueError(f"不支持的画幅比例: {aspect_ratio}")        if not 0 <= motion_intensity <= 1:            raise ValueError("运动强度应在0到1之间")        mode = "图生视频" if reference_image else "文生视频"        width, height = self.aspect_ratios[aspect_ratio]        result = {            "task_id": f"hailuo_{int(time.time())}",            "prompt": prompt,            "mode": mode,            "aspect_ratio": aspect_ratio,            "resolution": f"{width}x{height}",            "duration": duration,            "motion_intensity": motion_intensity,            "seed": seed if seed else int(time.time() * 1000) % 1000000,            "status": "completed",            "video_url": f"https://example.com/generated/{int(time.time())}.mp4",            "created_at": time.strftime("%Y-%m-%d %H:%M:%S")        }        self.generation_history.append(result)        return result     def batch_generate(self, prompt_list: List[Dict]) -> List[Dict]:        """批量生成多个视频"""        results = []        for params in prompt_list:            result = self.generate_video(**params)            results.append(result)            print(f"已生成: {result['task_id']} - {params.get('prompt', '')[:30]}...")        return results     def optimize_prompt(self, prompt: str, strategy: str = "balanced") -> str:        """提示词优化建议"""        style_keywords = {            "cinematic": "电影感光影, 浅景深, 柔和自然光, 高画质, 细节丰富",            "balanced": "高清画质, 自然光影, 细节清晰",            "minimal": "简洁干净, 主体突出"        }        if strategy not in style_keywords:            strategy = "balanced"        optimized = prompt + ", " + style_keywords[strategy]        return optimized     def get_history(self) -> List[Dict]:        """获取生成历史记录"""        return self.generation_history  if __name__ == "__main__":    generator = HailuoVideoGenerator()    prompt = generator.build_prompt(        subject="一个穿着白色连衣裙的年轻女性",        action="在花园中缓缓转身，裙摆自然摆动",        environment="阳光明媚的花园，鲜花盛开，蝴蝶飞舞",        camera_movement="中景镜头，缓慢环绕拍摄",        style="电影感画质，柔和自然光，梦幻氛围"    )    print(f"生成提示词: {prompt}")    print()    result = generator.generate_video(        prompt=prompt,        aspect_ratio="9:16",        duration=3,        motion_intensity=0.6,        seed=42    )    print(f"生成结果: {json.dumps(result, ensure_ascii=False, indent=2)}")    print()    sku_prompts = [        {            "prompt": generator.build_prompt(                subject="模特穿着白色雪纺连衣裙",                action="在花园中行走，裙摆随风飘动",                environment="春日花园，樱花飘落",                camera_movement="全身镜头，跟拍"            ),            "aspect_ratio": "9:16",            "motion_intensity": 0.5        },        {            "prompt": generator.build_prompt(                subject="模特穿着白色连衣裙",                action="转身微笑，发丝轻扬",                environment="海边沙滩，金色夕阳",                camera_movement="中景，慢动作"            ),            "aspect_ratio": "9:16",            "motion_intensity": 0.4        }    ]    print("批量生成电商视频...")    batch_results = generator.batch_generate(sku_prompts)    print(f"\n批量生成完成，共 {len(batch_results)} 个视频")