Update app.py
Browse fileschanging prompt box into dropdown
app.py
CHANGED
|
@@ -517,11 +517,20 @@ with gr.Blocks(css=css) as demo:
|
|
| 517 |
|
| 518 |
# with gr.Column(visible=False) as clear_button:
|
| 519 |
# remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm")
|
| 520 |
-
prompt = gr.Textbox(
|
| 521 |
-
|
| 522 |
-
|
| 523 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 524 |
)
|
|
|
|
| 525 |
gender_selector = gr.Radio(
|
| 526 |
label="Choose Gender",
|
| 527 |
choices=["woman", "man"],
|
|
@@ -537,16 +546,24 @@ with gr.Blocks(css=css) as demo:
|
|
| 537 |
gr.Markdown("### Suggested Prompts")
|
| 538 |
|
| 539 |
# theme_buttons = [gr.Button(theme, visible=False) for theme in man_prompts.keys()]
|
| 540 |
-
for theme in man_prompts.keys():
|
| 541 |
-
|
| 542 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 543 |
btn.click(
|
| 544 |
-
fn=lambda gender, t=theme: (get_prompts_by_gender(gender)[t]
|
| 545 |
inputs=gender_selector,
|
| 546 |
-
outputs=
|
| 547 |
)
|
| 548 |
|
| 549 |
|
|
|
|
| 550 |
# theme_buttons.append(gr.Button(theme))
|
| 551 |
# prompt_buttons = []
|
| 552 |
# for s in suggestions:
|
|
@@ -583,11 +600,18 @@ with gr.Blocks(css=css) as demo:
|
|
| 583 |
# # ✅ return prompt first, then button updates (match outputs length)
|
| 584 |
# return [new_prompt] + button_updates
|
| 585 |
|
|
|
|
|
|
|
|
|
|
| 586 |
|
| 587 |
gender_selector.change(
|
| 588 |
-
fn=update_prompt_on_gender_change,
|
| 589 |
-
inputs=[gender_selector, selected_theme, prompt],
|
| 590 |
-
outputs=[prompt] + theme_buttons
|
|
|
|
|
|
|
|
|
|
|
|
|
| 591 |
)
|
| 592 |
|
| 593 |
|
|
|
|
| 517 |
|
| 518 |
# with gr.Column(visible=False) as clear_button:
|
| 519 |
# remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm")
|
| 520 |
+
# prompt = gr.Textbox(
|
| 521 |
+
# label="Prompt",
|
| 522 |
+
# info="Try something like 'a photo of a man/woman img', 'img' is the trigger word.",
|
| 523 |
+
# placeholder="A photo of a [man/woman img]..."
|
| 524 |
+
# interactive=False
|
| 525 |
+
# )
|
| 526 |
+
|
| 527 |
+
prompt = gr.Dropdown(
|
| 528 |
+
label="Select Prompt",
|
| 529 |
+
choices=list(get_prompts_by_gender("woman").values()), # default to woman
|
| 530 |
+
value=list(get_prompts_by_gender("woman").values())[0],
|
| 531 |
+
interactive=True
|
| 532 |
)
|
| 533 |
+
|
| 534 |
gender_selector = gr.Radio(
|
| 535 |
label="Choose Gender",
|
| 536 |
choices=["woman", "man"],
|
|
|
|
| 546 |
gr.Markdown("### Suggested Prompts")
|
| 547 |
|
| 548 |
# theme_buttons = [gr.Button(theme, visible=False) for theme in man_prompts.keys()]
|
| 549 |
+
# for theme in man_prompts.keys():
|
| 550 |
+
# btn = gr.Button(theme, visible=False)
|
| 551 |
+
# theme_buttons.append(btn)
|
| 552 |
+
# btn.click(
|
| 553 |
+
# fn=lambda gender, t=theme: (get_prompts_by_gender(gender)[t], t, gr.update(visible=False)),
|
| 554 |
+
# inputs=gender_selector,
|
| 555 |
+
# outputs=[prompt, selected_theme, prompt_suggestions]
|
| 556 |
+
# )
|
| 557 |
+
|
| 558 |
+
for theme, btn in zip(man_prompts.keys(), theme_buttons):
|
| 559 |
btn.click(
|
| 560 |
+
fn=lambda gender, t=theme: gr.update(value=get_prompts_by_gender(gender)[t]),
|
| 561 |
inputs=gender_selector,
|
| 562 |
+
outputs=prompt
|
| 563 |
)
|
| 564 |
|
| 565 |
|
| 566 |
+
|
| 567 |
# theme_buttons.append(gr.Button(theme))
|
| 568 |
# prompt_buttons = []
|
| 569 |
# for s in suggestions:
|
|
|
|
| 600 |
# # ✅ return prompt first, then button updates (match outputs length)
|
| 601 |
# return [new_prompt] + button_updates
|
| 602 |
|
| 603 |
+
def update_prompt_dropdown(gender):
|
| 604 |
+
prompts = list(get_prompts_by_gender(gender).values())
|
| 605 |
+
return gr.update(choices=prompts, value=prompts[0]) # sets default to first
|
| 606 |
|
| 607 |
gender_selector.change(
|
| 608 |
+
# fn=update_prompt_on_gender_change,
|
| 609 |
+
# inputs=[gender_selector, selected_theme, prompt],
|
| 610 |
+
# outputs=[prompt] + theme_buttons
|
| 611 |
+
|
| 612 |
+
fn=update_prompt_dropdown,
|
| 613 |
+
inputs=gender_selector,
|
| 614 |
+
outputs=prompt
|
| 615 |
)
|
| 616 |
|
| 617 |
|