165 lines
4.8 KiB
Python
165 lines
4.8 KiB
Python
import os
|
|
|
|
from moviepy.video.io.VideoFileClip import VideoFileClip
|
|
from moviepy.video.VideoClip import ColorClip
|
|
from moviepy.video.compositing.CompositeVideoClip import CompositeVideoClip
|
|
from moviepy import TextClip
|
|
|
|
font = "./Montserrat.ttf"
|
|
font_size = 70
|
|
video_codec = "libx264"
|
|
|
|
def auto_wrap_text(text, max_width):
|
|
if not text:
|
|
return ""
|
|
|
|
words = text.split()
|
|
lines = []
|
|
|
|
line = ""
|
|
for word in words:
|
|
test_line = f"{line} {word}".strip()
|
|
test_clip = TextClip(text=test_line, font=font, font_size=font_size, color='white', method='label')
|
|
|
|
if test_clip.w > max_width and line != "":
|
|
lines.append(line)
|
|
line = word
|
|
else:
|
|
line = test_line
|
|
|
|
test_clip.close()
|
|
|
|
lines.append(line)
|
|
|
|
return "\n".join(lines)
|
|
|
|
def cut_video_new_clip(input_path: str, start: float, end: float, output_path: str):
|
|
|
|
|
|
with VideoFileClip(input_path) as clip:
|
|
segment = clip.subclipped(start, end)
|
|
fps = clip.fps or 30
|
|
|
|
segment.write_videofile(
|
|
output_path,
|
|
codec=video_codec,
|
|
remove_temp=True,
|
|
fps=fps,
|
|
bitrate="2000k",
|
|
ffmpeg_params=[
|
|
"-preset", "ultrafast",
|
|
"-tune", "zerolatency",
|
|
"-pix_fmt", "yuv420p",
|
|
"-profile:v", "high",
|
|
"-level", "4.1"
|
|
]
|
|
)
|
|
|
|
def process_segment(input_path: str, top_text: str = "", bottom_text: str = "", filename="", idx=1) -> str:
|
|
os.makedirs("outputs", exist_ok=True)
|
|
os.makedirs(f"outputs/{filename}", exist_ok=True)
|
|
|
|
final_width, final_height = 1080, 1920
|
|
top_h, middle_h, bottom_h = 480, 960, 480
|
|
|
|
with VideoFileClip(input_path) as clip:
|
|
dur = clip.duration
|
|
bg = ColorClip(size=(final_width, final_height), color=(0, 0, 0), duration=dur)
|
|
video_resized = clip.resized(width=final_width)
|
|
y = top_h + (middle_h - video_resized.h) // 2
|
|
video_resized = video_resized.with_position((0, y))
|
|
wrapped_top_text = auto_wrap_text(top_text, final_width - 40)
|
|
wrapped_bottom_text = auto_wrap_text(bottom_text, final_width - 40)
|
|
|
|
txt_top = TextClip(
|
|
text=wrapped_top_text,
|
|
font_size=70,
|
|
color="white",
|
|
font=font,
|
|
method="label",
|
|
size=(final_width, top_h)
|
|
).with_duration(dur).with_position((0, 0))
|
|
|
|
txt_bot = TextClip(
|
|
text=wrapped_bottom_text,
|
|
font_size=70,
|
|
color="white",
|
|
font=font,
|
|
method="label",
|
|
size=(final_width, bottom_h),
|
|
).with_duration(dur).with_position((0, final_height - bottom_h))
|
|
|
|
final = CompositeVideoClip([bg, video_resized, txt_top, txt_bot], size=(final_width, final_height))
|
|
|
|
output_path = f"outputs/{filename}/clip_{idx}.mp4"
|
|
|
|
final.write_videofile(
|
|
output_path,
|
|
codec=video_codec,
|
|
remove_temp=True,
|
|
fps=30,
|
|
bitrate="2000k",
|
|
ffmpeg_params=[
|
|
"-preset", "ultrafast",
|
|
"-tune", "zerolatency",
|
|
"-pix_fmt", "yuv420p",
|
|
"-profile:v", "high",
|
|
"-level", "4.1"
|
|
]
|
|
)
|
|
|
|
final.close()
|
|
|
|
return output_path
|
|
|
|
def timestamp_to_seconds(ts):
|
|
if isinstance(ts, (int, float)):
|
|
return ts
|
|
|
|
parts = ts.split(":")
|
|
parts = [float(p) for p in parts]
|
|
|
|
if len(parts) == 3:
|
|
h, m, s = parts
|
|
return h * 3600 + m * 60 + s
|
|
elif len(parts) == 2:
|
|
m, s = parts
|
|
return m * 60 + s
|
|
elif len(parts) == 1:
|
|
return parts[0]
|
|
else:
|
|
raise ValueError(f"Timestamp inválido: {ts}")
|
|
|
|
def process_full_video(filename: str, times: list = None) -> list:
|
|
os.makedirs("temp", exist_ok=True)
|
|
|
|
times = times or []
|
|
video_path = f"videos/{filename}"
|
|
processed = []
|
|
|
|
print(f"Total de trechos: {len(times)}")
|
|
print(f"Codec de render: {video_codec}")
|
|
|
|
for idx, interval in enumerate(times, start=1):
|
|
start = timestamp_to_seconds(interval.get("start", 0))
|
|
end_raw = interval.get("end", None)
|
|
end = timestamp_to_seconds(end_raw) if end_raw is not None else None
|
|
top_text = interval.get("topText", "")
|
|
bottom_text = interval.get("bottomText", "")
|
|
|
|
if end is None:
|
|
with VideoFileClip(video_path) as clip:
|
|
end = clip.duration
|
|
|
|
print(f"Cortando trecho {idx}: {start}s a {end}s")
|
|
|
|
temp_path = f"temp/{os.path.splitext(filename)[0]}_{idx}.mp4"
|
|
|
|
cut_video_new_clip(video_path, start, end, temp_path)
|
|
|
|
out = process_segment(temp_path, top_text, bottom_text, filename, idx)
|
|
|
|
processed.append(out)
|
|
|
|
return processed
|