Update app.py

app.py

@@ -18,16 +18,8 @@ else:
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-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}"
+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)
@@ -46,6 +38,12 @@ def infer(prompt_part1, color, shirt_type, design_prompt, prompt_part5, negative
     
     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;
@@ -68,46 +66,36 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             
-            prompt_part1 = gr.Text(
-                label="Prompt Part 1",
+            prompt_part1 = "a single"
+            gr.Markdown(f"**Prompt Part 1:** {prompt_part1}")
+            
+            prompt_part2 = gr.Text(
+                label="Prompt Part 2",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter prompt part 1",
+                placeholder="Enter prompt part 2 (e.g., color category)",
                 container=False,
-                value="a single",
-                visible=False,
             )
             
-            color = gr.Dropdown(
-                label="Color",
-                choices=color_options,
-                value="Red"
-            )
-            
-            shirt_type = gr.Dropdown(
-                label="Shirt Type",
-                choices=shirt_type_options,
-                value="T-shirt"
-            )
-            
-            design_prompt = gr.Text(
-                label="Design Prompt",
+            prompt_part3 = gr.Text(
+                label="Prompt Part 3",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter design prompt",
+                placeholder="Enter prompt part 3 (e.g., t-shirt, sweatshirt, shirt, hoodie)",
                 container=False,
             )
             
-            prompt_part5 = gr.Text(
-                label="Prompt Part 5",
+            prompt_part4 = gr.Text(
+                label="Prompt Part 4",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter prompt part 5",
+                placeholder="Enter design prompt",
                 container=False,
-                value="hanging on the plain grey wall",
-                visible=False,
             )
             
+            prompt_part5 = "hanging on the plain grey wall"
+            gr.Markdown(f"**Prompt Part 5:** {prompt_part5}")
+            
             run_button = gr.Button("Run", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
@@ -156,20 +144,25 @@ with gr.Blocks(css=css) as demo:
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
-                    value=7.5,
+                    value=0.0,
                 )
                 
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
                     minimum=1,
-                    maximum=50,
+                    maximum=12,
                     step=1,
-                    value=25,
+                    value=2,
                 )
+        
+        gr.Examples(
+            examples = examples,
+            inputs = [prompt_part2]
+        )
 
     run_button.click(
         fn = infer,
-        inputs = [prompt_part1, color, shirt_type, design_prompt, prompt_part5, 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]
     )