From 6ba9d1aab9f36df33ce0c49352402f8d14822b90 Mon Sep 17 00:00:00 2001 From: sun-guannan Date: Sun, 13 Jul 2025 16:12:23 +0800 Subject: [PATCH] add video impl --- add_video_track.py | 158 ++++++++++++++++++++++----------------------- 1 file changed, 79 insertions(+), 79 deletions(-) diff --git a/add_video_track.py b/add_video_track.py index bade303..1de2a93 100644 --- a/add_video_track.py +++ b/add_video_track.py @@ -25,134 +25,134 @@ def add_video_track( speed: float = 1.0, track_name: str = "main", relative_index: int = 0, - duration: Optional[float] = None, # 新增 duration 参数 - transition: Optional[str] = None, # 转场类型 - transition_duration: Optional[float] = 0.5, # 转场时长（秒） - # 蒙版相关参数 - mask_type: Optional[str] = None, # 蒙版类型 - mask_center_x: float = 0.5, # 蒙版中心X坐标（0-1） - mask_center_y: float = 0.5, # 蒙版中心Y坐标（0-1） - mask_size: float = 1.0, # 蒙版大小（0-1） - mask_rotation: float = 0.0, # 蒙版旋转角度（度） - mask_feather: float = 0.0, # 蒙版羽化程度（0-1） - mask_invert: bool = False, # 是否反转蒙版 - mask_rect_width: Optional[float] = None, # 矩形蒙版宽度(仅矩形蒙版) - mask_round_corner: Optional[float] = None, # 矩形蒙版圆角(仅矩形蒙版，0-100) - volume: float = 1.0 # 音量大小，默认1.0 + duration: Optional[float] = None, # Added duration parameter + transition: Optional[str] = None, # Transition type + transition_duration: Optional[float] = 0.5, # Transition duration (seconds) + # Mask related parameters + mask_type: Optional[str] = None, # Mask type + mask_center_x: float = 0.5, # Mask center X coordinate (0-1) + mask_center_y: float = 0.5, # Mask center Y coordinate (0-1) + mask_size: float = 1.0, # Mask size (0-1) + mask_rotation: float = 0.0, # Mask rotation angle (degrees) + mask_feather: float = 0.0, # Mask feather level (0-1) + mask_invert: bool = False, # Whether to invert mask + mask_rect_width: Optional[float] = None, # Rectangle mask width (only for rectangle mask) + mask_round_corner: Optional[float] = None, # Rectangle mask rounded corner (only for rectangle mask, 0-100) + volume: float = 1.0 # Volume level, default 1.0 ) -> Dict[str, str]: """ - 向指定草稿添加视频轨道 - :param draft_folder: 草稿文件夹路径，可选参数 - :param video_url: 视频URL - :param width: 视频宽度，默认1080 - :param height: 视频高度，默认1920 - :param start: 源视频开始时间（秒），默认0 - :param end: 源视频结束时间（秒），默认None（使用视频总时长） - :param target_start: 目标视频开始时间（秒），默认0 - :param draft_id: 草稿ID，如果为None或找不到对应的zip文件，则创建新草稿 - :param transform_y: Y轴变换，默认0 - :param scale_x: X轴缩放，默认1 - :param scale_y: Y轴缩放，默认1 - :param transform_x: X轴变换，默认0 - :param speed: 视频播放速度，默认1.0 - :param track_name: 当只有一个视频的时候，轨道名称可以省略 - :param relative_index: 轨道渲染顺序索引，默认0 - :param duration: 视频时长（秒），如果提供则跳过时长检测 - :param transition: 转场类型，可选参数 - :param transition_duration: 转场时长（秒），默认使用转场类型的默认时长 - :param mask_type: 蒙版类型（线性、镜面、圆形、矩形、爱心、星形），可选参数 - :param mask_center_x: 蒙版中心X坐标（0-1），默认0.5 - :param mask_center_y: 蒙版中心Y坐标（0-1），默认0.5 - :param mask_size: 蒙版大小（0-1），默认1.0 - :param mask_rotation: 蒙版旋转角度（度），默认0.0 - :param mask_feather: 蒙版羽化程度（0-1），默认0.0 - :param mask_invert: 是否反转蒙版，默认False - :param mask_rect_width: 矩形蒙版的宽度，仅在蒙版类型为矩形时允许设置，以占素材宽度的比例表示 - :param mask_round_corner: 矩形蒙版的圆角参数，仅在蒙版类型为矩形时允许设置，取值范围0~100 - :param volume: 音量大小，默认1.0（0.0为静音，1.0为原音量） - :return: 更新后的草稿信息，包含draft_id和draft_url + Add video track to specified draft + :param draft_folder: Draft folder path, optional parameter + :param video_url: Video URL + :param width: Video width, default 1080 + :param height: Video height, default 1920 + :param start: Source video start time (seconds), default 0 + :param end: Source video end time (seconds), default None (use total video duration) + :param target_start: Target video start time (seconds), default 0 + :param draft_id: Draft ID, if None or corresponding zip file not found, create new draft + :param transform_y: Y-axis transform, default 0 + :param scale_x: X-axis scale, default 1 + :param scale_y: Y-axis scale, default 1 + :param transform_x: X-axis transform, default 0 + :param speed: Video playback speed, default 1.0 + :param track_name: When there is only one video, track name can be omitted + :param relative_index: Track rendering order index, default 0 + :param duration: Video duration (seconds), if provided, skip duration detection + :param transition: Transition type, optional parameter + :param transition_duration: Transition duration (seconds), default uses the default duration of transition type + :param mask_type: Mask type (linear, mirror, circle, rectangle, heart, star), optional parameter + :param mask_center_x: Mask center X coordinate (0-1), default 0.5 + :param mask_center_y: Mask center Y coordinate (0-1), default 0.5 + :param mask_size: Mask size (0-1), default 1.0 + :param mask_rotation: Mask rotation angle (degrees), default 0.0 + :param mask_feather: Mask feather level (0-1), default 0.0 + :param mask_invert: Whether to invert mask, default False + :param mask_rect_width: Rectangle mask width, only allowed when mask type is rectangle, represented as a proportion of material width + :param mask_round_corner: Rectangle mask rounded corner parameter, only allowed when mask type is rectangle, range 0~100 + :param volume: Volume level, default 1.0 (0.0 is mute, 1.0 is original volume) + :return: Updated draft information, including draft_id and draft_url """ - # 获取或创建草稿 + # Get or create draft draft_id, script = get_or_create_draft( draft_id=draft_id, width=width, height=height ) - # 检查是否存在视频轨道，如果不存在则添加一条默认视频轨道 + # Check if video track exists, if not, add a default video track try: script.get_track(draft.Track_type.video, track_name=None) except exceptions.TrackNotFound: script.add_track(draft.Track_type.video, relative_index=0) except NameError: - # 如果存在多个视频轨道(NameError)，则不做任何操作 + # If multiple video tracks exist (NameError), do nothing pass - # 添加视频轨道（仅当轨道不存在时） + # Add video track (only when track doesn't exist) if track_name is not None: try: imported_track=script.get_imported_track(draft.Track_type.video, name=track_name) - # 如果没有抛出异常，说明轨道已存在 + # If no exception is thrown, the track already exists except exceptions.TrackNotFound: - # 轨道不存在，创建新轨道 + # Track doesn't exist, create new track script.add_track(draft.Track_type.video, track_name=track_name, relative_index=relative_index) else: script.add_track(draft.Track_type.video, relative_index=relative_index) - # 如果传递了 duration 参数，优先使用它；否则使用默认时长0秒，在下载草稿时，才会获取真实时长 + # If duration parameter is passed, use it preferentially; otherwise use default duration of 0 seconds, and get the real duration when downloading the draft if duration is not None: - # 使用传递进来的 duration，跳过时长获取和检查 + # Use the passed duration, skip duration retrieval and check video_duration = duration else: - # 使用默认时长0秒，在下载草稿时，才会获取真实时长 - video_duration = 0.0 # 默认视频时长为0秒 + # Use default duration of 0 seconds, and get the real duration when downloading the draft + video_duration = 0.0 # Default video duration is 0 seconds # duration_result = get_video_duration(video_url) # if not duration_result["success"]: - # print(f"获取视频时长失败: {duration_result['error']}") + # print(f"Failed to get video duration: {duration_result['error']}") - # # 检查视频时长是否超过2分钟 - # if duration_result["output"] > 120: # 120秒 = 2分钟 - # raise Exception(f"视频时长超过2分钟限制，当前时长: {duration_result['output']}秒") + # # Check if video duration exceeds 2 minutes + # if duration_result["output"] > 120: # 120 seconds = 2 minutes + # raise Exception(f"Video duration exceeds 2-minute limit, current duration: {duration_result['output']} seconds") # video_duration = duration_result["output"] - # 生成本地文件名 + # Generate local filename material_name = f"video_{url_to_hash(video_url)}.mp4" # local_video_path = download_video(video_url, draft_dir) - # 构建draft_video_path + # Build draft_video_path draft_video_path = None if draft_folder: - # 检测输入路径类型并处理 + # Detect input path type and process if is_windows_path(draft_folder): - # Windows路径处理 + # Windows path processing windows_drive, windows_path = re.match(r'([a-zA-Z]:)(.*)', draft_folder).groups() - parts = [p for p in windows_path.split('\\') if p] # 分割路径并过滤空部分 + parts = [p for p in windows_path.split('\\') if p] # Split path and filter empty parts draft_video_path = os.path.join(windows_drive, *parts, draft_id, "assets", "video", material_name) - # 规范化路径（确保分隔符一致） + # Normalize path (ensure consistent separators) draft_video_path = draft_video_path.replace('/', '\\') else: - # macOS/Linux路径处理 + # macOS/Linux path processing draft_video_path = os.path.join(draft_folder, draft_id, "assets", "video", material_name) - # 打印路径信息 + # Print path information print('replace_path:', draft_video_path) - # 设置视频结束时间 + # Set video end time video_end = end if end is not None else video_duration - # 计算源视频时长 + # Calculate source video duration source_duration = video_end - start - # 计算目标视频时长（考虑速度因素） + # Calculate target video duration (considering speed factor) target_duration = source_duration / speed - # 创建视频片段 + # Create video clip if draft_video_path: video_material = draft.Video_material(material_type='video', replace_path=draft_video_path, remote_url=video_url, material_name=material_name, duration=video_duration, width=0, height=0) else: video_material = draft.Video_material(material_type='video', remote_url=video_url, material_name=material_name, duration = video_duration, width=0, height=0) - # 创建source_timerange和target_timerange + # Create source_timerange and target_timerange source_timerange = trange(f"{start}s", f"{source_duration}s") target_timerange = trange(f"{target_start}s", f"{target_duration}s") @@ -170,24 +170,24 @@ def add_video_track( volume=volume ) - # 添加转场效果 + # Add transition effect if transition: try: - # 获取转场类型 + # Get transition type if IS_CAPCUT_ENV: transition_type = getattr(draft.CapCut_Transition_type, transition) else: transition_type = getattr(draft.Transition_type, transition) - # 设置转场时长（转换为微秒） + # Set transition duration (convert to microseconds) duration_microseconds = int(transition_duration * 1e6) - # 添加转场 + # Add transition video_segment.add_transition(transition_type, duration=duration_microseconds) except AttributeError: - raise ValueError(f"不支持的转场类型: {transition}，已跳过转场设置") + raise ValueError(f"Unsupported transition type: {transition}, transition setting skipped") - # 添加蒙版效果 + # Add mask effect if mask_type: try: if IS_CAPCUT_ENV: @@ -207,9 +207,9 @@ def add_video_track( round_corner=mask_round_corner ) except: - raise ValueError(f"不支持的蒙版类型 {mask_type}，支持的类型包括：线性、镜面、圆形、矩形、爱心、星形") + raise ValueError(f"Unsupported mask type {mask_type}, supported types include: linear, mirror, circle, rectangle, heart, star") - # 添加视频片段到轨道 + # Add video segment to track # if imported_track is not None: # imported_track.add_segment(video_segment) # else: