Update app.py

app.py

@@ -1,46 +1,51 @@
 import streamlit as st
 import os
 import subprocess
+from PIL import Image
+from io import BytesIO
 
-# Function to run the provided script with user inputs
-def run_script(image_path, video_path, output_quality):
-    script_command = (
-        f"python run.py --target {video_path} --output-video-quality {output_quality} "
-        f"--source {image_path} -o ./output/swapped.mp4 --execution-provider cuda --frame-processor face_swapper face_enhancer"
-    )
-    subprocess.run(script_command, shell=True)
+# Function to run the provided script
+def run_script(image_path, video_path):
+    script = """
+    !git clone https://github.com/s0md3v/roop.git
+    %cd roop
+
+    !wget https://huggingface.co/ezioruan/inswapper_128.onnx/resolve/main/inswapper_128.onnx -O inswapper_128.onnx
+    !mkdir models
+    !mv inswapper_128.onnx ./models
+
+    !python run.py --target {} --output-video-quality 80 --source {} -o /content/swapped.mp4 --execution-provider cuda --frame-processor face_swapper face_enhancer
+    """.format(video_path, image_path)
+    
+    subprocess.run(script, shell=True)
 
 # Streamlit app
 def main():
-    st.title("Face Swapper Streamlit App")
-    
-    # Upload image and video
-    st.header("Upload Image and Video")
-    image_file = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
-    video_file = st.file_uploader("Upload Video", type=["mp4"])
+    st.title("Face Swapper App")
 
-    # Output video quality
-    output_quality = st.slider("Select Output Video Quality", min_value=1, max_value=100, value=80)
+    # File upload
+    st.sidebar.header("Upload Files")
+    image_file = st.sidebar.file_uploader("Upload Image", type=["jpg", "png", "jpeg"])
+    video_file = st.sidebar.file_uploader("Upload Video", type=["mp4"])
 
-    # Run script button
-    if st.button("Run Face Swapper"):
+    if st.sidebar.button("Swap Faces"):
         if image_file is not None and video_file is not None:
             # Save uploaded files
-            image_path = "./input/user_image.jpg"
-            video_path = "./input/user_video.mp4"
-            image_file.save(image_path)
-            video_file.save(video_path)
+            image_path = "./uploaded_image.jpg"
+            video_path = "./uploaded_video.mp4"
+
+            with open(image_path, "wb") as f:
+                f.write(image_file.getvalue())
 
-            # Create output directory
-            os.makedirs("./output", exist_ok=True)
+            with open(video_path, "wb") as f:
+                f.write(video_file.getvalue())
 
             # Run the script
-            run_script(image_path, video_path, output_quality)
+            run_script(image_path, video_path)
 
-            # Display the output video
-            st.video("./output/swapped.mp4")
-        else:
-            st.warning("Please upload both an image and a video.")
+            # Display the swapped video
+            video_bytes = open("/content/swapped.mp4", "rb").read()
+            st.video(video_bytes)
 
 if __name__ == "__main__":
     main()