Update app.py

app.py

@@ -7,18 +7,18 @@ import numpy as np
 import os
 import time
 
-
 # 设置umask确保新创建的文件有正确的权限
 os.umask(0o022)
 
 
-def subliminal_insert(video_path,
-                      frame_img_path,
-                      interval_frames,
-                      duration_frames,
-                      output_dir,
-                      alpha=0.5,  # 添加透明度参数
-                      progress=None):
+def subliminal_insert(
+        video_path,
+        frame_img_path,
+        interval_frames,
+        duration_frames,
+        output_dir,
+        alpha=0.5,  # 添加透明度参数
+        progress=None):
     """
     将特定帧按间隔植入视频流中
     """
@@ -73,9 +73,9 @@ def subliminal_insert(video_path,
             v = ffmpeg.filter(
                 [input_video, input_image],
                 'overlay',
-                enable=f'if(lt(mod(n,{interval_frames}),{duration_frames}),1,0)',
-                alpha=alpha
-            )
+                enable=
+                f'if(lt(mod(n,{interval_frames}),{duration_frames}),1,0)',
+                alpha=alpha)
 
             # 检查输入视频是否有音频流
             has_audio = any(s['codec_type'] == 'audio'
@@ -158,12 +158,13 @@ def subliminal_insert(video_path,
                 pass
 
 
-def process_video(video_input,
-                  frame_input,
-                  interval,
-                  duration,
-                  alpha=0.5,  # 添加透明度参数
-                  progress=gr.Progress()):
+def process_video(
+    video_input,
+    frame_input,
+    interval,
+    duration,
+    alpha=0.5,  # 添加透明度参数
+    progress=gr.Progress()):
     """
     处理视频和图片输入
     """
@@ -215,7 +216,8 @@ def process_video(video_input,
             raise Exception(f"图片处理失败: {str(e)}")
 
         output_path = subliminal_insert(video_temp, frame_temp, int(interval),
-                                        int(duration), output_dir, alpha, progress)
+                                        int(duration), output_dir, alpha,
+                                        progress)
 
         print(f"处理完成，输出文件: {output_path}")
         return output_path