Update app.py

app.py

@@ -2,12 +2,8 @@ import streamlit as st
 import subprocess
 import shutil
 import os
-from PIL import Image
-import numpy as np
 
 def run_scripts(target, source, use_face_enhancer):
-    print(f"Source: {source.name}, Target: {target.name}")
-    
     if target is None or (not use_face_enhancer and source is None):
         return None
     target_extension = os.path.splitext(target.name)[-1]
@@ -29,32 +25,20 @@ def run_scripts(target, source, use_face_enhancer):
 
     return output_path2
 
-
-def main():
-    st.markdown('<p style="color:#191970;text-align:center;font-size:30px;">Aiconvert.online</p>', unsafe_allow_html=True)
-    st.title("AIconvert Face swap")
-    st.markdown('<style>h1{color: Crimson; text-align: center;}</style>', unsafe_allow_html=True)
-
-    source_file = st.file_uploader("Upload Source Image", type=["jpg", "png", "jpeg"])
-    target_file = st.file_uploader("Upload Target Image", type=["jpg", "png", "jpeg"])
-    doFaceEnhancer = st.checkbox("Enable Face Enhancer")
-
-    if source_file is not None and target_file is not None:
-        source_image = Image.open(source_file)
-        target_image = Image.open(target_file)
-
-        st.image(source_image, caption="Source Image", use_column_width=True)
-        st.image(target_image, caption="Target Image", use_column_width=True)
-
-        if st.button("Swap Faces"):
-            with st.spinner("Swapping Faces..."):
-                output_path = run_scripts(
-                    target_file,
-                    source_file if not doFaceEnhancer else None,
-                    doFaceEnhancer,
-                )
-                output_image = Image.open(output_path)
-                st.image(output_image, caption="Result", use_column_width=True)
-
-if __name__ == "__main__":
-    main()
+# Streamlit UI
+st.title("Face Swapper")
+st.sidebar.header("Options")
+source_file = st.sidebar.file_uploader("Upload source image")
+target_file = st.sidebar.file_uploader("Upload target image/video")
+use_face_enhancer = st.sidebar.checkbox("Use only Face Enhancer", False)
+
+if st.sidebar.button("Swap Faces"):
+    result = run_scripts(target_file, source_file, use_face_enhancer)
+    if result:
+        st.video(result)  # Display the result as a video
+
+# Cleanup uploaded files after processing
+if source_file:
+    os.remove(source_file.name)
+if target_file:
+    os.remove(target_file.name)