Update app.py

app.py

@@ -10,22 +10,26 @@ 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 "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
 
-    # Лог для видео и аудио путей
-    print(f"Video path: {video}, Audio path: {audio}")
+    # Поскольку Gradio возвращает пути к файлам, используем их напрямую
+    video_path = video  # Уже строка с путем к видео
+    audio_path = audio  # Уже строка с путем к аудио
 
+    # Создание временной директории для сохранения выходного видео
     with tempfile.TemporaryDirectory() as temp_dir:
-        output_path = os.path.join(temp_dir, "output.mp4")
+        # Создание поддиректории 'temp' внутри temp_dir для inference.py
+        temp_subdir = os.path.join(temp_dir, 'temp')
+        os.makedirs(temp_subdir, exist_ok=True)
 
-        # Лог для вывода пути к файлу
-        print(f"Output path: {output_path}")
+        # Определение выходного файла
+        output_path = os.path.join(temp_dir, "output.mp4")
 
         # Подготовка аргументов для инференса
         args = [
             "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
             "--segmentation_path", "checkpoints/face_segmentation.pth",
             "--enhance_face", "gfpgan",
-            "--face", video,
-            "--audio", audio,
+            "--face", video_path,
+            "--audio", audio_path,
             "--outfile", output_path,
             "--resize_factor", str(resize_factor),
             "--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
@@ -35,19 +39,18 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
             args.append("--nosmooth")
 
         try:
-            # Лог для команды инференса
-            print(f"Running inference with command: python inference.py {' '.join(args)}")
-
+            # Вызов команды с установкой рабочей директории в temp_dir
             cmd = ["python", "inference.py"] + args
-            subprocess.run(cmd, check=True)
+            subprocess.run(cmd, check=True, cwd=temp_dir)
         except subprocess.CalledProcessError as e:
             return f"Произошла ошибка при обработке: {e}"
 
+        # Проверка наличия выходного файла
         if not os.path.exists(output_path):
             return "Не удалось создать выходное видео."
 
-        return output_path
-
+        # Возвращаем путь к выходному файлу
+        return output_path  # Gradio автоматически обработает путь и выведет видео
 
 with gr.Blocks() as ui:
     gr.Markdown("## Wav2Lip - Синхронизация губ в видео")
@@ -72,7 +75,7 @@ with gr.Blocks() as ui:
         generate, 
         inputs=[video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right], 
         outputs=result,
-        concurrency_limit=1  
+        concurrency_limit=1  # Устанавливаем лимит на количество одновременно выполняемых операций
     )
 
 ui.launch(debug=True)