Update app.py

app.py

@@ -1,3 +1,9 @@
+# -*- coding: UTF-8 -*-
+# !/usr/bin/env python
+
+import os
+import shutil
+from PIL import Image
 import gradio as gr
 import roop.globals
 from roop.core import (
@@ -8,40 +14,35 @@ from roop.core import (
 )
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
-from PIL import Image
-import shutil
 
 
-def swap_face(source_file, target_file, doFaceEnhancer, mode):
-    # Save source image always
+def swap_face(source_file, target_file, do_face_enhancer, mode):
+    # Save source image
     source_path = "input.jpg"
     source_image = Image.open(source_file)
     source_image.save(source_path)
     roop.globals.source_path = source_path
 
+    # Handle target file
     if mode == "Image to Image":
-        # target is image
         target_path = "target.jpg"
         target_image = Image.open(target_file)
         target_image.save(target_path)
         output_path = "output.jpg"
-    else:
-        # mode == "Image to Video"
-        # target is video file path
+    else:  # Image to Video
         target_path = "target.mp4"
-        shutil.copy(target_file, target_path)
+        with open(target_path, "wb") as f:
+            f.write(target_file.read())
         output_path = "output.mp4"
 
+    # Assign paths
     roop.globals.target_path = target_path
     roop.globals.output_path = normalize_output_path(
         roop.globals.source_path, roop.globals.target_path, output_path
     )
 
-    if doFaceEnhancer:
-        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
-    else:
-        roop.globals.frame_processors = ["face_swapper"]
-
+    # Frame processor settings
+    roop.globals.frame_processors = ["face_swapper", "face_enhancer"] if do_face_enhancer else ["face_swapper"]
     roop.globals.headless = True
     roop.globals.keep_fps = True
     roop.globals.keep_audio = True
@@ -53,28 +54,31 @@ def swap_face(source_file, target_file, doFaceEnhancer, mode):
     roop.globals.execution_providers = decode_execution_providers(["cuda"])
     roop.globals.execution_threads = suggest_execution_threads()
 
+    # ✅ Fix: Set temp_frame_quality to avoid crash
+    roop.globals.temp_frame_quality = 80
+
+    # Check modules
     for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
         if not frame_processor.pre_check():
             return "Processor check failed"
 
+    # Start the process
     start()
 
     return roop.globals.output_path
 
 
+# Gradio UI
 app = gr.Interface(
     fn=swap_face,
     inputs=[
-        gr.Image(type="filepath", label="Source Face Image"),
-        gr.File(label="Target File (Image or Video)"),
-        gr.Checkbox(label="Apply Face Enhancer?"),
-        gr.Dropdown(
-            choices=["Image to Image", "Image to Video"],
-            value="Image to Image",
-            label="Select Swap Mode",
-        ),
+        gr.Image(label="Source Image", type="filepath"),
+        gr.File(label="Target (Image or Video)", file_types=[".jpg", ".jpeg", ".png", ".mp4"]),
+        gr.Checkbox(label="Enable Face Enhancer?"),
+        gr.Radio(choices=["Image to Image", "Image to Video"], label="Swap Mode", value="Image to Image")
     ],
-    outputs=gr.Video(label="Output (Video or Image)"),
+    outputs=gr.outputs.Image(type="filepath", label="Output")  # Will auto-handle video too
 )
 
-app.launch()
+# Launch app with public link
+app.launch(share=True)