Update app.py

app.py

@@ -10,20 +10,15 @@ 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 "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
 
-    # Создание временной директории для обработки
-    with tempfile.TemporaryDirectory() as temp_dir:
-        # Копирование загруженных файлов во временную директорию
-        video_path = os.path.join(temp_dir, video.name)
-        audio_path = os.path.join(temp_dir, audio.name)
-        with open(video_path, "wb") as f:
-            f.write(video.read())
-        with open(audio_path, "wb") as f:
-            f.write(audio.read())
+    # Поскольку Gradio теперь возвращает пути к файлам, мы используем их напрямую без необходимости копировать их содержимое
+    video_path = video  # Уже строка с путем к видео
+    audio_path = audio  # Уже строка с путем к аудио
 
-        # Определение выходного файла
+    # Создание временной директории для сохранения выходного видео
+    with tempfile.TemporaryDirectory() as temp_dir:
         output_path = os.path.join(temp_dir, "output.mp4")
 
-        # Подготовка аргументов для inference.py
+        # Подготовка аргументов для инференса
         args = [
             "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
             "--segmentation_path", "checkpoints/face_segmentation.pth",
@@ -39,11 +34,7 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
             args.append("--nosmooth")
 
         try:
-            # Вызов функции инференса напрямую
-            # Если вы интегрируете код из inference.py, вызовите функцию непосредственно
-            # Например:
-            # inference_main(args)
-            # Но для простоты, здесь мы будем использовать subprocess
+            # Вызов функции инференса через subprocess
             cmd = ["python", "inference.py"] + args
             subprocess.run(cmd, check=True)
         except subprocess.CalledProcessError as e:
@@ -53,11 +44,8 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
         if not os.path.exists(output_path):
             return "Не удалось создать выходное видео."
 
-        # Чтение выходного файла
-        with open(output_path, "rb") as f:
-            output_video = f.read()
-
-        return output_video
+        # Возвращаем путь к выходному файлу
+        return output_path 
 
 with gr.Blocks() as ui:
     gr.Markdown("## Wav2Lip - Синхронизация губ в видео")