Update app.py

app.py

@@ -467,7 +467,57 @@ def generate_video(
 
     export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
 
-    return video_path, current_seed
+    if check_ffmpeg():
+        try:
+            # Создаем временный файл для видео с звуком
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as audio_tmpfile:
+                video_with_audio_path = audio_tmpfile.name
+            
+            # Команда ffmpeg для добавления тихого аудио
+            cmd = [
+                'ffmpeg',
+                '-f', 'lavfi', 
+                '-i', 'anullsrc=channel_layout=stereo:sample_rate=44100',
+                '-i', video_path,
+                '-c:v', 'copy',
+                '-c:a', 'aac',
+                '-shortest',
+                '-y',
+                video_with_audio_path
+            ]
+            
+            # Запускаем ffmpeg
+            subprocess.run(cmd, capture_output=True, check=True)
+
+            # Создаем временный файл для заблюренного видео
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as blur_tmpfile:
+                blurred_video_path = blur_tmpfile.name
+            
+            # Команда ffmpeg для создания гауссова размытия
+            cmd_blur = [
+                'ffmpeg',
+                '-i', video_with_audio_path,  # Используем видео с аудио как источник
+                '-vf', 'gblur=sigma=25',       # Гауссово размытие с sigma=5
+                '-c:a', 'copy',               # Копируем аудио без изменений
+                '-y',
+                blurred_video_path
+            ]
+            
+            # Запускаем ffmpeg для создания блюра
+            subprocess.run(cmd_blur, capture_output=True, check=True)
+            
+            # Удаляем исходный файл без звука
+            os.unlink(video_path)
+            
+            return video_with_audio_path, blurred_video_path, current_seed
+            
+        except Exception as e:
+            print(f"Error adding audio: {e}")
+            # В случае ошибки возвращаем видео без звука
+            return video_path, video_path, current_seed
+    else:
+        print("FFmpeg not available, returning video without audio")
+        return video_path, video_path, current_seed
 
 with gr.Blocks() as demo:
     gr.Markdown("# Wan22 AOT")
@@ -488,7 +538,8 @@ with gr.Blocks() as demo:
 
             generate_button = gr.Button("Generate Video", variant="primary")
         with gr.Column():
-            video_output = gr.Video(label="Generated Video", autoplay=True, interactive=False)
+            video_output_1 = gr.Video(label="Generated Video", autoplay=True, interactive=False)
+            video_output_2 = gr.Video(label="Generated Video", autoplay=True, interactive=False)
   
     #upload_image_and_prompt(input_image_component, prompt_input)
     ui_inputs = [
@@ -497,7 +548,14 @@ with gr.Blocks() as demo:
         guidance_scale_input, guidance_scale_2_input, seed_input, randomize_seed_checkbox
     ]
    
-    generate_button.click(fn=generate_video_with_upload, inputs=ui_inputs, outputs=[video_output, seed_input])
+    generate_button.click(fn=generate_video_with_upload, inputs=ui_inputs, outputs=[video_output_1, video_output_2, seed_input])
+
+def check_ffmpeg():
+    try:
+        subprocess.run(['ffmpeg', '-version'], capture_output=True, check=True)
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return False
 
 if __name__ == "__main__":
     demo.queue().launch(mcp_server=True)