Update app.py

app.py

@@ -1,3 +1,5 @@
+# -* coding:UTF-8 -*
+# !/usr/bin/env python
 import numpy as np
 import gradio as gr
 import roop.globals
@@ -12,8 +14,12 @@ from roop.utilities import normalize_output_path
 import os
 from PIL import Image
 
+# Define HTML content for banners and additional sections
+html_banner = "<h2>Welcome to Face Swapper App</h2>"
+html_section1 = "<h3>Instructions:</h3><p>Upload source and target images, and select 'Face Enhancer' option if desired.</p>"
+html_section2 = "<h3>Additional Info:</h3><p>This app uses CUDA for faster processing.</p>"
 
-def swap_face(source_file, target_file,doFaceEnhancer):
+def swap_face(source_file, target_file, doFaceEnhancer):
 
     source_path = "input.jpg"
     target_path = "target.jpg"
@@ -32,8 +38,8 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     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"]
+    if doFaceEnhancer:
+        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
     else:
         roop.globals.frame_processors = ["face_swapper"]
     roop.globals.headless = True
@@ -63,33 +69,14 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     start()
     return output_path
 
-
+# Define Gradio interface with HTML banners and sections
 app = gr.Interface(
     fn=swap_face,
-    inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="face_enhancer?", info="do face enhancer?")],
+    inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="Enable Face Enhancer", default=False)],
     outputs="image",
-    layout="vertical"
+    title="Face Swapper",
+    description="Swap faces between two images with optional face enhancement.",
+    html=[html_banner, html_section1, html_section2]
 )
 
-# Add HTML sections
-app.components.insert(0, gr.HTML("""
-    <div style="background-color: #f0f0f0; padding: 10px; border-radius: 5px;">
-        <h3>Banner 1: Welcome to Face Swap!</h3>
-        <p>Upload your source and target images to begin.</p>
-    </div>
-"""))
-
-app.components.insert(2, gr.HTML("""
-    <div style="background-color: #e0f0e0; padding: 10px; border-radius: 5px;">
-        <h4>Banner 2: Optional Face Enhancement</h4>
-        <p>Check the box to apply face enhancement to the output.</p>
-    </div>
-"""))
-
-app.components.append(gr.HTML("""
-    <div style="background-color: #f0f0f0; padding: 10px; border-radius: 5px;">
-        <p>Banner 3: Results will appear below.</p>
-    </div>
-"""))
-
-app.launch()
+app.launch()