Update app.py

app.py

@@ -1,5 +1,6 @@
-# -* coding:UTF-8 -*
+# -* coding:UTF-8 *-
 # !/usr/bin/env python
+
 import numpy as np
 import gradio as gr
 import roop.globals
@@ -13,31 +14,50 @@ from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
 import os
 from PIL import Image
+import mimetypes
 
 
-def swap_face(source_file, target_file,doFaceEnhancer):
+def swap_face(source_file, target_file, swap_type, doFaceEnhancer):
+    supported_images = ['.jpg', '.jpeg', '.png']
+    supported_videos = ['.mp4', '.mov', '.avi']
 
+    # Save source image
     source_path = "input.jpg"
-    target_path = "target.jpg"
-
     source_image = Image.fromarray(source_file)
     source_image.save(source_path)
-    target_image = Image.fromarray(target_file)
-    target_image.save(target_path)
 
-    print("source_path: ", source_path)
-    print("target_path: ", target_path)
+    # Save target as image or video
+    mime_type, _ = mimetypes.guess_type(target_file.name)
+    ext = os.path.splitext(target_file.name)[1].lower()
+
+    if mime_type and "video" in mime_type:
+        target_path = "target" + ext
+        with open(target_path, "wb") as f:
+            f.write(target_file.read())
+    else:
+        target_path = "target.jpg"
+        target_image = Image.fromarray(target_file)
+        target_image.save(target_path)
+
+    print("source_path:", source_path)
+    print("target_path:", target_path)
 
     roop.globals.source_path = source_path
     roop.globals.target_path = target_path
-    output_path = "output.jpg"
+
+    # Output path based on type
+    if "video" in swap_type.lower() or ext in supported_videos:
+        output_path = "output.mp4"
+    else:
+        output_path = "output.jpg"
+
     roop.globals.output_path = normalize_output_path(
         roop.globals.source_path, roop.globals.target_path, output_path
     )
-    if doFaceEnhancer == True:
-        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
-    else:
-        roop.globals.frame_processors = ["face_swapper"]
+
+    roop.globals.frame_processors = (
+        ["face_swapper", "face_enhancer"] if doFaceEnhancer else ["face_swapper"]
+    )
     roop.globals.headless = True
     roop.globals.keep_fps = True
     roop.globals.keep_audio = True
@@ -60,13 +80,25 @@ def swap_face(source_file, target_file,doFaceEnhancer):
         roop.globals.frame_processors
     ):
         if not frame_processor.pre_check():
-            return
+            return "Frame processor pre-check failed"
 
     start()
-    return output_path
+    return roop.globals.output_path
 
 
 app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
+    fn=swap_face,
+    inputs=[
+        gr.Image(label="Source Image"),
+        gr.File(label="Target (Image or Video)"),
+        gr.Radio(choices=["Image to Image", "Image to Video"], label="Swap Type"),
+        gr.Checkbox(label="Face Enhancer?", info="Apply face enhancement")
+    ],
+    outputs=[
+        gr.Image(label="Output Image") if not os.path.exists("output.mp4") else gr.Video(label="Output Video")
+    ],
+    title="Face Swap Tool (Roop)",
+    description="Swap face from source image to a target image or video using Roop.",
 )
-app.launch()
+
+app.launch()