Update app.py

app.py

@@ -8,22 +8,27 @@ def generate_image(text, seed):
     if seed is not None:
         random.seed(seed)
 
-    if text in [example[0] for example in examples]:
-        print(f"Using example: {text}")
+    # Define slight variations of the original prompt for different poses/positions
+    variations = [
+        f"{text}, front view",
+        f"{text}, side view",
+        f"{text}, dynamic pose",
+        f"{text}, close-up view"
+    ]
 
-    # Generate four images by calling the model four times
-    images = [model(text) for _ in range(4)]
+    # Generate an image for each variation
+    images = [model(variation) for variation in variations]
     return images
 
-# Examples for user guidance
+# Define examples with pre-generated image previews
 examples = [
-    ["Humanoid Cat Warrior, Full View", None],
-    ["Warhammer Sisterhood", None],
-    ["Future Robots war", None],
-    ["Fantasy dragon", None]
+    ["Humanoid Cat Warrior, Full View", None, "path/to/humanoid_cat_preview.jpg"],
+    ["Warhammer Sisterhood", None, "path/to/warhammer_sisterhood_preview.jpg"],
+    ["Future Robots war", None, "path/to/future_robots_war_preview.jpg"],
+    ["Fantasy dragon", None, "path/to/fantasy_dragon_preview.jpg"]
 ]
 
-# Custom CSS for gradient buttons and capsule shapes
+# Custom CSS for styling
 css = """
 /* General button styles */
 button {
@@ -63,17 +68,20 @@ input[type=range] {
 }
 """
 
-# Gradio interface with custom styles
+# Define a function to create the examples with preview images
+def create_example_row(example):
+    return [gr.Example(value=example[0], examples=None, label=example[0], image=example[2])]
+
+# Gradio interface
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
         gr.Textbox(label="Type your imagination:", placeholder="Type or select an example..."),
         gr.Slider(minimum=0, maximum=10000, step=1, label="Seed (optional)")
     ],
-    # Specify a list of 4 Image outputs for displaying multiple images
     outputs=[gr.Image(label=f"Generated Image {i+1}") for i in range(4)],
-    examples=examples,
-    css=css,  # Apply custom CSS styling
+    examples=create_example_row(examples),  # Use pre-generated images for previews
+    css=css,
     description="Note: The model is running on CPU; processing might take a bit longer. Thank you for your patience!"
 )