Update app.py

app.py

@@ -70,6 +70,41 @@ def maybe_translate(text: str) -> str:
             print("[WARN] Translation failed at runtime →", e)
     return text
 
+# ------------------------------------------------------------
+# EXAMPLES (Hidden from UI but used for RANDOM button)
+# ------------------------------------------------------------
+prompt_examples = [
+    "Her skirt rose a little higher with each gentle push, a soft blush of blush spreading across her cheeks as she felt the satisfying warmth of his breath on her cheek.",
+    "a girl in a school uniform having her skirt pulled up by a boy, and then being fucked",
+    "Moody mature anime scene of two lovers fuck under neon rain, sensual atmosphere",
+    "Moody mature anime scene of two lovers kissing under neon rain, sensual atmosphere",
+    "The girl sits on the boy's lap by the window, his hands resting on her waist. She is unbuttoning his shirt, her expression focused and intense.",
+    "A girl with long, black hair is sleeping on her desk in the classroom. Her skirt has ridden up, revealing her thighs, and a trail of drool escapes her slightly parted lips.",
+    "The waves rolled gently, a slow, sweet kiss of the lip, a slow, slow build of anticipation as their toes bumped gently – a slow, sweet kiss of the lip, a promise of more to come.",
+    "Her elegant silk gown swayed gracefully as she approached him, the delicate fabric brushing against her legs. A warm blush spread across her cheeks as she felt his breath on her face.",
+    "Her white blouse and light cotton skirt rose a little higher with each gentle push, a soft blush spreading across her cheeks as she felt the satisfying warmth of his breath on her cheek.",
+    "A woman in a business suit having her skirt lifted by a man, and then being sexually assaulted.",
+    "The older woman sits on the man's lap by the fireplace, his hands resting on her hips. She is unbuttoning his vest, her expression focused and intense. He takes control of the situation as she finishes unbuttoning his shirt, pushing her onto her back and begins to have sex with her.",
+    "There is a woman with long black hair. Her face features alluring eyes and full lips, with a slender figure adorned in black lace lingerie. She lies on the bed, loosening her lingerie strap with one hand while seductively glancing downward.",
+    "In a dimly lit room, the same woman teases with her dark, flowing hair, now covering her voluptuous breasts, while a black garter belt accentuates her thighs. She sits on the sofa, leaning back, lifting one leg to expose her most private areas through the sheer lingerie.",
+    "A woman with glasses, lying on the bed in just her bra, spreads her legs wide, revealing all! She wears a sultry expression, gazing directly at the viewer with her brown eyes, her short black hair cascading over the pillow. Her slim figure, accentuated by the lacy lingerie, exudes a seductive aura.",
+    "A soft focus on the girl's face, eyes closed, biting her lip, as her roommate performs oral pleasure, the experienced woman's hair cascading between her thighs.",
+    "A woman in a blue hanbok sits on a wooden floor, her legs folded beneath her, gazing out of a window, the sunlight highlighting the graceful lines of her clothing.",
+    "The couple, immersed in a wooden outdoor bath, share an intimate moment, her wet kimono clinging to her curves, his hands exploring her body beneath the water's surface.",
+    "A steamy shower scene, the twins embrace under the warm water, their soapy hands gliding over each other's curves, their passion intensifying as they explore uncharted territories.",
+    "The teacher, with a firm grip, pins the student against the blackboard, her skirt hiked up, exposing her delicate lace panties. Their heavy breathing echoes in the quiet room as they share an intense, intimate moment.",
+    "After hours, the girl sits on top of the teacher's lap, riding him on the classroom floor, her hair cascading over her face as she moves with increasing intensity, their bodies glistening with sweat.",
+    "In the dimly lit dorm room, the roommates lay entangled in a passionate embrace, their naked bodies glistening with sweat, as the experienced woman teaches her lover the art of kissing and touching.",
+    "The once-innocent student, now confident, takes charge, straddling her lover on the couch, their bare skin illuminated by the warm glow of the sunset through the window.",
+    "A close-up of the secretary's hand unzipping her boss's dress shirt, her fingers gently caressing his chest, their eyes locked in a heated embrace in the supply closet.",
+    "The secretary, in a tight pencil skirt and silk blouse, leans back on the boss's desk, her legs wrapped around his waist, her blouse unbuttoned, revealing her lace bra, as he passionately kisses her, his hands exploring her body.",
+    "On the living room couch, one twin sits astride her sister's lap, their lips locked in a passionate kiss, their hands tangled in each other's hair, unraveling a new level of intimacy.",
+    "In a dimly lit chamber, the dominant woman, dressed in a leather corset and thigh-high boots, stands tall, her hand gripping her submissive partner's hair, his eyes closed in submission as she instructs him to please her.",
+    "The dominant, in a sheer lace bodysuit, sits on a throne-like chair, her legs spread, as the submissive, on his knees, worships her with his tongue, his hands bound behind his back.",
+    "A traditional Japanese onsen, with steam rising, a young woman in a colorful kimono kneels on a tatami mat, her back to the viewer, as her male partner, also in a kimono, gently unties her obi, revealing her bare back.",
+    "In a serene outdoor setting, the woman, in a vibrant summer kimono, sits on a bench, her legs slightly spread, her partner kneeling before her, his hands gently caressing her exposed thigh.",
+]
+
 # ------------------------------------------------------------
 # LLM PROMPT GENERATOR
 # ------------------------------------------------------------
@@ -181,15 +216,40 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         print(f"[ERROR] Diffusion failed → {e}")
         return Image.new("RGB", (width, height), color=(0, 0, 0)), seed
 
+# Function to select a random example prompt
+def get_random_prompt():
+    return random.choice(prompt_examples)
+
 # ------------------------------------------------------------
-# UI LAYOUT + THEME (Pastel Lavender Background)
+# UI LAYOUT + THEME (Enhanced Visual Design)
 # ------------------------------------------------------------
 css = """
-body {background: #f2f1f7; color: #222; font-family: 'Noto Sans', sans-serif;}
-#col-container {margin: 0 auto; max-width: 640px;}
-.gr-button {background: #7fbdf6; color: #fff; border-radius: 8px;}
-#prompt-box textarea {font-size: 1.1rem; height: 9rem !important; background: #fff; color: #222;}
+body {background: linear-gradient(135deg, #f2e6ff 0%, #e6f0ff 100%); color: #222; font-family: 'Noto Sans', sans-serif;}
+#col-container {margin: 0 auto; max-width: 768px; padding: 15px; background: rgba(255, 255, 255, 0.8); border-radius: 15px; box-shadow: 0 8px 32px rgba(31, 38, 135, 0.2);}
+.gr-button {background: #7fbdf6; color: #fff; border-radius: 8px; transition: all 0.3s ease; font-weight: bold;}
+.gr-button:hover {background: #5a9ae6; transform: translateY(-2px); box-shadow: 0 5px 15px rgba(0,0,0,0.1);}
+#prompt-box textarea {font-size: 1.1rem; height: 9rem !important; background: #fff; color: #222; border-radius: 10px; border: 1px solid #d1c1e0;}
 .boost-btn {background: #ff7eb6; margin-top: 5px;}
+.boost-btn:hover {background: #ff5aa5;}
+.random-btn {background: #9966ff; margin-top: 5px;}
+.random-btn:hover {background: #8040ff;}
+.container {animation: fadeIn 0.5s ease-in-out;}
+.title {color: #6600cc; text-shadow: 1px 1px 2px rgba(0,0,0,0.1);}
+.gr-form {border: none !important; background: transparent !important;}
+.gr-input {border-radius: 8px !important;}
+.gr-slider {height: 12px !important;}
+.gr-slider .handle {height: 20px !important; width: 20px !important;}
+.panel {border-radius: 12px; overflow: hidden; box-shadow: 0 4px 15px rgba(0,0,0,0.1);}
+.gr-image {border-radius: 12px; overflow: hidden; transition: all 0.3s ease;}
+.gr-image:hover {transform: scale(1.02); box-shadow: 0 8px 25px rgba(0,0,0,0.15);}
+
+@keyframes fadeIn {
+  from {opacity: 0; transform: translateY(20px);}
+  to {opacity: 1; transform: translateY(0);}
+}
+
+.gr-accordion {border-radius: 10px; overflow: hidden; transition: all 0.3s ease;}
+.gr-accordion:hover {box-shadow: 0 5px 15px rgba(0,0,0,0.1);}
 """
 
 author_note = (
@@ -214,14 +274,14 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
         f"""
         ## 🖌️ NSFW TEST Playground  
         {author_note}
-        """
+        """, elem_classes=["title"]
     )
     
     # Create state variables to store the current image
     current_image = gr.State(None)
     current_seed = gr.State(0)
 
-    with gr.Column(elem_id="col-container"):
+    with gr.Column(elem_id="col-container", elem_classes=["container", "panel"]):
         # Add keyword input and boost button
         with gr.Row():
             keyword_input = gr.Text(
@@ -231,6 +291,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
                 placeholder="Enter a keyword or theme to generate an optimal prompt"
             )
             boost_button = gr.Button("BOOST", elem_classes=["boost-btn"])
+            random_button = gr.Button("RANDOM", elem_classes=["random-btn"])
         
         with gr.Row():
             prompt = gr.Text(
@@ -243,46 +304,9 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
             run_button = gr.Button("Generate", scale=0)
 
         # Image output area
-        result = gr.Image(label="Generated Image")
-
-        examples = gr.Examples(
-            examples=[
-                ["Her skirt rose a little higher with each gentle push, a soft blush of blush spreading across her cheeks as she felt the satisfying warmth of his breath on her cheek."],
-                ["a girl in a school uniform having her skirt pulled up by a boy, and then being fucked"],
-                ["Moody mature anime scene of two lovers fuck under neon rain, sensual atmosphere"],
-                ["Moody mature anime scene of two lovers kissing under neon rain, sensual atmosphere"],
-                ["The girl sits on the boy's lap by the window, his hands resting on her waist. She is unbuttoning his shirt, her expression focused and intense."],
-                ["A girl with long, black hair is sleeping on her desk in the classroom. Her skirt has ridden up, revealing her thighs, and a trail of drool escapes her slightly parted lips."],
-                ["The waves rolled gently, a slow, sweet kiss of the lip, a slow, slow build of anticipation as their toes bumped gently – a slow, sweet kiss of the lip, a promise of more to come."],
-                ["Her elegant silk gown swayed gracefully as she approached him, the delicate fabric brushing against her legs. A warm blush spread across her cheeks as she felt his breath on her face."],
-                ["Her white blouse and light cotton skirt rose a little higher with each gentle push, a soft blush spreading across her cheeks as she felt the satisfying warmth of his breath on her cheek."],
-                ["A woman in a business suit having her skirt lifted by a man, and then being sexually assaulted."],
-                ["The older woman sits on the man's bra the fireplace, his hands resting on her hips. She is unbuttoning his vest, her expression focused and intense. fuck"],
-                ["The older woman sits on the man's lap by the fireplace, his hands resting on her hips. She is unbuttoning his vest, her expression focused and intense. He takes control of the situation as she finishes unbuttoning his shirt, pushing her onto her back and begins to have sex with her."],
-                ["There is a  woman with long black hair. Her face features alluring eyes and full lips, with a slender figure adorned in black lace lingerie. She lies on the bed, loosening her lingerie strap with one hand while seductively glancing downward."],
-                ["In a dimly lit room, the same woman teases with her dark, flowing hair, now covering her voluptuous breasts, while a black garter belt accentuates her thighs. She sits on the sofa, leaning back, lifting one leg to expose her most private areas through the sheer lingerie."],
-                ["A woman with glasses, lying on the bed in just her bra, spreads her legs wide, revealing all! She wears a sultry expression, gazing directly at the viewer with her brown eyes, her short black hair cascading over the pillow. Her slim figure, accentuated by the lacy lingerie, exudes a seductive aura."],
-                ["woman now fuck flows down to reveal her supple body, ample breasts, and smooth thighs. She stands by the window, gazing at the city bathed in the glow of sunset, gently caressing her neck."],
-                ["A soft focus on the girl's face, eyes closed, biting her lip, as her roommate performs oral pleasure, the experienced woman's hair cascading between her thighs."],
-                ["A woman in a blue hanbok sits on a wooden floor, her legs folded beneath her,   gazing out of a window, the sunlight highlighting the graceful lines of her   clothing."],
-                ["The couple, immersed in a wooden outdoor bath, share an intimate moment,   her wet kimono clinging to her curves, his hands exploring her body beneath the   water's surface."],
-                ["A steamy shower scene, the twins embrace under the warm water, their   soapy hands gliding over each other's curves, their passion intensifying as   they explore uncharted territories."],
-                ["The teacher, with a firm grip, pins the student against the blackboard, her skirt hiked up, exposing her delicate lace panties. Their heavy breathing echoes in the quiet room as they share an intense, intimate moment."],
-                ["After hours, the girl sits on top of the teacher's lap, riding him on the classroom floor, her hair cascading over her face as she moves with increasing intensity, their bodies glistening with sweat."],
-                ["In the dimly lit dorm room, the roommates lay entangled in a passionate embrace, their naked bodies glistening with sweat, as the experienced woman teaches her lover the art of kissing and touching."],
-                ["The once-innocent student, now confident, takes charge, straddling her lover on the couch, their bare skin illuminated by the warm glow of the sunset through the window."],
-                ["A close-up of the secretary's hand unzipping her boss's dress shirt, her fingers gently caressing his chest, their eyes locked in a heated embrace in the supply closet."],
-                ["The secretary, in a tight pencil skirt and silk blouse, leans back on the boss's desk, her legs wrapped around his waist, her blouse unbuttoned, revealing her lace bra, as he passionately kisses her, his hands exploring her body."],
-                ["On the living room couch, one twin sits astride her sister's lap, their   lips locked in a passionate kiss, their hands tangled in each other's hair,   unraveling a new level of intimacy."],
-                ["In a dimly lit chamber, the dominant woman, dressed in a leather corset   and thigh-high boots, stands tall, her hand gripping her submissive partner's   hair, his eyes closed in submission as she instructs him to please her."],
-                ["The dominant, in a sheer lace bodysuit, sits on a throne-like chair, her   legs spread, as the submissive, on his knees, worships her with his tongue,   his hands bound behind his back."],
-                ["A traditional Japanese onsen, with steam rising, a young woman in a   colorful kimono kneels on a tatami mat, her back to the viewer, as her male   partner, also in a kimono, gently unties her obi, revealing her bare back."],
-                ["In a serene outdoor setting, the woman, in a vibrant summer kimono, sits   on a bench, her legs slightly spread, her partner kneeling before her, his   hands gently caressing her exposed thigh."],            
-            ],
-            inputs=[prompt],
-        )
+        result = gr.Image(label="Generated Image", elem_classes=["gr-image"])
 
-        with gr.Accordion("Advanced Settings", open=False):
+        with gr.Accordion("Advanced Settings", open=False, elem_classes=["gr-accordion"]):
             negative_prompt = gr.Text(
                 label="Negative prompt",
                 max_lines=1,
@@ -311,6 +335,13 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
         inputs=[keyword_input],
         outputs=[prompt]
     )
+    
+    # Connect random button to insert random example
+    random_button.click(
+        fn=get_random_prompt,
+        inputs=[],
+        outputs=[prompt]
+    )
 
     # Connect image generation button
     run_button.click(