Apply gradio-i18n

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
+from gradio_i18n import Translate, gettext as _
 
 from modules.live_portrait.live_portrait_inferencer import LivePortraitInferencer
+from modules.utils.paths import *
 
 
 class App:
@@ -8,62 +10,64 @@ class App:
                  args=None):
         self.args = args
         self.app = gr.Blocks()
+        self.i18n = Translate(I18N_YAML_PATH)
         self.inferencer = LivePortraitInferencer()
 
     @staticmethod
     def create_parameters():
         return [
-            gr.Slider(label="Rotate Pitch", minimum=-20, maximum=20, step=0.5, value=0),
-            gr.Slider(label="Rotate Yaw", minimum=-20, maximum=20, step=0.5, value=0),
-            gr.Slider(label="Rotate Roll", minimum=-20, maximum=20, step=0.5, value=0),
-            gr.Slider(label="Blink", minimum=-20, maximum=20, step=0.5, value=0),
-            gr.Slider(label="Eyebrow", minimum=-20, maximum=20, step=0.5, value=0),
-            gr.Slider(label="Wink", minimum=0, maximum=25, step=0.5, value=0),
-            gr.Slider(label="Pupil X", minimum=-20, maximum=20, step=0.5, value=0),
-            gr.Slider(label="Pupil Y", minimum=-20, maximum=20, step=0.5, value=0),
-            gr.Slider(label="AAA", minimum=-30, maximum=120, step=1, value=0),
-            gr.Slider(label="EEE", minimum=-20, maximum=20, step=0.2, value=0),
-            gr.Slider(label="WOO", minimum=-20, maximum=20, step=0.2, value=0),
-            gr.Slider(label="Smile", minimum=-2.0, maximum=2.0, step=0.01, value=0),
-            gr.Slider(label="Source Ratio", minimum=0, maximum=1, step=0.01, value=1),
-            gr.Slider(label="Sample Ratio", minimum=-0.2, maximum=1.2, step=0.01, value=1),
-            gr.Dropdown(label="Sample Parts",
+            gr.Slider(label=_("Rotate Pitch"), minimum=-20, maximum=20, step=0.5, value=0),
+            gr.Slider(label=_("Rotate Yaw"), minimum=-20, maximum=20, step=0.5, value=0),
+            gr.Slider(label=_("Rotate Roll"), minimum=-20, maximum=20, step=0.5, value=0),
+            gr.Slider(label=_("Blink"), minimum=-20, maximum=20, step=0.5, value=0),
+            gr.Slider(label=_("Eyebrow"), minimum=-20, maximum=20, step=0.5, value=0),
+            gr.Slider(label=_("Wink"), minimum=0, maximum=25, step=0.5, value=0),
+            gr.Slider(label=_("Pupil X"), minimum=-20, maximum=20, step=0.5, value=0),
+            gr.Slider(label=_("Pupil Y"), minimum=-20, maximum=20, step=0.5, value=0),
+            gr.Slider(label=_("AAA"), minimum=-30, maximum=120, step=1, value=0),
+            gr.Slider(label=_("EEE"), minimum=-20, maximum=20, step=0.2, value=0),
+            gr.Slider(label=_("WOO"), minimum=-20, maximum=20, step=0.2, value=0),
+            gr.Slider(label=_("Smile"), minimum=-2.0, maximum=2.0, step=0.01, value=0),
+            gr.Slider(label=_("Source Ratio"), minimum=0, maximum=1, step=0.01, value=1),
+            gr.Slider(label=_("Sample Ratio"), minimum=-0.2, maximum=1.2, step=0.01, value=1),
+            gr.Dropdown(label=_("Sample Parts"),
                         choices=["OnlyExpression", "OnlyRotation", "OnlyMouth", "OnlyEyes", "All"], value="All"),
-            gr.Slider(label="Crop Factor", minimum=1.5, maximum=2.5, step=0.1, value=1.7)
+            gr.Slider(label=_("Crop Factor"), minimum=1.5, maximum=2.5, step=0.1, value=1.7)
         ]
 
     def launch(self):
         with self.app:
-            with gr.Row():
-                with gr.Column():
-                    img_ref = gr.Image(label="Reference Image")
-            with gr.Row():
-                btn_gen = gr.Button("GENERATE", visible=False)
-            with gr.Row():
-                with gr.Column(scale=8):
-                    img_out = gr.Image(label="Output Image")
-                with gr.Column(scale=2):
-                    expression_parameters = self.create_parameters()
-                    with gr.Accordion("Opt in features", visible=False):
-                        img_sample = gr.Image()
-                        img_motion_link = gr.Image()
-                        tb_exp = gr.Textbox()
+            with self.i18n:
+                with gr.Row():
+                    with gr.Column():
+                        img_ref = gr.Image(label=_("Reference Image"))
+                with gr.Row():
+                    btn_gen = gr.Button("GENERATE", visible=False)
+                with gr.Row():
+                    with gr.Column(scale=8):
+                        img_out = gr.Image(label=_("Output Image"))
+                    with gr.Column(scale=2):
+                        expression_parameters = self.create_parameters()
+                        with gr.Accordion("Opt in features", visible=False):
+                            img_sample = gr.Image()
+                            img_motion_link = gr.Image()
+                            tb_exp = gr.Textbox()
 
-            params = expression_parameters + [img_ref]
-            opt_in_features_params = [img_sample, img_motion_link, tb_exp]
+                params = expression_parameters + [img_ref]
+                opt_in_features_params = [img_sample, img_motion_link, tb_exp]
 
-            gr.on(
-                triggers=[param.change for param in params],
-                fn=self.inferencer.edit_expression,
-                inputs=params + opt_in_features_params,
-                outputs=img_out
-            )
+                gr.on(
+                    triggers=[param.change for param in params],
+                    fn=self.inferencer.edit_expression,
+                    inputs=params + opt_in_features_params,
+                    outputs=img_out
+                )
 
-            btn_gen.click(self.inferencer.edit_expression,
-                          inputs=params + opt_in_features_params,
-                          outputs=img_out)
+                btn_gen.click(self.inferencer.edit_expression,
+                              inputs=params + opt_in_features_params,
+                              outputs=img_out)
 
-        self.app.launch(inbrowser=True)
+            self.app.launch(inbrowser=True)
 
 
 app = App()

modules/utils/paths.py

@@ -17,6 +17,7 @@ MODEL_PATHS = {
     "face_yolov8n": os.path.join(MODELS_DIR, "face_yolov8n.pt")
 }
 MASK_TEMPLATES = os.path.join(PROJECT_ROOT_DIR, "modules", "utils", "resources", "mask_template.png")
+I18N_YAML_PATH = os.path.join(PROJECT_ROOT_DIR, "i18n", "translation.yaml")
 
 
 def get_auto_incremental_file_path(dir_path: str, extension: str, prefix: str = ""):