Update app.py

app.py

@@ -15,7 +15,6 @@ import spaces
 
 @spaces.GPU
 def swap_face(source_file, target_file, doFaceEnhancer):
-
     source_path = "input.jpg"
     target_path = "target.jpg"
 
@@ -27,50 +26,52 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     print("source_path: ", source_path)
     print("target_path: ", target_path)
 
-    roop.globals.source_path = source_path
-    roop.globals.target_path = target_path
     output_path = "output.jpg"
-    roop.globals.output_path = normalize_output_path(
-        roop.globals.source_path, roop.globals.target_path, output_path
-    )
-    if doFaceEnhancer:
-        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
-    else:
-        roop.globals.frame_processors = ["face_swapper"]
-    
-    roop.globals.headless = True
-    roop.globals.keep_fps = True
-    roop.globals.keep_audio = True
-    roop.globals.keep_frames = False
-    roop.globals.many_faces = False
-    roop.globals.video_encoder = "libx264"
-    roop.globals.video_quality = 18
-    roop.globals.max_memory = "4G"  # Sử dụng giá trị mặc định
-    roop.globals.execution_providers = decode_execution_providers(["cuda"])
-    roop.globals.execution_threads = 4  # Sử dụng giá trị mặc định
+    normalized_output_path = normalize_output_path(source_path, target_path, output_path)
 
-    # Đảm bảo reference_face_position không phải là None
-    if roop.globals.reference_face_position is None:
-        roop.globals.reference_face_position = 0  # Hoặc một giá trị mặc định khác
-
-    # Đảm bảo similar_face_distance không phải là None
-    if roop.globals.similar_face_distance is None:
-        roop.globals.similar_face_distance = 0.6  # Hoặc một giá trị mặc định khác
+    frame_processors = ["face_swapper", "face_enhancer"] if doFaceEnhancer else ["face_swapper"]
+    headless = True
+    keep_fps = True
+    keep_audio = True
+    keep_frames = False
+    many_faces = False
+    video_encoder = "libx264"
+    video_quality = 18
+    max_memory = "4G"
+    execution_providers = decode_execution_providers(["cuda"])
+    execution_threads = 4
+    reference_face_position = 0
+    similar_face_distance = 0.6
 
     print(
         "start process",
-        roop.globals.source_path,
-        roop.globals.target_path,
-        roop.globals.output_path,
+        source_path,
+        target_path,
+        normalized_output_path,
     )
 
-    for frame_processor in get_frame_processors_modules(
-        roop.globals.frame_processors
-    ):
+    for frame_processor in get_frame_processors_modules(frame_processors):
         if not frame_processor.pre_check():
             print(f"Pre-check failed for {frame_processor}")
             return
 
+    roop.globals.source_path = source_path
+    roop.globals.target_path = target_path
+    roop.globals.output_path = normalized_output_path
+    roop.globals.frame_processors = frame_processors
+    roop.globals.headless = headless
+    roop.globals.keep_fps = keep_fps
+    roop.globals.keep_audio = keep_audio
+    roop.globals.keep_frames = keep_frames
+    roop.globals.many_faces = many_faces
+    roop.globals.video_encoder = video_encoder
+    roop.globals.video_quality = video_quality
+    roop.globals.max_memory = max_memory
+    roop.globals.execution_providers = execution_providers
+    roop.globals.execution_threads = execution_threads
+    roop.globals.reference_face_position = reference_face_position
+    roop.globals.similar_face_distance = similar_face_distance
+
     start()
     return output_path