Update app.py

app.py

@@ -22,6 +22,9 @@ def process_video_with_mask(video_path, mask_path, text="Joker"):
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
     
+    # 初始化固定角度为None
+    fixed_angle = None
+    
     while True:
         ret_video, frame_video = cap_video.read()
         ret_mask, frame_mask = cap_mask.read()
@@ -55,7 +58,11 @@ def process_video_with_mask(video_path, mask_path, text="Joker"):
             # 计算中心点和尺寸
             center = rect[0]
             size = rect[1]
-            angle = rect[2]
+            current_angle = rect[2]
+            
+            # 如果是第一帧，记录角度
+            if fixed_angle is None:
+                fixed_angle = current_angle
             
             # 创建透明画布绘制文字
             text_image = Image.new("RGBA", (int(size[0]), int(size[1])), (0, 0, 0, 0))
@@ -78,8 +85,8 @@ def process_video_with_mask(video_path, mask_path, text="Joker"):
             # 绘制文字
             text_draw.text((pos_x, pos_y), text, fill=(255, 255, 0, 255), font=font)
             
-            # 旋转文字
-            rotated_text = text_image.rotate(angle, expand=True, center=(size[0]/2, size[1]/2))
+            # 使用固定角度旋转文字
+            rotated_text = text_image.rotate(fixed_angle, expand=True, center=(size[0]/2, size[1]/2))
             
             # 计算粘贴位置
             paste_x = int(center[0] - rotated_text.width / 2)