Update app.py

app.py

@@ -272,9 +272,29 @@ woman_prompts = {
 def get_prompts_by_gender(gender):
     return man_prompts if gender == "man" else woman_prompts
 
+# ✅ Place this here, above gr.Blocks
+def update_theme_buttons(gender):
+    prompts = get_prompts_by_gender(gender)  # returns {theme: prompt}
+    updates = []
+    for i, (theme, prompt_text) in enumerate(prompts.items()):
+        if i < len(theme_btns):
+            updates.append(gr.update(value=theme, visible=True))
+        else:
+            break
+    # Hide remaining unused buttons
+    for j in range(len(prompts), len(theme_btns)):
+        updates.append(gr.update(visible=False))
+    return updates   
+
+def get_prompt_from_theme(gender, theme):
+    return get_prompts_by_gender(gender)[theme], gr.update(visible=False)
+    
+
 def update_suggestion_buttons(gender):
     prompts = get_prompts_by_gender(gender)
-    return gr.update(visible=True), prompts        
+    return gr.update(visible=True), prompts 
+
+    
 
 ### Description and style
 logo = r"""
@@ -411,23 +431,17 @@ input[type="radio"] {
 '''
 
 def update_prompt_buttons_visibility(gender):
-    if gender == "man":
-        return [gr.update(visible=True)] + \
-               [gr.update(visible=True) for _ in man_btns] + \
-               [gr.update(visible=False) for _ in woman_btns]
-    else:
-        return [gr.update(visible=True)] + \
-               [gr.update(visible=False) for _ in man_btns] + \
-               [gr.update(visible=True) for _ in woman_btns]
-
+    prompts = man_prompts if gender == "man" else woman_prompts
+    return [gr.update(value=theme, visible=True) for theme in prompts.keys()]
 
+# theme_btns = [gr.Button(theme, visible=False) for theme in list(man_prompts.keys())]
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown(logo)
     gr.Markdown(title)
     #gr.Markdown(description)
 
-    current_suggestions = gr.State(woman_prompts)
+    # current_suggestions = gr.State(woman_prompts)
     
     # gr.DuplicateButton(
     #     value="Duplicate Space for private use ",
@@ -466,8 +480,10 @@ with gr.Blocks(css=css) as demo:
                 gr.Markdown("### Suggested Prompts")
                  
                 theme_buttons = []
-                for theme in man_prompts.keys():  # initially load man themes
-                    theme_buttons.append(gr.Button(theme))
+                for theme in list(man_prompts.keys()):  # initially load man themes
+                   btn = gr.Button(theme)
+                   theme_btns.append(btn)
+                #     theme_buttons.append(gr.Button(theme))
                 # prompt_buttons = []
                 # for s in suggestions:
                 #     b = gr.Button(s)
@@ -482,27 +498,28 @@ with gr.Blocks(css=css) as demo:
                 outputs=prompt_suggestions
             )
             
-            def update_buttons_hidden(gender):
-                updates = update_prompt_buttons_visibility(gender)
-                # Force popup to remain hidden
-                updates[0] = gr.update(visible=False)
-                return updates
-
-            def update_buttons_and_show(gender):
-                updates = update_prompt_buttons_visibility(gender)
-                updates[0] = gr.update(visible=True)    # force popup visible
-                return updates    
+            # def update_buttons_hidden(gender):
+            #     updates = update_prompt_buttons_visibility(gender)
+            #     # Force popup to remain hidden
+            #     updates[0] = gr.update(visible=False)
+            #     return updates
+
+            # def update_buttons_and_show(gender):
+            #     updates = update_prompt_buttons_visibility(gender)
+            #     updates[0] = gr.update(visible=True)    # force popup visible
+            #     return updates    
             
             gender_selector.change(
-                fn=update_buttons_hidden,
+                fn=lambda gender: [gr.update(visible=True) for _ in theme_btns],
                 inputs=gender_selector,
-                outputs=[prompt_suggestions] + man_btns + woman_btns
+                outputs=theme_btns
             )
             
             show_prompts_btn.click(
-                fn=update_buttons_and_show,
-                inputs=gender_selector,
-                outputs=[prompt_suggestions] + man_btns + woman_btns
+                fn=lambda: gr.update(visible=True),
+                # fn=update_theme_buttons,
+                # inputs=gender_selector,
+                outputs=theme_buttons
             )
 
             
@@ -642,19 +659,13 @@ with gr.Blocks(css=css) as demo:
         # )
         
         # When user clicks on a suggestion, fill the prompt textbox and close popup
-        for theme, prompt_text in man_prompts.items():
-            b = gr.Button(theme)
-            b.click(
-                fn=lambda _, t=prompt_text: (gr.update(value=t), gr.update(visible=False)),
-                inputs=None,
+        for btn in theme_btns:
+            btn.click(
+                fn=lambda theme_name, g: get_prompt_from_theme(g, theme_name),
+                inputs=[btn, gender_selector],
                 outputs=[prompt, prompt_suggestions]
             )
 
-
-
-
-
-
     # gr.Examples(
     #     examples=get_example(),
     #     inputs=[files, prompt, style, negative_prompt],