Kling 可灵视频生成 API

import requests
import json

# Kling 可灵 AI 视频生成 API
def generate_video(prompt: str, duration: int = 5):
    """文本生成视频"""
    url = "https://api.n1n.ai/v1/video/generate"
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    
    data = {
        "model": "kling-v1.5",
        "prompt": prompt,
        "duration": duration,  # 视频时长（秒）: 5, 10
        "aspect_ratio": "16:9",  # 宽高比: 16:9, 9:16, 1:1
        "quality": "high",  # 质量: standard, high, ultra
        "fps": 30,  # 帧率
        "style": "realistic"  # 风格: realistic, anime, cartoon
    }
    
    response = requests.post(url, headers=headers, json=data)
    
    if response.status_code == 200:
        result = response.json()
        return result["video_url"], result["task_id"]
    else:
        raise Exception(f"生成失败: {response.text}")

# 图片生成视频
def image_to_video(image_url: str, motion_type: str = "auto"):
    """静态图片转动态视频"""
    data = {
        "model": "kling-v1.5",
        "image_url": image_url,
        "motion_type": motion_type,  # auto, zoom_in, zoom_out, pan_left, pan_right
        "motion_strength": 0.7,  # 运动强度 0-1
        "duration": 5
    }
    
    response = requests.post(
        "https://api.n1n.ai/v1/video/animate",
        headers=headers,
        json=data
    )
    
    return response.json()

# 视频编辑增强
def enhance_video(video_url: str, enhancements: dict):
    """视频后期处理"""
    data = {
        "source_video": video_url,
        "enhancements": {
            "resolution": "4K",  # 分辨率提升
            "fps_interpolation": 60,  # 帧率插值
            "stabilization": True,  # 视频防抖
            "color_grading": "cinematic",  # 调色风格
            "noise_reduction": True  # 降噪
        }
    }
    
    response = requests.post(
        "https://api.n1n.ai/v1/video/enhance",
        headers=headers,
        json=data
    )
    
    return response.json()

# 高级视频生成功能
class VideoCreator:
    def __init__(self, api_key: str):
        self.api_key = api_key
        self.base_url = "https://api.n1n.ai/v1/video"
    
    def create_story_video(self, script: list):
        """根据脚本生成故事视频"""
        scenes = []
        
        for scene in script:
            response = requests.post(
                f"{self.base_url}/scene",
                headers=self._get_headers(),
                json={
                    "description": scene["description"],
                    "duration": scene["duration"],
                    "transition": scene.get("transition", "fade"),
                    "camera_movement": scene.get("camera", "static"),
                    "style_reference": scene.get("style")
                }
            )
            scenes.append(response.json()["scene_id"])
        
        # 合并场景
        final_video = self.merge_scenes(scenes)
        return final_video
    
    def merge_scenes(self, scene_ids: list):
        """合并多个场景为完整视频"""
        response = requests.post(
            f"{self.base_url}/merge",
            headers=self._get_headers(),
            json={
                "scenes": scene_ids,
                "transitions": "smooth",
                "add_music": True,
                "export_format": "mp4"
            }
        )
        return response.json()
    
    def style_transfer(self, video_url: str, reference_image: str):
        """视频风格迁移"""
        response = requests.post(
            f"{self.base_url}/style-transfer",
            headers=self._get_headers(),
            json={
                "source_video": video_url,
                "style_reference": reference_image,
                "transfer_strength": 0.8,
                "preserve_content": True
            }
        )
        return response.json()
    
    def generate_variations(self, prompt: str, count: int = 3):
        """生成多个视频变体"""
        variations = []
        
        for i in range(count):
            response = requests.post(
                f"{self.base_url}/generate",
                headers=self._get_headers(),
                json={
                    "prompt": prompt,
                    "seed": i * 1000,  # 不同种子值
                    "variation_strength": 0.3 + (i * 0.2)
                }
            )
            variations.append(response.json())
        
        return variations

# 批量视频处理
def batch_process_videos(video_list: list):
    creator = VideoCreator(API_KEY)
    results = []
    
    for video_config in video_list:
        if video_config["type"] == "text_to_video":
            result = creator.generate_video(
                prompt=video_config["prompt"],
                duration=video_config["duration"]
            )
        elif video_config["type"] == "image_to_video":
            result = creator.image_to_video(
                image_url=video_config["image"],
                motion_type=video_config["motion"]
            )
        
        results.append(result)
    
    return results