Update app.py

app.py

@@ -18,9 +18,9 @@ else:
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-def infer(prompt_part1, color, dress_type, design, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
-    front_prompt = f"{prompt_part1} {color} colored plain {dress_type} with {design} design in front, {prompt_part5}"
-    back_prompt = f"{prompt_part1} {color} colored plain {dress_type} with {design} design in back, {prompt_part5}"
+def infer(prompt_part1, color, dress_type, front_design, back_design, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+    front_prompt = f"front view of {prompt_part1} {color} colored plain {dress_type} with {front_design} design, {prompt_part5}"
+    back_prompt = f"back view of {prompt_part1} {color} colored plain {dress_type} with {back_design} design, {prompt_part5}"
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -50,9 +50,9 @@ def infer(prompt_part1, color, dress_type, design, prompt_part5, negative_prompt
     return front_image, back_image
 
 examples = [
-    "red, t-shirt, yellow stripes",
-    "blue, hoodie, minimalist",
-    "red, sweat shirt, geometric design",
+    "red, t-shirt, yellow stripes, polka dots",
+    "blue, hoodie, minimalist, abstract art",
+    "red, sweat shirt, geometric design, plain",
 ]
 
 css = """
@@ -103,11 +103,19 @@ with gr.Blocks(css=css) as demo:
                 container=False,
             )
             
-            prompt_part4 = gr.Textbox(
-                label="design",
+            prompt_part4_front = gr.Textbox(
+                label="front design",
                 show_label=False,
                 max_lines=1,
-                placeholder="design",
+                placeholder="front design",
+                container=False,
+            )
+
+            prompt_part4_back = gr.Textbox(
+                label="back design",
+                show_label=False,
+                max_lines=1,
+                placeholder="back design",
                 container=False,
             )
             
@@ -184,12 +192,12 @@ with gr.Blocks(css=css) as demo:
         
         gr.Examples(
             examples=examples,
-            inputs=[prompt_part2]
+            inputs=[prompt_part2, prompt_part3, prompt_part4_front, prompt_part4_back]
         )
 
     run_button.click(
         fn=infer,
-        inputs=[prompt_part1, prompt_part2, prompt_part3, prompt_part4, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs=[prompt_part1, prompt_part2, prompt_part3, prompt_part4_front, prompt_part4_back, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs=[front_result, back_result]
     )