Update app.py

app.py

@@ -17,6 +17,7 @@ def subliminal_insert(video_path,
                       interval_frames,
                       duration_frames,
                       output_dir,
+                      alpha=0.5,  # 添加透明度参数
                       progress=None):
     """
     将特定帧按间隔植入视频流中
@@ -72,7 +73,8 @@ 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)'
+                enable=f'if(lt(mod(n,{interval_frames}),{duration_frames}),1,0)',
+                alpha=alpha
             )
 
             # 检查输入视频是否有音频流
@@ -160,6 +162,7 @@ def process_video(video_input,
                   frame_input,
                   interval,
                   duration,
+                  alpha=0.5,  # 添加透明度参数
                   progress=gr.Progress()):
     """
     处理视频和图片输入
@@ -212,7 +215,7 @@ def process_video(video_input,
             raise Exception(f"图片处理失败: {str(e)}")
 
         output_path = subliminal_insert(video_temp, frame_temp, int(interval),
-                                        int(duration), output_dir, progress)
+                                        int(duration), output_dir, alpha, progress)
 
         print(f"处理完成，输出文件: {output_path}")
         return output_path
@@ -247,7 +250,13 @@ iface = gr.Interface(fn=process_video,
                                    value=1,
                                    step=1,
                                    label="植入持续时间(帧数)",
-                                   info="每次插入的图片持续多少帧")
+                                   info="每次插入的图片持续多少帧"),
+                         gr.Slider(minimum=0.0,
+                                   maximum=1.0,
+                                   value=0.5,
+                                   step=0.1,
+                                   label="叠加透明度",
+                                   info="图片叠加的透明度(0完全透明,1完全不透明)")
                      ],
                      outputs=gr.Video(label="输出视频"),
                      title="视频流插入帧",