Update app.py

app.py

@@ -18,8 +18,16 @@ else:
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-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}"
+color_options = [
+    "Red", "Blue", "Green", "Yellow", "Black", "White", "Purple", "Orange", "Pink", "Brown", "Mixed"
+]
+
+shirt_type_options = [
+    "T-shirt", "Sweatshirt", "Shirt", "Hoodie"
+]
+
+def infer(prompt_part1, color, shirt_type, design_prompt, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+    prompt = f"{prompt_part1} {color} {shirt_type} {design_prompt} {prompt_part5}"
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -38,12 +46,6 @@ def infer(prompt_part1, prompt_part2, prompt_part3, prompt_part4, prompt_part5,
     
     return image
 
-examples = [
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
-    "An astronaut riding a green horse",
-    "A delicious ceviche cheesecake slice",
-]
-
 css="""
 #col-container {
     margin: 0 auto;
@@ -73,26 +75,23 @@ with gr.Blocks(css=css) as demo:
                 placeholder="Enter prompt part 1",
                 container=False,
                 value="a single",
+                visible=False,
             )
             
-            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,
+            color = gr.Dropdown(
+                label="Color",
+                choices=color_options,
+                value="Red"
             )
             
-            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,
+            shirt_type = gr.Dropdown(
+                label="Shirt Type",
+                choices=shirt_type_options,
+                value="T-shirt"
             )
             
-            prompt_part4 = gr.Text(
-                label="Prompt Part 4",
+            design_prompt = gr.Text(
+                label="Design Prompt",
                 show_label=False,
                 max_lines=1,
                 placeholder="Enter design prompt",
@@ -106,6 +105,7 @@ with gr.Blocks(css=css) as demo:
                 placeholder="Enter prompt part 5",
                 container=False,
                 value="hanging on the plain grey wall",
+                visible=False,
             )
             
             run_button = gr.Button("Run", scale=0)
@@ -156,25 +156,20 @@ with gr.Blocks(css=css) as demo:
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
-                    value=0.0,
+                    value=7.5,
                 )
                 
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
                     minimum=1,
-                    maximum=12,
+                    maximum=50,
                     step=1,
-                    value=2,
+                    value=25,
                 )
-        
-        gr.Examples(
-            examples = examples,
-            inputs = [prompt_part1]
-        )
 
     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, color, shirt_type, design_prompt, prompt_part5, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result]
     )