Update app.py

app.py

@@ -11,12 +11,9 @@ from roop.core import (
 )
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
-import os
 from PIL import Image
 
-
-def swap_face(source_file, target_file,doFaceEnhancer):
-
+def swap_face(source_file, target_file, do_face_enhancer):
     source_path = "input.jpg"
     target_path = "target.jpg"
 
@@ -34,10 +31,12 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     roop.globals.output_path = normalize_output_path(
         roop.globals.source_path, roop.globals.target_path, output_path
     )
-    if doFaceEnhancer == True:
-        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
+
+    if do_face_enhancer:
+        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
@@ -65,8 +64,16 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     start()
     return output_path
 
-
 app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
+    fn=swap_face, 
+    inputs=[
+        gr.Image(), 
+        gr.Image(),
+        gr.Checkbox(label="Enable Face Enhancer", default=False, key="do_face_enhancer", info="Toggle face enhancer on/off")
+    ],
+    outputs="image"
 )
 app.launch()
+
+
+