Update app.py

app.py

@@ -10,7 +10,7 @@ os.makedirs('temp', exist_ok=True)
 
 
 
-def run_scripts(target, source, mode, use_face_enhancer):
+def run_scripts(target, source, mode):
     if target is None or (source is None):
         return None
     with st.spinner("Processing..."):
@@ -33,7 +33,14 @@ def run_scripts(target, source, mode, use_face_enhancer):
             cmd1 = ["python3", "run.py", "-s", f'temp/source{target_extension}', "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_swapper"]
             subprocess.run(cmd1)
 
-        if use_face_enhancer and mode != "Face Swapper":
+        if mode == "Face Enhancer":
+            cmd2 = ["python3", "run.py", "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_enhancer"]
+            subprocess.run(cmd2)
+
+        if mode == "Both":
+            cmd1 = ["python3", "run.py", "-s", f'temp/source{target_extension}', "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_swapper"]
+            subprocess.run(cmd1)
+
             cmd2 = ["python3", "run.py", "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_enhancer"]
             subprocess.run(cmd2)
 
@@ -50,9 +57,8 @@ st.markdown('<style>h1{color: Crimson; text-align: center;}</style>', unsafe_all
 source_file = st.file_uploader("Upload source image")
 target_file = st.file_uploader("Upload target image/video")
 mode = st.radio("Choose Mode", ("Face Swapper", "Face Enhancer", "Both"))
-use_face_enhancer = st.checkbox("Use Face Enhancer")
 
 if source_file and target_file and st.button("Swap Faces"):
-    result = run_scripts(target_file, source_file, mode, use_face_enhancer)
+    result = run_scripts(target_file, source_file, mode)
     if result:
         st.image(result)  # Display the result as an image