Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 import subprocess
 import shutil
 import os
@@ -6,12 +6,13 @@ 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
 
     if not use_face_enhancer:
-        # Run both scripts
+        # Run the first script
         cmd1 = ["python3", "run.py", "-s", source.name, "-t", target.name, "-o", output_path1, "--frame-processor", "face_swapper"]
         subprocess.run(cmd1)
 
@@ -25,17 +26,19 @@ def run_scripts(target, source, use_face_enhancer):
 
     return output_path2
 
-iface = gr.Interface(
-    fn=run_scripts,
-    inputs=[
-        "file",
-        "file",
-        gr.inputs.Checkbox(default=False, label="Use only Face Enhancer")  # New checkbox input
-    ],
-    outputs="file",
-    title="Face swapper",
-    description="Upload a target image/video and a source image to swap faces.",
-    live=True
-)
-
-iface.launch()
+st.title("Face swapper")
+
+target_image = st.file_uploader("Upload target image", type=["jpg", "png"])
+
+source_image = st.file_uploader("Upload source image (optional)", type=["jpg", "png"])
+
+use_face_enhancer = st.checkbox("Use only Face Enhancer", value=False)
+
+if st.button("Swap Faces"):
+    output_file = run_scripts(target_image, source_image, use_face_enhancer)
+    if output_file:
+        st.success("Face swapping completed!")
+        st.image(output_file)
+    else:
+        st.error("Please provide a target image.")
+