mirror of
https://github.com/sun-guannan/CapCutAPI.git
synced 2025-11-25 03:15:00 +08:00
273 lines
11 KiB
Python
273 lines
11 KiB
Python
import pyJianYingDraft as draft
|
||
from settings.local import IS_CAPCUT_ENV
|
||
from util import generate_draft_url, hex_to_rgb
|
||
from pyJianYingDraft import trange, Font_type
|
||
from typing import Optional, List # add List type hint
|
||
from pyJianYingDraft import exceptions
|
||
from create_draft import get_or_create_draft
|
||
from pyJianYingDraft.text_segment import TextBubble, TextEffect, TextStyleRange
|
||
|
||
def add_text_impl(
|
||
text: str,
|
||
start: float,
|
||
end: float,
|
||
draft_id: str | None = None, # Python 3.10+ 新语法
|
||
transform_y: float = -0.8,
|
||
transform_x: float = 0,
|
||
font: Optional[str] = None,
|
||
font_color: str = "#ffffff",
|
||
font_size: float = 8.0,
|
||
track_name: str = "text_main",
|
||
vertical: bool = False,
|
||
font_alpha: float = 1.0,
|
||
# Border parameters
|
||
border_alpha: float = 1.0,
|
||
border_color: str = "#000000",
|
||
border_width: float = 0.0,
|
||
# Background parameters
|
||
background_color: str = "#000000",
|
||
background_style: int = 1,
|
||
background_alpha: float = 0.0,
|
||
background_round_radius: float = 0.0,
|
||
background_height: float = 0.14,
|
||
background_width: float = 0.14,
|
||
background_horizontal_offset: float = 0.5,
|
||
background_vertical_offset: float = 0.5,
|
||
# Shadow parameters
|
||
shadow_enabled: bool = False,
|
||
shadow_alpha: float = 0.9,
|
||
shadow_angle: float = -45.0,
|
||
shadow_color: str = "#000000",
|
||
shadow_distance: float = 5.0,
|
||
shadow_smoothing: float = 0.15,
|
||
# Bubble effect
|
||
bubble_effect_id: str | None = None,
|
||
bubble_resource_id: str | None = None,
|
||
# Text effect
|
||
effect_effect_id: str | None = None,
|
||
intro_animation: str | None = None,
|
||
intro_duration: float = 0.5,
|
||
outro_animation: str | None = None,
|
||
outro_duration: float = 0.5,
|
||
width: int = 1080,
|
||
height: int = 1920,
|
||
fixed_width: float = -1, # Text fixed width ratio, default -1 means not fixed
|
||
fixed_height: float = -1, # Text fixed height ratio, default -1 means not fixed
|
||
# 多样式文本参数
|
||
text_styles: Optional[List[TextStyleRange]] = None, # 文本的不同部分的样式列表
|
||
):
|
||
"""
|
||
Add text subtitle to the specified draft (configurable parameter version)
|
||
:param text: Text content
|
||
:param start: Start time (seconds)
|
||
:param end: End time (seconds)
|
||
:param draft_id: Draft ID (optional, default None creates a new draft)
|
||
:param transform_y: Y-axis position (default -0.8, bottom of screen)
|
||
:param transform_x: X-axis position (default 0, center of screen)
|
||
:param font: Font name (supports all fonts in Font_type)
|
||
:param font_color: Font color #FFF0FF
|
||
:param font_size: Font size (float value, default 8.0)
|
||
:param track_name: Track name
|
||
:param vertical: Whether to display vertically (default False)
|
||
:param font_alpha: Text transparency, range 0.0-1.0 (default 1.0, completely opaque)
|
||
:param border_alpha: Border transparency, range 0.0-1.0 (default 1.0)
|
||
:param border_color: Border color (default black)
|
||
:param border_width: Border width (default 0.0, no border display)
|
||
:param background_color: Background color (default black)
|
||
:param background_style: Background style (default 1)
|
||
:param background_alpha: Background transparency (default 0.0, no background display)
|
||
:param background_round_radius: 背景圆角半径，范围0.0-1.0（默认0.0）
|
||
:param background_height: 背景高度，范围0.0-1.0（默认0.14）
|
||
:param background_width: 背景宽度，范围0.0-1.0（默认0.14）
|
||
:param background_horizontal_offset: 背景水平偏移，范围0.0-1.0（默认0.5）
|
||
:param background_vertical_offset: 背景垂直偏移，范围0.0-1.0（默认0.5）
|
||
:param shadow_enabled: 是否启用阴影（默认False）
|
||
:param shadow_alpha: 阴影透明度，范围0.0-1.0（默认0.9）
|
||
:param shadow_angle: 阴影角度，范围-180.0-180.0（默认-45.0）
|
||
:param shadow_color: 阴影颜色（默认黑色）
|
||
:param shadow_distance: 阴影距离（默认5.0）
|
||
:param shadow_smoothing: 阴影平滑度，范围0.0-1.0（默认0.15）
|
||
:param bubble_effect_id: Bubble effect ID
|
||
:param bubble_resource_id: Bubble resource ID
|
||
:param effect_effect_id: Text effect ID
|
||
:param intro_animation: Intro animation type
|
||
:param intro_duration: Intro animation duration (seconds), default 0.5 seconds
|
||
:param outro_animation: Outro animation type
|
||
:param outro_duration: Outro animation duration (seconds), default 0.5 seconds
|
||
:param width: Video width (pixels)
|
||
:param height: Video height (pixels)
|
||
:param fixed_width: Text fixed width ratio, range 0.0-1.0, default -1 means not fixed
|
||
:param fixed_height: Text fixed height ratio, range 0.0-1.0, default -1 means not fixed
|
||
:param text_styles: 文本的不同部分的样式列表，每个元素是一个TextStyleRange
|
||
:return: Updated draft information
|
||
"""
|
||
# Validate if font is in Font_type
|
||
if font is None:
|
||
font_type = None
|
||
else:
|
||
try:
|
||
font_type = getattr(Font_type, font)
|
||
except:
|
||
available_fonts = [attr for attr in dir(Font_type) if not attr.startswith('_')]
|
||
raise ValueError(f"Unsupported font: {font}, please use one of the fonts in Font_type: {available_fonts}")
|
||
|
||
# Validate alpha value range
|
||
if not 0.0 <= font_alpha <= 1.0:
|
||
raise ValueError("alpha value must be between 0.0 and 1.0")
|
||
if not 0.0 <= border_alpha <= 1.0:
|
||
raise ValueError("border_alpha value must be between 0.0 and 1.0")
|
||
if not 0.0 <= background_alpha <= 1.0:
|
||
raise ValueError("background_alpha value must be between 0.0 and 1.0")
|
||
|
||
# Get or create draft
|
||
draft_id, script = get_or_create_draft(
|
||
draft_id=draft_id,
|
||
width=width,
|
||
height=height
|
||
)
|
||
|
||
# Add text track
|
||
if track_name is not None:
|
||
try:
|
||
imported_track = script.get_imported_track(draft.Track_type.text, name=track_name)
|
||
# If no exception is thrown, the track already exists
|
||
except exceptions.TrackNotFound:
|
||
# Track doesn't exist, create a new track
|
||
script.add_track(draft.Track_type.text, track_name=track_name)
|
||
else:
|
||
script.add_track(draft.Track_type.audio)
|
||
|
||
# Convert hexadecimal color to RGB tuple
|
||
try:
|
||
rgb_color = hex_to_rgb(font_color)
|
||
rgb_border_color = hex_to_rgb(border_color)
|
||
except ValueError as e:
|
||
raise ValueError(f"Color parameter error: {str(e)}")
|
||
|
||
# Create text_border
|
||
text_border = None
|
||
if border_width > 0:
|
||
text_border = draft.Text_border(
|
||
alpha=border_alpha,
|
||
color=rgb_border_color,
|
||
width=border_width
|
||
)
|
||
|
||
# Create text_background
|
||
text_background = None
|
||
if background_alpha > 0:
|
||
text_background = draft.Text_background(
|
||
color=background_color,
|
||
style=background_style,
|
||
alpha=background_alpha,
|
||
round_radius=background_round_radius,
|
||
height=background_height,
|
||
width=background_width,
|
||
horizontal_offset=background_horizontal_offset,
|
||
vertical_offset=background_vertical_offset
|
||
)
|
||
|
||
# 创建text_shadow (阴影)
|
||
text_shadow = None
|
||
if shadow_enabled:
|
||
text_shadow = draft.Text_shadow(
|
||
has_shadow=shadow_enabled,
|
||
alpha=shadow_alpha,
|
||
angle=shadow_angle,
|
||
color=shadow_color,
|
||
distance=shadow_distance,
|
||
smoothing=shadow_smoothing
|
||
)
|
||
|
||
# Create bubble effect
|
||
text_bubble = None
|
||
if bubble_effect_id and bubble_resource_id:
|
||
text_bubble = TextBubble(
|
||
effect_id=bubble_effect_id,
|
||
resource_id=bubble_resource_id
|
||
)
|
||
|
||
# Create text effect
|
||
text_effect = None
|
||
if effect_effect_id:
|
||
text_effect = TextEffect(
|
||
effect_id=effect_effect_id
|
||
)
|
||
|
||
# Convert ratio to pixel value
|
||
pixel_fixed_width = -1
|
||
pixel_fixed_height = -1
|
||
if fixed_width > 0:
|
||
pixel_fixed_width = int(fixed_width * script.width)
|
||
if fixed_height > 0:
|
||
pixel_fixed_height = int(fixed_height * script.height)
|
||
|
||
# Create text segment (using configurable parameters)
|
||
text_segment = draft.Text_segment(
|
||
text,
|
||
trange(f"{start}s", f"{end-start}s"),
|
||
font=font_type, # Use font from Font_type
|
||
style=draft.Text_style(
|
||
color=rgb_color,
|
||
size=font_size,
|
||
align=1,
|
||
vertical=vertical, # Set whether to display vertically
|
||
alpha=font_alpha # Set transparency
|
||
),
|
||
clip_settings=draft.Clip_settings(transform_y=transform_y, transform_x=transform_x),
|
||
border=text_border,
|
||
background=text_background,
|
||
shadow=text_shadow,
|
||
fixed_width=pixel_fixed_width,
|
||
fixed_height=pixel_fixed_height
|
||
)
|
||
|
||
# 应用多样式文本设置
|
||
if text_styles:
|
||
for style_range in text_styles:
|
||
# 验证范围有效性
|
||
if style_range.start < 0 or style_range.end > len(text) or style_range.start >= style_range.end:
|
||
raise ValueError(f"无效的文本范围: [{style_range.start}, {style_range.end}), 文本长度: {len(text)}")
|
||
|
||
# 应用样式到特定文本范围
|
||
text_segment.add_text_style(style_range)
|
||
|
||
if text_bubble:
|
||
text_segment.add_bubble(text_bubble.effect_id, text_bubble.resource_id)
|
||
if text_effect:
|
||
text_segment.add_effect(text_effect.effect_id)
|
||
|
||
# Add intro animation
|
||
if intro_animation:
|
||
try:
|
||
if IS_CAPCUT_ENV:
|
||
animation_type = getattr(draft.CapCut_Text_intro, intro_animation)
|
||
else:
|
||
animation_type = getattr(draft.Text_intro, intro_animation)
|
||
# Convert seconds to microseconds
|
||
duration_microseconds = int(intro_duration * 1000000)
|
||
text_segment.add_animation(animation_type, duration_microseconds) # Add intro animation, set duration
|
||
except:
|
||
print(f"Warning: Unsupported intro animation type {intro_animation}, this parameter will be ignored")
|
||
|
||
# Add outro animation
|
||
if outro_animation:
|
||
try:
|
||
if IS_CAPCUT_ENV:
|
||
animation_type = getattr(draft.CapCut_Text_outro, outro_animation)
|
||
else:
|
||
animation_type = getattr(draft.Text_outro, outro_animation)
|
||
# Convert seconds to microseconds
|
||
duration_microseconds = int(outro_duration * 1000000)
|
||
text_segment.add_animation(animation_type, duration_microseconds) # Add outro animation, set duration
|
||
except:
|
||
print(f"Warning: Unsupported outro animation type {outro_animation}, this parameter will be ignored")
|
||
|
||
# Add text segment to track
|
||
script.add_segment(text_segment, track_name=track_name)
|
||
|
||
return {
|
||
"draft_id": draft_id,
|
||
"draft_url": generate_draft_url(draft_id)
|
||
}
|