Update app.py

app.py

@@ -6,39 +6,37 @@ import os
 def run_scripts(target, source, use_face_enhancer):
     if target is None or (not use_face_enhancer and source is None):
         return None
-    target_extension = os.path.splitext(target.name)[-1]
-    output_path1 = "output1" + target_extension
-    output_path2 = "output2" + target_extension
+    with st.spinner("Processing..."):
+        target_extension = os.path.splitext(target)[-1]
+        output_path1 = "output1" + target_extension
+        output_path2 = "output2" + target_extension
 
-    if not use_face_enhancer:
-        # Run both scripts
-        cmd1 = ["python3", "run.py", "-s", source.name, "-t", target.name, "-o", output_path1, "--frame-processor", "face_swapper"]
-        subprocess.run(cmd1)
+        if not use_face_enhancer:
+            # Run both scripts
+            cmd1 = ["python3", "run.py", "-s", source, "-t", target, "-o", output_path1, "--frame-processor", "face_swapper"]
+            subprocess.run(cmd1)
 
-    # Run the second script
-    cmd2 = ["python3", "run.py", "-t", target.name if use_face_enhancer else output_path1, "-o", output_path2, "--frame-processor", "face_enhancer"]
-    subprocess.run(cmd2)
+        # Run the second script
+        cmd2 = ["python3", "run.py", "-t", target if use_face_enhancer else output_path1, "-o", output_path2, "--frame-processor", "face_enhancer"]
+        subprocess.run(cmd2)
 
-    if not use_face_enhancer:
-        os.remove(source.name)
-    os.remove(target.name)
+        if not use_face_enhancer:
+            os.remove(source)
+        os.remove(target)
 
-    return output_path2
+        return output_path2
 
 # Streamlit UI
 st.title("Face Swapper")
-st.sidebar.header("Options")
-source_file = st.sidebar.file_uploader("Upload source image")
-target_file = st.sidebar.file_uploader("Upload target image/video")
-use_face_enhancer = st.sidebar.checkbox("Use only Face Enhancer", False)
+st.write("Upload a target image/video and a source image to swap faces.")
 
-if st.sidebar.button("Swap Faces"):
-    result = run_scripts(target_file, source_file, use_face_enhancer)
+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)
+
+if st.button("Swap Faces"):
+    result = run_scripts(target_file.name, source_file.name, use_face_enhancer)
     if result:
-        st.video(result)  # Display the result as a video
+        st.image(result)  # Display the result as an image
 
-# Cleanup uploaded files after processing
-if source_file:
-    os.remove(source_file.name)
-if target_file:
-    os.remove(target_file.name)
+# No need to remove uploaded files, as they are now passed as strings.