Update app.py

app.py

@@ -6,28 +6,22 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
     if video is None or audio is None or checkpoint is None:
         return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
 
-    # Отладка: проверка текущей директории и наличия inference.py
     print(f"Текущая рабочая директория: {os.getcwd()}")
     print(f"Содержимое текущей директории: {os.listdir('.')}")
     print(f"Проверка наличия 'inference.py': {os.path.exists('inference.py')}")
 
-    # Используем пути к файлам напрямую
-    video_path = video  # Строка с путем к видео
-    audio_path = audio  # Строка с путем к аудио
+    video_path = video  
+    audio_path = audio  
 
-    # Логирование путей для отладки
     print(f"Путь к видео: {video_path}")
     print(f"Путь к аудио: {audio_path}")
 
-    # Создание директории 'outputs' для сохранения выходного видео
     output_dir = "outputs"
     os.makedirs(output_dir, exist_ok=True)
 
-    # Определение выходного файла
     output_path = os.path.join(output_dir, "output.mp4")
     print(f"Путь к выходному файлу: {output_path}")
 
-    # Подготовка аргументов для инференса
     args = [
     "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
     "--segmentation_path", "checkpoints/face_segmentation.pth",    
@@ -51,7 +45,6 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
         args.append("--save_as_video")
 
     try:
-        # Вызов команды инференса без изменения рабочей директории
         cmd = ["python", "inference.py"] + args
         print(f"Запуск инференса с командой: {' '.join(cmd)}")
         subprocess.run(cmd, check=True)
@@ -59,16 +52,13 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
         print(f"Ошибка при выполнении команды: {e}")
         return f"Произошла ошибка при обработке: {e}"
 
-    # Проверка наличия выходного файла
     if not os.path.exists(output_path):
         print("Выходной файл не существует.")
         return "Не удалось создать выходное видео."
 
-    # Возвращаем путь к выходному файлу
     print(f"Выходной файл создан по пути: {output_path}")
-    return output_path  # Gradio автоматически обработает путь и выведет видео
+    return output_path  
 
-# Корректная структура Gradio Blocks
 with gr.Blocks() as ui:
     gr.Markdown("## Wav2Lip - Синхронизация губ в видео")
     with gr.Row():
@@ -84,18 +74,16 @@ with gr.Blocks() as ui:
             pad_bottom = gr.Slider(minimum=0, maximum=50, step=1, value=10, label="Отступ снизу")
             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)  # Новый элемент управления
+            save_as_video = gr.Checkbox(label="Сохранять как видео", value=True)  
             generate_btn = gr.Button("Сгенерировать")
         with gr.Column():
             result = gr.Video(label="Результат")
 
-    # Вызов метода .click() внутри контекста gr.Blocks()
     generate_btn.click(
         generate, 
         inputs=[video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right, save_as_video], 
         outputs=result,
-        concurrency_limit=10  # Устанавливаем лимит на количество одновременно выполняемых операций
+        concurrency_limit=30  
     )
 
-# Запуск приложения
 ui.launch(debug=True)