Update app.py

app.py

@@ -1,38 +1,47 @@
 import streamlit as st
 import subprocess
 import shutil
+from io import BytesIO
 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..."):
-        # 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
 
+        target_bytes = target.read()  # Read target file as bytes
+        source_bytes = source.read()  # Read source file as bytes
+
+        target_io = BytesIO(target_bytes)  # Convert bytes to BytesIO
+        source_io = BytesIO(source_bytes)  # Convert bytes to BytesIO
+
+        # Save files to temporary directory
+        with open(f'temp/target{target_extension}', 'wb') as f:
+            f.write(target_bytes)
+        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.name}', "-t", f'temp/{target.name}', "-o", output_path1, "--frame-processor", "face_swapper"]
+            cmd1 = ["python3", "run.py", "-s", f'temp/source{target_extension}', "-t", f'temp/target{target_extension}', "-o", output_path1, "--frame-processor", "face_swapper"]
             subprocess.run(cmd1)
 
         # Run the second script
-        cmd2 = ["python3", "run.py", "-t", f'temp/{target.name}' if use_face_enhancer else output_path1, "-o", output_path2, "--frame-processor", "face_enhancer"]
+        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 not use_face_enhancer:
-            os.remove(f'temp/{source.name}')
-        os.remove(f'temp/{target.name}')
+            os.remove(f'temp/source{target_extension}')
+        os.remove(f'temp/target{target_extension}')
 
         return output_path2