Significant commited on
Commit
e4cfbdf
·
verified ·
1 Parent(s): 3a57694

Update app.py

Browse files

changing prompt box into dropdown

Files changed (1) hide show
  1. app.py +36 -12
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
- 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
  )
 
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
- btn = gr.Button(theme, visible=False)
542
- theme_buttons.append(btn)
 
 
 
 
 
 
 
543
  btn.click(
544
- fn=lambda gender, t=theme: (get_prompts_by_gender(gender)[t], t, gr.update(visible=False)),
545
  inputs=gender_selector,
546
- outputs=[prompt, selected_theme, prompt_suggestions]
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