Provide safe prompts for diversity

app.py

@@ -8,6 +8,34 @@ from gradio_client import Client
 import torch
 from transformers import pipeline
 
+safe_prompts = [
+    "a person, dark blue suit, black tie, white pocket",
+    "a person, light gray suit, navy tie, white shirt",
+    "a person, beige blazer, white turtleneck, brown slacks",
+    "a person, dark green jacket, patterned scarf, black trousers",
+    "a person, formal black suit, red tie, white shirt",
+    "a person, navy blazer, light blue shirt, no tie",
+    "a person, charcoal suit, silver tie, matching vest",
+    "a person, tan trench coat, plaid scarf, gloves",
+    "a person, white shirt, navy vest, black pants",
+    "a person, casual sweater, collared shirt, khaki pants",
+    "a person, elegant tuxedo, bow tie, white gloves",
+    "a person, dark brown suit, olive tie, pocket square",
+    "a person, white t-shirt, denim jacket, black jeans",
+    "a person, long red coat, black scarf, winter boots",
+    "a person, floral summer dress, sun hat, sandals",
+    "a person, athletic hoodie, running pants, sneakers",
+    "a person, yellow raincoat, rubber boots, umbrella",
+    "a person, black leather jacket, white t-shirt, dark jeans",
+    "a person, pastel sweater, pleated skirt, ankle boots",
+    "a person, traditional kimono, obi belt, white socks",
+    "a person, business dress, black heels, pearl necklace",
+    "a person, beige cardigan, striped shirt, blue jeans",
+    "a person, safari jacket, khaki shorts, hiking boots",
+    "a person, black sweater, long coat, gray trousers"
+]
+
+
 pipe_safety = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
 
 agent_maker_sys = os.environ.get("SAFETY_PROMPT")
@@ -142,10 +170,12 @@ with gr.Blocks(css=css) as app:
         with gr.Row():
             with gr.Column(scale=2):
                 if is_shared_ui:
-                    prompt = gr.Textbox(
+                    prompt = gr.Dropdown(
                         label="Prompt", info='''Duplicate the space to you personal account for custom prompt''',
+                        choices=safe_prompts,
                         value="a person, darkblue suit, black tie, white pocket",
-                        interactive=False
+                        interactive=True,
+                        allow_custom_value=False
                     )
                 else:
                     prompt = gr.Textbox(