Update app.py

app.py

@@ -3,26 +3,36 @@ import subprocess
 import shutil
 import os
 
+
+# Create 'temp' directory if it doesn't exist
+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):
         return None
     with st.spinner("Processing..."):
-        target_extension = os.path.splitext(target)[-1]
+        # Save uploaded files to temporary directory
+        with open(f'temp/{target.name}', 'wb') as f:
+            f.write(target.getvalue())
+        with open(f'temp/{source.name}', 'wb') as f:
+            f.write(source.getvalue())
+
+        target_extension = os.path.splitext(target.name)[-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, "-t", target, "-o", output_path1, "--frame-processor", "face_swapper"]
+            cmd1 = ["python3", "run.py", "-s", f'temp/{source.name}', "-t", f'temp/{target.name}', "-o", output_path1, "--frame-processor", "face_swapper"]
             subprocess.run(cmd1)
 
         # Run the second script
-        cmd2 = ["python3", "run.py", "-t", target if use_face_enhancer else output_path1, "-o", output_path2, "--frame-processor", "face_enhancer"]
+        cmd2 = ["python3", "run.py", "-t", f'temp/{target.name}' 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)
-        os.remove(target)
+            os.remove(f'temp/{source.name}')
+        os.remove(f'temp/{target.name}')
 
         return output_path2
 
@@ -35,6 +45,6 @@ target_file = st.file_uploader("Upload target image/video")
 use_face_enhancer = st.checkbox("Use only Face Enhancer", False)
 
 if source_file and target_file and st.button("Swap Faces"):
-    result = run_scripts(target_file.name, source_file.name, use_face_enhancer)
+    result = run_scripts(target_file, source_file, use_face_enhancer)
     if result:
         st.image(result)  # Display the result as an image