Initial commit: CapCutAPI - 轻量、灵活、易上手的剪映/CapCut API工具

add_subtitle_impl.py

@@ -0,0 +1,160 @@
+import pyJianYingDraft as draft
+from util import generate_draft_url, hex_to_rgb
+from create_draft import get_or_create_draft
+from pyJianYingDraft.text_segment import TextBubble, TextEffect
+from typing import Optional
+import requests
+
+def add_subtitle_impl(
+    srt_path: str,
+    draft_id: str = None,
+    track_name: str = "subtitle",
+    time_offset: float = 0,
+    # 字体样式参数
+    font_size: float = 8.0,
+    bold: bool = False,
+    italic: bool = False,
+    underline: bool = False,
+    font_color: str = "#FFFFFF",
+    
+    # 描边参数
+    border_alpha: float = 1.0,
+    border_color: str = "#000000",
+    border_width: float = 0.0,  # 默认不显示描边
+    
+    # 背景参数
+    background_color: str = "#000000",
+    background_style: int = 1,
+    background_alpha: float = 0.0,  # 默认不显示背景
+    
+    # 气泡效果
+    bubble_effect_id: Optional[str] = None,
+    bubble_resource_id: Optional[str] = None,
+    
+    # 文本花字
+    effect_effect_id: Optional[str] = None,
+    # 图像调节参数
+    transform_x: float = 0.0,
+    transform_y: float = -0.8,  # 字幕默认位置在底部
+    scale_x: float = 1.0,
+    scale_y: float = 1.0,
+    rotation: float = 0.0,
+    style_reference: draft.Text_segment = None,
+    vertical: bool = True,  # 新增参数：是否垂直显示
+    alpha: float = 0.4,
+    width: int = 1080,  # 新增参数
+    height: int = 1920  # 新增参数
+):
+    """
+    向草稿添加字幕
+    :param srt_path: 字幕文件路径或URL或SRT文本内容
+    :param draft_id: 草稿ID，如果为None则创建新草稿
+    :param track_name: 轨道名称，默认为"subtitle"
+    :param time_offset: 时间偏移量，默认为"0"s
+    :param text_style: 文本样式，默认为None
+    :param clip_settings: 片段设置，默认为None
+    :param style_reference: 样式参考，默认为None
+    :return: 草稿信息
+    """
+    # 获取或创建草稿
+    draft_id, script = get_or_create_draft(
+        draft_id=draft_id,
+        width=width,
+        height=height
+    )
+    
+    # 处理字幕内容
+    srt_content = None
+    
+    # 检查是否为URL
+    if srt_path.startswith(('http://', 'https://')):
+        try:
+            response = requests.get(srt_path)
+            response.raise_for_status()
+
+            response.encoding = 'utf-8'
+            srt_content = response.text
+        except Exception as e:
+            raise Exception(f"下载字幕文件失败: {str(e)}")
+    else:
+        # 如果不是URL，直接使用内容
+        srt_content = srt_path
+        # 处理可能包含的转义字符
+        srt_content = srt_content.replace('\\n', '\n').replace('/n', '\n')
+    
+    # 导入字幕
+    # 转换十六进制颜色为RGB
+    rgb_color = hex_to_rgb(font_color)
+
+    # 创建text_border (描边)
+    text_border = None
+    if border_width > 0:
+        text_border = draft.Text_border(
+            alpha=border_alpha,
+            color=hex_to_rgb(border_color),
+            width=border_width
+        )
+    
+    # 创建text_background (背景)
+    text_background = None
+    if background_alpha > 0:
+        text_background = draft.Text_background(
+            color=background_color,
+            style=background_style,
+            alpha=background_alpha
+        )
+    
+    # 创建text_style
+    text_style = draft.Text_style(
+        size=font_size,
+        bold=bold,
+        italic=italic,
+        underline=underline,
+        color=rgb_color,
+        align=1,  # 保持居中对齐
+        vertical=vertical,  # 使用传入的vertical参数
+        alpha=alpha  # 使用传入的alpha参数
+    )
+    
+    # 创建气泡效果
+    text_bubble = None
+    if bubble_effect_id and bubble_resource_id:
+        text_bubble = TextBubble(
+            effect_id=bubble_effect_id,
+            resource_id=bubble_resource_id
+        )
+    
+    # 创建花字效果
+    text_effect = None
+    if effect_effect_id:
+        text_effect = TextEffect(
+            effect_id=effect_effect_id,
+            resource_id=effect_effect_id
+        )
+    
+    # 创建clip_settings
+    clip_settings = draft.Clip_settings(
+        transform_x=transform_x,
+        transform_y=transform_y,
+        scale_x=scale_x,
+        scale_y=scale_y,
+        rotation=rotation
+    )
+
+    script.import_srt(
+        srt_content,
+        track_name=track_name,
+        time_offset=int(time_offset * 1000000),  # 将秒转换为微秒
+        text_style=text_style,
+        clip_settings=clip_settings,
+        style_reference=style_reference,
+        border=text_border,
+        background=text_background,
+        bubble=text_bubble,
+        effect=text_effect
+    )
+
+    return {
+        "draft_id": draft_id,
+        "draft_url": generate_draft_url(draft_id)
+    }