Update app.py

app.py

@@ -189,21 +189,21 @@ def process(
 
     
 
-    yield "### \n ⌛ Loading face analyser model...", *ui_before()
+    yield "### \n 💊 Loading face analyser model...", *ui_before()
     load_face_analyser_model()
 
-    yield "### \n ⌛ Loading face swapper model...", *ui_before()
+    yield "### \n 💊 Loading face swapper model...", *ui_before()
     load_face_swapper_model()
 
     if face_enhancer_name != "NONE":
         if face_enhancer_name not in cv2_interpolations:
-            yield f"### \n ⌛ Loading {face_enhancer_name} model...", *ui_before()
+            yield f"### \n 🪜 Loading {face_enhancer_name} model...", *ui_before()
         FACE_ENHANCER = load_face_enhancer_model(name=face_enhancer_name, device=device)
     else:
         FACE_ENHANCER = None
 
     if enable_face_parser:
-        yield "### \n ⌛ Loading face parsing model...", *ui_before()
+        yield "### \n 🧲 Loading face parsing model...", *ui_before()
         load_face_parser_model()
 
     includes = mask_regions_to_list(mask_includes)
@@ -221,7 +221,7 @@ def process(
         ## ------------------------------ CONTENT CHECK ------------------------------
 
     
-        yield "### \n ⌛ Analysing face data...", *ui_before()
+        yield "### \n 📡 Analysing face data...", *ui_before()
         if condition != "Specific Face":
             source_data = source_path, age
         else:
@@ -237,7 +237,7 @@ def process(
 
         ## ------------------------------ SWAP FUNC ------------------------------
 
-        yield "### \n ⌛ Generating faces...", *ui_before()
+        yield "### \n ⚙️ Generating faces...", *ui_before()
         preds = []
         matrs = []
         count = 0
@@ -251,13 +251,13 @@ def process(
             if USE_CUDA:
                 image_grid = create_image_grid(batch_pred, size=128)
                 PREVIEW = image_grid[:, :, ::-1]
-                yield f"### \n ⌛ Generating face Batch {count}", *ui_before()
+                yield f"### \n ⚙️ Generating face Batch {count}", *ui_before()
 
         ## ------------------------------ FACE ENHANCEMENT ------------------------------
 
         generated_len = len(preds)
         if face_enhancer_name != "NONE":
-            yield f"### \n ⌛ Upscaling faces with {face_enhancer_name}...", *ui_before()
+            yield f"### \n 📐 Upscaling faces with {face_enhancer_name}...", *ui_before()
             for idx, pred in tqdm(enumerate(preds), total=generated_len, desc=f"Upscaling with {face_enhancer_name}"):
                 enhancer_model, enhancer_model_runner = FACE_ENHANCER
                 pred = enhancer_model_runner(pred, enhancer_model)
@@ -267,7 +267,7 @@ def process(
         ## ------------------------------ FACE PARSING ------------------------------
 
         if enable_face_parser:
-            yield "### \n ⌛ Face-parsing mask...", *ui_before()
+            yield "### \n 🖇️ Face-parsing mask...", *ui_before()
             masks = []
             count = 0
             for batch_mask in get_parsed_mask(FACE_PARSER, preds, classes=includes, device=device, batch_size=BATCH_SIZE, softness=int(mask_soft_iterations)):
@@ -278,7 +278,7 @@ def process(
                 if len(batch_mask) > 1:
                     image_grid = create_image_grid(batch_mask, size=128)
                     PREVIEW = image_grid[:, :, ::-1]
-                    yield f"### \n ⌛ Face parsing Batch {count}", *ui_before()
+                    yield f"### \n ✏️ Face parsing Batch {count}", *ui_before()
             masks = np.concatenate(masks, axis=0) if len(masks) >= 1 else masks
         else:
             masks = [None] * generated_len
@@ -294,7 +294,7 @@ def process(
 
         ## ------------------------------ PASTE-BACK ------------------------------
 
-        yield "### \n ⌛ Pasting back...", *ui_before()
+        yield "### \n 🛠️ Pasting back...", *ui_before()
         def post_process(frame_idx, frame_img, split_preds, split_matrs, split_masks, enable_laplacian_blend, crop_mask, blur_amount, erode_amount):
             whole_img_path = frame_img
             whole_img = cv2.imread(whole_img_path)
@@ -350,7 +350,7 @@ def process(
         temp_path = os.path.join(output_path, output_name, "sequence")
         os.makedirs(temp_path, exist_ok=True)
 
-        yield "### \n ⌛ Extracting video frames...", *ui_before()
+        yield "### \n 💽 Extracting video frames...", *ui_before()
         image_sequence = []
         cap = cv2.VideoCapture(video_path)
         curr_idx = 0
@@ -367,12 +367,12 @@ def process(
         for info_update in swap_process(image_sequence):
             yield info_update
 
-        yield "### \n ⌛ Merging sequence...", *ui_before()
+        yield "### \n 🔗 Merging sequence...", *ui_before()
         output_video_path = os.path.join(output_path, output_name + ".mp4")
         merge_img_sequence_from_ref(video_path, image_sequence, output_video_path)
 
         if os.path.exists(temp_path) and not keep_output_sequence:
-            yield "### \n ⌛ Removing temporary files...", *ui_before()
+            yield "### \n 🚽 Removing temporary files...", *ui_before()
             shutil.rmtree(temp_path)
 
         WORKSPACE = output_path
@@ -490,7 +490,7 @@ def video_changed(video_path):
 
 
 def analyse_settings_changed(detect_condition, detection_size, detection_threshold):
-    yield "### \n ⌛ Applying new values..."
+    yield "### \n 💡 Applying new values..."
     global FACE_ANALYSER
     global DETECT_CONDITION
     DETECT_CONDITION = detect_condition
@@ -526,7 +526,7 @@ def slider_changed(show_frame, video_path, frame_index):
 
 
 def trim_and_reload(video_path, output_path, output_name, start_frame, stop_frame):
-    yield video_path, f"### \n ⌛ Trimming video frame {start_frame} to {stop_frame}..."
+    yield video_path, f"### \n 🛠️ Trimming video frame {start_frame} to {stop_frame}..."
     try:
         output_path = os.path.join(output_path, output_name)
         trimmed_video = trim_video(video_path, output_path, start_frame, stop_frame)
@@ -543,12 +543,12 @@ footer{display:none !important}
 """
 
 with gr.Blocks(css=css) as interface:
-    gr.Markdown("# 🗿 Swap Mukham")
+    gr.Markdown("# 🧸 Deepfake Faceswap")
     gr.Markdown("### Face swap app based on insightface inswapper.")
     with gr.Row():
         with gr.Row():
             with gr.Column(scale=0.4):
-                with gr.Tab("📄 Swap Condition"):
+                with gr.Tab("⚖️ Swap Condition"):
                     swap_option = gr.Dropdown(
                         swap_options_list,
                         info="Choose which face or faces in the target image to swap.",
@@ -561,7 +561,7 @@ with gr.Blocks(css=css) as interface:
                         value=25, label="Value", interactive=True, visible=False
                     )
 
-                with gr.Tab("🎚️ Detection Settings"):
+                with gr.Tab("🧫 Detection Settings"):
                     detect_condition_dropdown = gr.Dropdown(
                         detect_conditions,
                         label="Condition",
@@ -579,7 +579,7 @@ with gr.Blocks(css=css) as interface:
                     )
                     apply_detection_settings = gr.Button("Apply settings")
 
-                with gr.Tab("📤 Output Settings"):
+                with gr.Tab("♻️ Output Settings"):
                     output_directory = gr.Text(
                         label="Output Directory",
                         value=DEF_OUTPUT_PATH,
@@ -592,7 +592,7 @@ with gr.Blocks(css=css) as interface:
                         label="Keep output sequence", value=False, interactive=True
                     )
 
-                with gr.Tab("🪄 Other Settings"):
+                with gr.Tab("☢️ Other Settings"):
                     face_scale = gr.Slider(
                         label="Face Scale",
                         minimum=0,
@@ -707,7 +707,7 @@ with gr.Blocks(css=css) as interface:
                         video_input = gr.Video(
                             label="Target Video", interactive=True
                         )
-                        with gr.Accordion("✂️ Trim video", open=False):
+                        with gr.Accordion("🎨 Trim video", open=False):
                             with gr.Column():
                                 with gr.Row():
                                     set_slider_range_btn = gr.Button(
@@ -754,8 +754,8 @@ with gr.Blocks(css=css) as interface:
                 info = gr.Markdown(value="...")
 
                 with gr.Row():
-                    swap_button = gr.Button("✨ Swap", variant="primary")
-                    cancel_button = gr.Button("⛔ Cancel")
+                    swap_button = gr.Button("🎯 Swap", variant="primary")
+                    cancel_button = gr.Button("❌ Cancel")
 
                 preview_image = gr.Image(label="Output", interactive=False)
                 preview_video = gr.Video(
@@ -764,28 +764,28 @@ with gr.Blocks(css=css) as interface:
 
                 with gr.Row():
                     output_directory_button = gr.Button(
-                        "📂", interactive=False, visible=False
+                        "💌", interactive=False, visible=False
                     )
                     output_video_button = gr.Button(
-                        "🎬", interactive=False, visible=False
+                        "📽️", interactive=False, visible=False
                     )
 
                 with gr.Box():
                     with gr.Row():
                         gr.Markdown(
-                            "### [🤝 Sponsor](https://github.com/sponsors/harisreedhar)"
+                            "### [🎭 Sponsor]"
                         )
                         gr.Markdown(
-                            "### [👨‍💻 Source code](https://github.com/harisreedhar/Swap-Mukham)"
+                            "### [🖥️ Source code](https://huggingface.co/spaces/victorisgeek/SwapFace2Pon)"
                         )
                         gr.Markdown(
-                            "### [⚠️ Disclaimer](https://github.com/harisreedhar/Swap-Mukham#disclaimer)"
+                            "### [ 🧩 Playground](https://huggingface.co/spaces/victorisgeek/SwapFace2Pon)"
                         )
                         gr.Markdown(
-                            "### [🌐 Run in Colab](https://colab.research.google.com/github/harisreedhar/Swap-Mukham/blob/main/swap_mukham_colab.ipynb)"
+                            "### [📸 Run in Colab](https://colab.research.google.com/github/victorgeel/FaceSwapNoNfsw/blob/main/SwapFace.ipynb)"
                         )
                         gr.Markdown(
-                            "### [🤗 Acknowledgements](https://github.com/harisreedhar/Swap-Mukham#acknowledgements)"
+                            "### [🤗 Modified Version](https://github.com/victorgeel/FaceSwapNoNfsw)"
                         )
 
     ## ------------------------------ GRADIO EVENTS ------------------------------