Add constant and apply

app.py

@@ -39,7 +39,7 @@ class App:
             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"),
+                        choices=[part.value for part in SamplePart], value=SamplePart.ALL.value),
             gr.Slider(label=_("Crop Factor"), minimum=1.5, maximum=2.5, step=0.1, value=1.7)
         ]
 

modules/live_portrait/live_portrait_inferencer.py

@@ -201,15 +201,15 @@ class LivePortraitInferencer:
                     self.d_info['exp'][0, 5, 1] = 0
 
                 # "OnlyExpression", "OnlyRotation", "OnlyMouth", "OnlyEyes", "All"
-                if sample_parts == "OnlyExpression" or sample_parts == "All":
+                if sample_parts == SamplePart.ONLY_EXPRESSION.value or sample_parts == SamplePart.ONLY_EXPRESSION.ALL.value:
                     es.e += self.d_info['exp'] * sample_ratio
-                if sample_parts == "OnlyRotation" or sample_parts == "All":
+                if sample_parts == SamplePart.ONLY_ROTATION.value or sample_parts == SamplePart.ONLY_ROTATION.ALL.value:
                     rotate_pitch += self.d_info['pitch'] * sample_ratio
                     rotate_yaw += self.d_info['yaw'] * sample_ratio
                     rotate_roll += self.d_info['roll'] * sample_ratio
-                elif sample_parts == "OnlyMouth":
+                elif sample_parts == SamplePart.ONLY_MOUTH.value:
                     self.retargeting(es.e, self.d_info['exp'], sample_ratio, (14, 17, 19, 20))
-                elif sample_parts == "OnlyEyes":
+                elif sample_parts == SamplePart.ONLY_EYES.value:
                     self.retargeting(es.e, self.d_info['exp'], sample_ratio, (1, 2, 11, 13, 15, 16))
 
             es.r = self.calc_fe(es.e, blink, eyebrow, wink, pupil_x, pupil_y, aaa, eee, woo, smile,

modules/utils/constants.py

@@ -7,6 +7,14 @@ class ModelType(Enum):
     ANIMAL = _("Animal")
 
 
+class SamplePart(Enum):
+    ONLY_EXPRESSION = _("OnlyExpression")
+    ONLY_ROTATION = _("OnlyRotation")
+    ONLY_MOUTH = _("OnlyMouth")
+    ONLY_EYES = _("OnlyEyes")
+    ALL = _("All")
+
+
 REPO_MARKDOWN = """
 ## [AdvancedLivePortrait-WebUI](https://github.com/jhj0517/AdvancedLivePortrait-WebUI/tree/master)
 """