Update app.py

app.py

@@ -1,4 +1,4 @@
-# -* coding:UTF-8 -*
+# -*- coding: UTF-8 -*-
 # !/usr/bin/env python
 import numpy as np
 import gradio as gr
@@ -16,6 +16,7 @@ from PIL import Image
 
 
 def swap_face(source_file, target_file, doFaceEnhancer):
+
     source_path = "input.jpg"
     target_path = "target.jpg"
 
@@ -33,7 +34,7 @@ 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:
+    if doFaceEnhancer == True:
         roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
     else:
         roop.globals.frame_processors = ["face_swapper"]
@@ -65,16 +66,25 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     return output_path
 
 
-with gr.Blocks() as app:
-    gr.HTML("<div style='border: 1px solid black; padding: 10px; margin-bottom: 10px;'>Banner 1</div>")
-    gr.HTML("<div style='border: 1px solid black; padding: 10px; margin-bottom: 10px;'>Banner 2</div>")
-    gr.HTML("<div style='border: 1px solid black; padding: 10px; margin-bottom: 10px;'>Banner 3</div>")
-    
-    interface = gr.Interface(
-        fn=swap_face, 
-        inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="Face Enhancer?", info="Do face enhancer?")], 
-        outputs="image"
-    )
-    interface.launch()
+# Define HTML banners
+banner1 = gr.HTML("<h1>Welcome to the Face Swap App</h1>")
+banner2 = gr.HTML("<h2>Upload your source and target images below</h2>")
+banner3 = gr.HTML("<p>Check the box if you want to use the face enhancer feature</p>")
+
+# Define the interface with HTML banners
+app = gr.Interface(
+    fn=swap_face,
+    inputs=[
+        banner1,
+        banner2,
+        gr.Image(label="Source Image"),
+        gr.Image(label="Target Image"),
+        gr.Checkbox(label="Face Enhancer?", info="Do face enhancer?")
+    ],
+    outputs="image",
+    live=False,
+    title="Face Swap Application",
+    description="A simple app to swap faces between two images"
+)
 
 app.launch()