Update app.py

app.py

@@ -11,7 +11,7 @@ dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
 prompt_df = dataset["train"].to_pandas()
 
 def get_samples():
-    prompt_list = prompt_df.sample(n = 5)["Prompt"].map(lambda x: [x]).values.tolist()
+    prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
     return prompt_list
 
 def get_params(request: gr.Request):
@@ -27,6 +27,7 @@ def get_params(request: gr.Request):
 client = InferenceClient()
 models = client.list_deployed_models()
 list_models = models["text-to-image"]
+list_prompts = get_samples()
 
 '''
 list_models = [
@@ -227,10 +228,14 @@ with gr.Blocks(css=css) as demo:
             current_model = gr.Dropdown(label="Current Model", choices=list_models, value="stabilityai/stable-diffusion-2-1")
         
     with gr.Row(elem_id="prompt-container"):
-        text_prompt = gr.Textbox(label="Prompt", placeholder="a cute dog", 
-                                 value = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt",
-                                 lines=2, elem_id="prompt-text-input")
-        text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
+        with gr.Row():
+            text_prompt = gr.Textbox(label="Prompt", placeholder="a cute dog", 
+                                     value = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt",
+                                     lines=2, elem_id="prompt-text-input")
+            text_button = gr.Button("Manualy input Generate", variant='primary', elem_id="gen-button")
+        with gr.Row():
+            select_prompt = gr.Dropdown(label="prompt selected", choices=list_prompts, value=list_prompts[1])
+            select_button = gr.Button("Select Prompt Generate", variant='primary', elem_id="gen-button")
         
     with gr.Row():
         image_output = gr.Image(type="pil", label="Output Image", elem_id="gallery")
@@ -239,6 +244,7 @@ with gr.Blocks(css=css) as demo:
         negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1, elem_id="negative-prompt-text-input")
         image_style = gr.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="Portrait", allow_custom_value=False)
 
+    '''
     with gr.Row():
         with gr.Column():
             exps = gr.Examples(
@@ -247,18 +253,11 @@ with gr.Blocks(css=css) as demo:
                 label = "Prompt Examples",
                 elem_id = "Examples"
             )
-            '''
-            exps_l = get_samples()
-            exps = gr.Dataset(components=[text_prompt],
-                        label="Prompt Examples",
-                        samples=exps_l,
-                    )
-            #exp_refresh = gr.Button(value="Click Refresh to get newly prompt examples")
-            #exp_refresh.click(get_samples, None, exps)
-            exps.select(lambda x: x[0], exps, text_prompt)
-            '''
+    '''
             
     text_button.click(generate_txt2img, inputs=[current_model, text_prompt, negative_prompt, image_style], outputs=image_output)
+    select_button.click(generate_txt2img, inputs=[current_model, select_prompt, negative_prompt, image_style], outputs=image_output)
+    
     demo.load(get_params, None, current_model)
 
 demo.launch(show_api=False)