Update app.py

app.py

@@ -69,7 +69,6 @@ def run(domain_source, action_source, hair_source, top_source, bottom_source, do
     # == Source Avatar ==
     # body
     body_source = '0'
-    body_target = '1'
     
     # hair
     if hair_source == "green": hair_source = '0'
@@ -95,12 +94,28 @@ def run(domain_source, action_source, hair_source, top_source, bottom_source, do
     gif = display_gif_pad(file_name_source, 'avatar_source.gif')
     
     # == Target Avatar ==
+    # body
     body_target = '1'
-    hair_target = np.random.choice(['1', '3', '5', '6', '9'])
-    top_target = np.random.choice(['3', '4', '5', '6'])
-    bottom_target = np.random.choice(['4', '5', '6'])
     
-    file_name_target = './Sprite/frames/domain_2/' + action + '/'
+    # hair
+    if hair_target == "violet": hair_target = '1'
+    elif hair_target == "silver": hair_target = '3'
+    elif hair_target == "purple": hair_target = '5'
+    elif hair_target == "grey": hair_target = '6'
+    elif hair_target == "golden": hair_target = '9'
+    
+    # top
+    if top_target == "grey": top_target = '3'
+    elif top_target == "khaki": top_target = '4'
+    elif top_target == "linen": top_target = '5'
+    elif top_target == "ocre": top_target = '6'
+    
+    # bottom
+    if bottom_source == "denim": bottom_source = '4'
+    elif bottom_source == "olive": bottom_source = '5'
+    elif bottom_source == "brown": bottom_source = '6'
+    
+    file_name_target = './Sprite/frames/domain_2/' + action_target + '/'
     file_name_target = file_name_target + 'front' + '_' + str(body_target) + str(bottom_target) + str(top_target) + str(hair_target) + '_'
     
     gif_target = display_gif_pad(file_name_target, 'avatar_target.gif')
@@ -118,9 +133,9 @@ gr.Interface(
         gr.Radio(choices=["white", "golden", "red", "silver"], value="white"),
         gr.Textbox(value="Target Avatar - Alien", interactive=False),
         gr.Radio(choices=["slash", "spellcard", "walk"], value="walk"),
-        gr.Radio(choices=["green", "yellow", "rose", "red", "wine"], value="green"),
-        gr.Radio(choices=["brown", "blue", "white"], value="brown"),
-        gr.Radio(choices=["white", "golden", "red", "silver"], value="white"),
+        gr.Radio(choices=["violet", "silver", "purple", "grey", "golden"], value="golden"),
+        gr.Radio(choices=["grey", "khaki", "linen", "ocre"], value="ocre"),
+        gr.Radio(choices=["denim", "olive", "brown"], value="white"),
     ],
     outputs=[
         gr.components.Image(type="file", label="Source Avatar (Costumed by You)"),