Update app.py

app.py

@@ -18,8 +18,9 @@ else:
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
-
+def infer(prompt_part1, prompt_part2, prompt_part3, prompt_part4, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+    prompt = f"{prompt_part1} {prompt_part2} {prompt_part3} {prompt_part4} {prompt_part5}"
+    
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
         
@@ -65,12 +66,46 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             
-            prompt = gr.Text(
-                label="Prompt",
+            prompt_part1 = gr.Text(
+                label="Prompt Part 1",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter prompt part 1",
+                container=False,
+                value="a single",
+            )
+            
+            prompt_part2 = gr.Text(
+                label="Prompt Part 2",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter prompt part 2 (e.g., color category)",
+                container=False,
+            )
+            
+            prompt_part3 = gr.Text(
+                label="Prompt Part 3",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter prompt part 3 (e.g., t-shirt, sweatshirt, shirt, hoodie)",
+                container=False,
+            )
+            
+            prompt_part4 = gr.Text(
+                label="Prompt Part 4",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter design prompt",
+                container=False,
+            )
+            
+            prompt_part5 = gr.Text(
+                label="Prompt Part 5",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter your prompt",
+                placeholder="Enter prompt part 5",
                 container=False,
+                value="hanging on the plain grey wall",
             )
             
             run_button = gr.Button("Run", scale=0)
@@ -134,13 +169,13 @@ with gr.Blocks(css=css) as demo:
         
         gr.Examples(
             examples = examples,
-            inputs = [prompt]
+            inputs = [prompt_part1]
         )
 
     run_button.click(
         fn = infer,
-        inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs = [prompt_part1, prompt_part2, prompt_part3, prompt_part4, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result]
     )
 
-demo.queue().launch()
+demo.queue().launch()