Update app.py

app.py

@@ -1,5 +1,3 @@
-# -* coding:UTF-8 -*
-# !/usr/bin/env python
 import numpy as np
 import gradio as gr
 import roop.globals
@@ -14,10 +12,6 @@ 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):
 
@@ -69,14 +63,21 @@ 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="Enable Face Enhancer", default=False)],
-    outputs="image",
-    title="Face Swapper",
-    description="Swap faces between two images with optional face enhancement.",
-    html=[html_banner, html_section1, html_section2]
-)
+
+html_section_1 = "<div><h1>Welcome to the Face Swap App</h1></div>"
+html_section_2 = "<div><p>Upload your source and target images to swap faces. Optionally, use the face enhancer feature.</p></div>"
+html_section_3 = "<div><p>After processing, the swapped image will be displayed below.</p></div>"
+
+app = gr.Blocks()
+
+with app:
+    gr.HTML(html_section_1)
+    gr.HTML(html_section_2)
+    gr.HTML(html_section_3)
+    gr.Interface(
+        fn=swap_face,
+        inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="face_enhancer?", info="do face enhancer?")],
+        outputs="image"
+    )
 
 app.launch()