Update app.py

app.py

@@ -10,13 +10,12 @@ os.makedirs('temp', exist_ok=True)
 
 
 
-def run_scripts(target, source, use_face_enhancer):
-    if target is None or (not use_face_enhancer and source is None):
+def run_scripts(target, source, mode, use_face_enhancer):
+    if target is None or (source is None):
         return None
     with st.spinner("Processing..."):
         target_extension = os.path.splitext(target.name)[-1]
-        output_path1 = "output1" + target_extension
-        output_path2 = "output2" + target_extension
+        output_path = "output" + target_extension
 
         target_bytes = target.read()  # Read target file as bytes
         source_bytes = source.read()  # Read source file as bytes
@@ -30,20 +29,18 @@ def run_scripts(target, source, use_face_enhancer):
         with open(f'temp/source{target_extension}', 'wb') as f:
             f.write(source_bytes)
 
-        if not use_face_enhancer:
-            # Run both scripts
-            cmd1 = ["python3", "run.py", "-s", f'temp/source{target_extension}', "-t", f'temp/target{target_extension}', "-o", output_path1, "--frame-processor", "face_swapper"]
+        if mode == "Face Swapper" or 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)
 
-        # Run the second script
-        cmd2 = ["python3", "run.py", "-t", f'temp/target{target_extension}' if use_face_enhancer else output_path1, "-o", output_path2, "--frame-processor", "face_enhancer"]
-        subprocess.run(cmd2)
+        if mode == "Face Enhancer" or mode == "Both":
+            cmd2 = ["python3", "run.py", "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_enhancer"]
+            subprocess.run(cmd2)
 
-        if not use_face_enhancer:
-            os.remove(f'temp/source{target_extension}')
+        os.remove(f'temp/source{target_extension}')
         os.remove(f'temp/target{target_extension}')
 
-        return output_path2
+        return output_path
 
 # Streamlit UI
 st.markdown('<p style="color:#191970;text-align:center;font-size:30px;">Aiconvert.online</p>', unsafe_allow_html=True)
@@ -52,9 +49,10 @@ 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")
-use_face_enhancer = st.checkbox("Use only Face Enhancer", False)
+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, use_face_enhancer)
+    result = run_scripts(target_file, source_file, mode, use_face_enhancer)
     if result:
-        st.image(result)
+        st.image(result)  # Display the result as an image