app.py

app.py

@@ -26,7 +26,7 @@ from utils import trim_video, StreamerThread, ProcessBar, open_directory, split_
 
 ## ------------------------------ USER ARGS ------------------------------
 
-parser = argparse.ArgumentParser(description="Swap-Mukham Face Swapper")
+parser = argparse.ArgumentParser(description="FSawap-BypassNSFW")
 parser.add_argument("--out_dir", help="Default Output directory", default=os.getcwd())
 parser.add_argument("--batch_size", help="Gpu batch size", default=32)
 parser.add_argument("--cuda", action="store_true", help="Enable cuda", default=False)
@@ -183,27 +183,27 @@ def process(
 
     start_time = time.time()
     total_exec_time = lambda start_time: divmod(time.time() - start_time, 60)
-    get_finsh_text = lambda start_time: f"✔️ Completed in {int(total_exec_time(start_time)[0])} min {int(total_exec_time(start_time)[1])} sec."
+    get_finsh_text = lambda start_time: f"📌 Completed in {int(total_exec_time(start_time)[0])} min {int(total_exec_time(start_time)[1])} sec."
 
     ## ------------------------------ PREPARE INPUTS & LOAD MODELS ------------------------------
 
     
 
-    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 ⏰ 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 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 🔮 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 🩵 Face ကိုဖန်တီးနေပါပြီ...", *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 🧲 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 ပြုလုပ်နေပြီ...", *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
@@ -500,7 +500,7 @@ def analyse_settings_changed(detect_condition, detection_size, detection_thresho
         det_size=(int(detection_size), int(detection_size)),
         det_thresh=float(detection_threshold),
     )
-    yield f"### \n ✔️ Applied detect condition:{detect_condition}, detection size: {detection_size}, detection threshold: {detection_threshold}"
+    yield f"### \n 📌 Applied detect condition:{detect_condition}, detection size: {detection_size}, detection threshold: {detection_threshold}"
 
 
 def stop_running():
@@ -526,14 +526,14 @@ 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)
-        yield trimmed_video, "### \n ✔️ Video trimmed and reloaded."
+        yield trimmed_video, "### \n ✂️✂️ Video trimmed and reloaded."
     except Exception as e:
         print(e)
-        yield video_path, "### \n ❌ Video trimming failed. See console for more info."
+        yield video_path, "### \n 🔴 Video trimming failed. See console for more info."
 
 
 ## ------------------------------ GRADIO GUI ------------------------------
@@ -543,15 +543,15 @@ footer{display:none !important}
 """
 
 with gr.Blocks(css=css) as interface:
-    gr.Markdown("# 🗿 Swap Mukham")
-    gr.Markdown("### Face swap app based on insightface inswapper.")
+    gr.Markdown("# 💞 GeepDeep 💞")
+    gr.Markdown("###  FaceSwap app အား 🗂️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.",
+                        info="🎛️ပြောင်းလဲလိုသည့် မျက်နှာ အနေအထားအား ရွေးချယ်ပါ🎛️.",
                         multiselect=False,
                         show_label=False,
                         value=swap_options_list[0],
@@ -561,13 +561,13 @@ 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",
                         value=DETECT_CONDITION,
                         interactive=True,
-                        info="This condition is only used when multiple faces are detected on source or specific image.",
+                        info="👫 မျက်နှာနှစ်ခုရွေးရန်အတွက်ဖြစ်သည်၊👫.",
                     )
                     detection_size = gr.Number(
                         label="Detection Size", value=DETECT_SIZE, interactive=True
@@ -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,
@@ -668,7 +668,7 @@ with gr.Blocks(css=css) as interface:
 
 
                 source_image_input = gr.Image(
-                    label="Source face", type="filepath", interactive=True
+                    label="🍭အသုံးပြုမည့်မျက်နှာ🍭", type="filepath", interactive=True
                 )
 
                 with gr.Box(visible=False) as specific_face:
@@ -699,15 +699,15 @@ with gr.Blocks(css=css) as interface:
 
                     with gr.Box(visible=True) as input_image_group:
                         image_input = gr.Image(
-                            label="Target Image", interactive=True, type="filepath"
+                            label="📸 ပြောင်းလဲလိုသည့်မျက်နှာ 📸", interactive=True, type="filepath"
                         )
 
                     with gr.Box(visible=False) as input_video_group:
                         vid_widget = gr.Video if USE_COLAB else gr.Text
                         video_input = gr.Video(
-                            label="Target Video", interactive=True
+                            label="🖥️ပြောင်းမည့်video🖥️", interactive=True
                         )
-                        with gr.Accordion("✂️ Trim video", open=False):
+                        with gr.Accordion("✂️ video တည်းဖြတ်ရန်", open=False):
                             with gr.Column():
                                 with gr.Row():
                                     set_slider_range_btn = gr.Button(
@@ -754,38 +754,35 @@ 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_image = gr.Image(label="💕Output💕", interactive=False)
                 preview_video = gr.Video(
                     label="Output", interactive=False, visible=False
                 )
 
                 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)"
+                            "[🆚မူရင်းကုဒ်](https://huggingface.co/spaces/victorisgeek/SwapYoFace)"
                         )
                         gr.Markdown(
-                            "### [👨‍💻 Source code](https://github.com/harisreedhar/Swap-Mukham)"
+                            "[🚨ရှင်းလင်းချက် ကြည့်ကျက်လုပ်လိုက်]"
                         )
                         gr.Markdown(
-                            "### [⚠️ Disclaimer](https://github.com/harisreedhar/Swap-Mukham#disclaimer)"
+                            "[🧿 Run in Colab](https://colab.research.google.com/github/victorgeel/FaceSwapNoNfsw)"
                         )
                         gr.Markdown(
-                            "### [🌐 Run in Colab](https://colab.research.google.com/github/harisreedhar/Swap-Mukham/blob/main/swap_mukham_colab.ipynb)"
-                        )
-                        gr.Markdown(
-                            "### [🤗 Acknowledgements](https://github.com/harisreedhar/Swap-Mukham#acknowledgements)"
+                            "[💥🖕မအေလိုးမင်းအောင်လှိုင်🖕💥]"
                         )
 
     ## ------------------------------ GRADIO EVENTS ------------------------------
@@ -903,4 +900,4 @@ if __name__ == "__main__":
     if USE_COLAB:
         print("Running in colab mode")
 
-    interface.queue(concurrency_count=2, max_size=20).launch(share=USE_COLAB)
+    interface.queue(concurrency_count=2, max_size=20).launch(share=USE_COLAB)