Update app.py

app.py

@@ -3,7 +3,7 @@ import subprocess
 import os
 import tempfile
 
-def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right):
+def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right, save_as_video):
     if video is None or audio is None or checkpoint is None:
         return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
 
@@ -40,6 +40,9 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
 
         if no_smooth:
             args.append("--nosmooth")
+        
+        if save_as_video:
+            args.append("--save_as_video")
 
         try:
             # Вызов команды инференса без изменения рабочей директории
@@ -75,6 +78,7 @@ with gr.Blocks() as ui:
             pad_bottom = gr.Slider(minimum=0, maximum=50, step=1, value=10, label="Отступ снизу (рекомендуется 20 для включения подбородка)")
             pad_left = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ слева")
             pad_right = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ справа")
+            save_as_video = gr.Checkbox(label="Сохранять как видео", value=True)  # Новый элемент управления
             generate_btn = gr.Button("Сгенерировать")
         with gr.Column():
             result = gr.Video(label="Результат")
@@ -82,7 +86,7 @@ with gr.Blocks() as ui:
     # Вызов метода .click() внутри контекста gr.Blocks()
     generate_btn.click(
         generate, 
-        inputs=[video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right], 
+        inputs=[video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right, save_as_video], 
         outputs=result,
         concurrency_limit=1  # Устанавливаем лимит на количество одновременно выполняемых операций
     )