Update app.py

app.py

@@ -71,6 +71,13 @@ def vote(filename, vote_type):
     save_vote_counts(vote_counts)
     return get_image_gallery()
 
+def get_random_style():
+    styles = [
+        "Impressionist", "Cubist", "Surrealist", "Abstract Expressionist",
+        "Pop Art", "Minimalist", "Baroque", "Art Nouveau", "Pointillist", "Fauvism"
+    ]
+    return random.choice(styles)
+
 MAX_SEED = np.iinfo(np.int32).max
 
 if not torch.cuda.is_available():
@@ -125,9 +132,19 @@ def generate(
 
     return image_paths, seed, download_links, get_image_gallery()
 
-examples = [
-    "A bustling cityscape at sunset, where sleek, translucent skyscrapers shimmer with holographic displays. Gentle AI-powered drones weave through the air, delivering packages and tending to floating gardens. In the foreground, a diverse group of humans and lifelike androids share a laugh at a streetside café, their animated conversation bringing warmth to the scene.",
-    # ... (other examples remain unchanged)
+example_sets = [
+    [
+        f"{get_random_style()} painting of a majestic lighthouse on a rocky coast. Use bold brushstrokes and a vibrant color palette to capture the interplay of light and shadow as the lighthouse beam cuts through a stormy night sky.",
+        f"{get_random_style()} still life featuring a pair of vintage eyeglasses. Focus on the intricate details of the frames and lenses, using a warm color scheme to evoke a sense of nostalgia and wisdom.",
+        f"{get_random_style()} depiction of a rustic wooden stool in a sunlit artist's studio. Emphasize the texture of the wood and the interplay of light and shadow, using a mix of earthy tones and highlights.",
+        f"{get_random_style()} scene viewed through an ornate window frame. Contrast the intricate details of the window with a dreamy, soft-focus landscape beyond, using a palette that transitions from cool interior tones to warm exterior hues.",
+        f"{get_random_style()} close-up study of interlaced fingers. Use a monochromatic color scheme to emphasize the form and texture of the hands, with dramatic lighting to create depth and emotion.",
+        f"{get_random_style()} composition featuring a set of dice in motion. Capture the energy and randomness of the throw, using a dynamic color palette and blurred lines to convey movement.",
+        f"{get_random_style()} interpretation of heaven. Create an ethereal atmosphere with soft, billowing clouds and radiant light, using a palette of celestial blues, golds, and whites.",
+        f"{get_random_style()} portrayal of an ancient, mystical gate. Combine architectural details with elements of fantasy, using a rich, jewel-toned palette to create an air of mystery and magic.",
+        f"{get_random_style()} portrait of a curious cat. Focus on capturing the feline's expressive eyes and sleek form, using a mix of bold and subtle colors to bring out the cat's personality.",
+        f"{get_random_style()} abstract representation of toes in sand. Use textured brushstrokes to convey the feeling of warm sand, with a palette inspired by a sun-drenched beach."
+    ] for _ in range(3)
 ]
 
 css = '''
@@ -205,14 +222,36 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
     
     selected_image = gr.State(None)
 
-    gr.Examples(
-        examples=examples,
+    example_set_dropdown = gr.Dropdown(
+        choices=["Set 1", "Set 2", "Set 3"],
+        label="Select Example Set",
+        value="Set 1"
+    )
+
+    def update_examples(choice):
+        set_index = int(choice.split()[-1]) - 1
+        return gr.Examples(
+            examples=example_sets[set_index],
+            inputs=prompt,
+            outputs=[result, seed],
+            fn=generate,
+            cache_examples=False,
+        )
+
+    examples_component = gr.Examples(
+        examples=example_sets[0],
         inputs=prompt,
         outputs=[result, seed],
         fn=generate,
         cache_examples=False,
     )
 
+    example_set_dropdown.change(
+        fn=update_examples,
+        inputs=[example_set_dropdown],
+        outputs=[examples_component]
+    )
+
     use_negative_prompt.change(
         fn=lambda x: gr.update(visible=x),
         inputs=use_negative_prompt,