Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import subprocess
 import os
-import tempfile
 
 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:
@@ -20,47 +19,49 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
     print(f"Путь к видео: {video_path}")
     print(f"Путь к аудио: {audio_path}")
 
-    # Создание временной директории для сохранения выходного видео
-    with tempfile.TemporaryDirectory() as temp_dir:
-        # Определение выходного файла
-        output_path = os.path.join(temp_dir, "output.mp4")
-        print(f"Путь к выходному файлу: {output_path}")
+    # Создание директории 'outputs' для сохранения выходного видео
+    output_dir = "outputs"
+    os.makedirs(output_dir, exist_ok=True)
 
-        # Подготовка аргументов для инференса
-        args = [
-            "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
-            "--segmentation_path", "checkpoints/face_segmentation.pth",
-            "--enhance_face", "gfpgan",
-            "--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)
-        ]
+    # Определение выходного файла
+    output_path = os.path.join(output_dir, "output.mp4")
+    print(f"Путь к выходному файлу: {output_path}")
 
-        if no_smooth:
-            args.append("--nosmooth")
-        
-        if save_as_video:
-            args.append("--save_as_video")
+    # Подготовка аргументов для инференса
+    args = [
+        "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
+        "--segmentation_path", "checkpoints/face_segmentation.pth",
+        "--enhance_face", "gfpgan",
+        "--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)
+    ]
 
-        try:
-            # Вызов команды инференса без изменения рабочей директории
-            cmd = ["python", "inference.py"] + args
-            print(f"Запуск инференса с командой: {' '.join(cmd)}")
-            subprocess.run(cmd, check=True)
-        except subprocess.CalledProcessError as e:
-            print(f"Ошибка при выполнении команды: {e}")
-            return f"Произошла ошибка при обработке: {e}"
+    if no_smooth:
+        args.append("--nosmooth")
 
-        # Проверка наличия выходного файла
-        if not os.path.exists(output_path):
-            print("Выходной файл не существует.")
-            return "Не удалось создать выходное видео."
+    if save_as_video:
+        args.append("--save_as_video")
 
-        # Возвращаем путь к выходному файлу
-        print(f"Выходной файл создан по пути: {output_path}")
-        return output_path  # Gradio автоматически обработает путь и выведет видео
+    try:
+        # Вызов команды инференса без изменения рабочей директории
+        cmd = ["python", "inference.py"] + args
+        print(f"Запуск инференса с командой: {' '.join(cmd)}")
+        subprocess.run(cmd, check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Ошибка при выполнении команды: {e}")
+        return f"Произошла ошибка при обработке: {e}"
+
+    # Проверка наличия выходного файла
+    if not os.path.exists(output_path):
+        print("Выходной файл не существует.")
+        return "Не удалось создать выходное видео."
+
+    # Возвращаем путь к выходному файлу
+    print(f"Выходной файл создан по пути: {output_path}")
+    return output_path  # Gradio автоматически обработает путь и выведет видео
 
 # Корректная структура Gradio Blocks
 with gr.Blocks() as ui: