Update app.py

app.py

@@ -1,6 +1,6 @@
-# -* coding:UTF-8 -*
-# !/usr/bin/env python
-import numpy as np
+import os
+from PIL import Image
+import gradio as gr
 import roop.globals
 from roop.core import (
     start,
@@ -10,26 +10,17 @@ from roop.core import (
 )
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
-from PIL import Image
-import gradio as gr
-import os
+from roop.globals import clean_temp
 
 def swap_face(source_file, target_file):
     source_image = Image.fromarray(source_file)
     source_path = "input.jpg"
     source_image.save(source_path, format="JPEG", quality=95)
 
-    
     output_path = "output.mp4"
 
-    roop.globals.source_path = source_path
-    roop.globals.target_path = target_file
-    
-    
+    roop.globals.set_global_paths(source_path, target_file, output_path)
 
-    roop.globals.output_path = normalize_output_path(
-        roop.globals.source_path, roop.globals.target_path, output_path
-    )
     roop.globals.frame_processors = ["face_swapper"]
     roop.globals.headless = True
     roop.globals.keep_fps = True
@@ -38,21 +29,26 @@ def swap_face(source_file, target_file):
     roop.globals.many_faces = False
     roop.globals.video_encoder = "libx264"
     roop.globals.video_quality = 50
-    roop.globals.max_memory = 14#suggest_max_memory()
+    roop.globals.max_memory = 14  # suggest_max_memory()
     roop.globals.execution_providers = decode_execution_providers(["cpu"])
-    roop.globals.execution_threads = 32#suggest_execution_threads()
-    for frame_processor in get_frame_processors_modules(
-        roop.globals.frame_processors
-    ):
+    roop.globals.execution_threads = 32  # suggest_execution_threads()
+
+    for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
         if not frame_processor.pre_check():
             return
 
     start()
 
+    clean_temp(roop.globals.target_path)
+
     return os.path.join(os.getcwd(), output_path)
 
 app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Video()], outputs=[gr.Video()], description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",theme="NoCrypt/miku"
+    fn=swap_face,
+    inputs=[gr.Image(), gr.Video()],
+    outputs=[gr.Video()],
+    description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
+    theme="NoCrypt/miku"
 )
 
-app.launch()
+app.launch()