Update app.py

app.py

@@ -80,7 +80,7 @@ def swap_video_fct(video_path, output_path, source_face, destination_face, toler
     dest_face = sorted(dest_face, key = lambda x : x.bbox[0])
 
     if(len(dest_face) == 0):
-        print("No dest face found")
+        print("💡 No dest face found")
         return -1
  
     dest_face_feats = []
@@ -91,7 +91,7 @@ def swap_video_fct(video_path, output_path, source_face, destination_face, toler
     src_face = app.get(source_face)
     src_face = sorted(src_face, key = lambda x : x.bbox[0])
     if(len(src_face) == 0):
-        print("No source face found")
+        print("🚨 No source face found")
         return -1
     
     cap = cv2.VideoCapture(video_path)
@@ -115,7 +115,7 @@ def swap_video_fct(video_path, output_path, source_face, destination_face, toler
         for_range = range(preview-1,preview)
 
     for i in for_range:
-        progress(i/frame_count, desc="Processing")
+        progress(i/frame_count, desc="⏳Processing")
         cap.set(cv2.CAP_PROP_POS_FRAMES, i)
         ret, frame = cap.read()
         #frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
@@ -170,44 +170,44 @@ def show_preview(video_path, frame_number):
     return frame
 
 def create_interface():
-    title = 'Face Swap UI'
+    title = '🧸 🧸Face Swap UI'
     with gr.Blocks(analytics_enabled=False, title=title) as face_swap_ui:
-        with gr.Tab("Swap Face Image"):
+        with gr.Tab("📀 Swap Face Image"):
             with gr.Row():
                 with gr.Column():
-                    image_input = gr.Image(label='Input Image (Click to select a face)', scale=0.5)
+                    image_input = gr.Image(label='🎞️Input Image (📺 Click to select a face)', scale=0.5)
                     with gr.Row():
-                        analyze_button = gr.Button("Analyze")
+                        analyze_button = gr.Button("⌛ Analyze")
                     with gr.Row():
                         with gr.Column():
-                            face_num = gr.Number(label='Recognized Faces')
-                            face_index_num = gr.Number(label='Face Index', precision=0)
-                        selected_face = gr.Image(label='Face to swap', interactive=False)
-                        swap_face = gr.Image(label='Swap Face')
-                    swap_button = gr.Button("Swap")
+                            face_num = gr.Number(label='📹 Recognized Faces')
+                            face_index_num = gr.Number(label='📀 Face Index', precision=0)
+                        selected_face = gr.Image(label='💊Face to swap', interactive=False)
+                        swap_face = gr.Image(label='🧪Swap Face')
+                    swap_button = gr.Button("🧩 Swap")
                 with gr.Column():
-                    image_output = gr.Image(label='Output Image',interactive=False)
+                    image_output = gr.Image(label='📤Output Image',interactive=False)
                     #text_output = gr.Textbox(placeholder="What is your name?")
             swap_button.click(fn=swap_face_fct, inputs=[image_input, face_index_num, swap_face], outputs=[image_output])
             image_input.select(select_handler, image_input, [selected_face, face_index_num])
             analyze_button.click(fn=get_faces, inputs=image_input, outputs=[image_input,face_num])
-        with gr.Tab("Swap Face Video"):
+        with gr.Tab("📺Swap Face Video"):
             with gr.Row():
                 with gr.Column():
                     source_video = gr.Video()
-                    video_info = gr.Textbox(label="Video Information")
-                    gr.Markdown("Select a frame for preview with the slider. Then select the face which should be swapped by clicking on it with the cursor")
-                    video_position = gr.Slider(label="Frame preview",interactive=True)
-                    frame_preview = gr.Image(label="Frame preview")
-                    face_index = gr.Textbox(label="Face-Index",interactive=False)
+                    video_info = gr.Textbox(label="📡Video Information")
+                    gr.Markdown("🛠️Select a frame for preview with the slider. Then select the face which should be swapped by clicking on it with the cursor")
+                    video_position = gr.Slider(label="📐Frame preview",interactive=True)
+                    frame_preview = gr.Image(label="📏Frame preview")
+                    face_index = gr.Textbox(label="📉Face-Index",interactive=False)
                     with gr.Row():
-                        dest_face_vid = gr.Image(label="Face tow swap",interactive=True)
-                        source_face_vid = gr.Image(label="New Face")
-                    gr.Markdown("The higher the tolerance the more likely a wrong face will be swapped. 30-40 is a good starting point.")
-                    face_tolerance = gr.Slider(label="Tolerance",value=40,interactive=True)
-                    preview_video = gr.Button("Preview")
-                    video_file_path = gr.Text(label="Output Video path incl. file.mp4 (when left empty it will be put in the gradio temp dir)")
-                    process_video = gr.Button("Process")
+                        dest_face_vid = gr.Image(label="👑Face to swap",interactive=True)
+                        source_face_vid = gr.Image(label="🔮New Face")
+                    gr.Markdown("🔑The higher the tolerance the more likely a wrong face will be swapped. 30-40 is a good starting point.")
+                    face_tolerance = gr.Slider(label="⏳Tolerance",value=40,interactive=True)
+                    preview_video = gr.Button("🪞Preview")
+                    video_file_path = gr.Text(label="🗳️Output Video path incl. file.mp4 (when left empty it will be put in the gradio temp dir)")
+                    process_video = gr.Button("⌛Process")
                 with gr.Column():
                     with gr.Column(scale=1):
                         image_output = gr.Image()