updated

app.py

@@ -1,4 +1,46 @@
 import streamlit as st
+import os
+import subprocess
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+# 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)
+
+# 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"])
+
+    # Output video quality
+    output_quality = st.slider("Select Output Video Quality", min_value=1, max_value=100, value=80)
+
+    # Run script button
+    if st.button("Run Face Swapper"):
+        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)
+
+            # Create output directory
+            os.makedirs("./output", exist_ok=True)
+
+            # Run the script
+            run_script(image_path, video_path, output_quality)
+
+            # Display the output video
+            st.video("./output/swapped.mp4")
+        else:
+            st.warning("Please upload both an image and a video.")
+
+if __name__ == "__main__":
+    main()