Spaces:
Running
Running
| import cv2 | |
| import numpy as np | |
| import os | |
| from PIL import ImageFont | |
| def parse_color(color_str, default=(255, 255, 255)): | |
| color_map = { | |
| "gold": (255, 215, 0), "red": (255, 60, 60), "blue": (60, 120, 255), | |
| "white": (255, 255, 255), "black": (0, 0, 0), "cyan": (0, 255, 255), | |
| "magenta": (255, 0, 255), "yellow": (255, 255, 0), "orange": (255, 165, 0), | |
| "pink": (255, 105, 180), "lime": (50, 255, 50), | |
| } | |
| if color_str.lower() in color_map: | |
| return color_map[color_str.lower()] | |
| if color_str.startswith("#") and len(color_str) == 7: | |
| try: | |
| return (int(color_str[1:3], 16), int(color_str[3:5], 16), int(color_str[5:7], 16)) | |
| except: | |
| pass | |
| return default | |
| def get_chinese_font(): | |
| fonts = [ | |
| "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc", | |
| "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc", | |
| "/usr/share/fonts/noto-cjk/NotoSansCJK-Regular.ttc", | |
| "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc", | |
| "/usr/share/fonts/truetype/wqy/wqy-microhei.ttc", | |
| "NotoSansCJK-Regular.ttc", | |
| ] | |
| for f in fonts: | |
| if os.path.exists(f): | |
| return f | |
| return None | |
| def get_title_font(font_path, size): | |
| try: | |
| return ImageFont.truetype(font_path, size) if font_path else ImageFont.load_default() | |
| except: | |
| return ImageFont.load_default() | |
| def fit_image_to_canvas(img, target_w, target_h): | |
| h, w = img.shape[:2] | |
| scale = min(target_w / w, target_h / h) | |
| new_w, new_h = int(w * scale), int(h * scale) | |
| resized = cv2.resize(img, (new_w, new_h)) | |
| canvas = np.zeros((target_h, target_w, 3), dtype=np.uint8) | |
| x_offset, y_offset = (target_w - new_w) // 2, (target_h - new_h) // 2 | |
| canvas[y_offset:y_offset + new_h, x_offset:x_offset + new_w] = resized | |
| return canvas | |
| def ken_burns_crop(img, progress, w_out, h_out): | |
| h, w = img.shape[:2] | |
| zoom = 1.0 + 0.15 * progress | |
| crop_w, crop_h = w_out / zoom, h_out / zoom | |
| dx = max(0, min((w - crop_w) * progress, w - crop_w)) if w > crop_w else 0 | |
| dy = max(0, min((h - crop_h) * progress, h - crop_h)) if h > crop_h else 0 | |
| crop = img[int(dy):int(dy + crop_h), int(dx):int(dx + crop_w)] | |
| return cv2.resize(crop, (w_out, h_out)) | |
| def apply_cinemascope(frame, bar_height_ratio=0.08): | |
| h = frame.shape[0] | |
| bar_h = int(h * bar_height_ratio) | |
| frame[:bar_h, :] = 0 | |
| frame[h - bar_h:, :] = 0 | |
| return frame |