Update app.py

app.py

@@ -273,26 +273,28 @@ 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 update_theme_buttons(gender):
-    prompts = man_prompts if gender == "man" else woman_prompts
+    prompts = get_prompts_by_gender(gender)
     updates = []
-    for theme in prompts.keys():
-        updates.append(gr.update(value=theme, visible=True))
+    themes = list(prompts.keys())
+
+    for i, btn in enumerate(theme_buttons):
+        theme = list(man_prompts.keys())[i]
+    btn.click(
+        fn=make_click_fn(theme),
+        inputs=gender_selector,
+        outputs=[prompt, prompt_suggestions]
+    )
     return updates
 
+
+# def update_theme_buttons(gender):
+#     prompts = man_prompts if gender == "man" else woman_prompts
+#     updates = []
+#     for theme in prompts.keys():
+#         updates.append(gr.update(value=theme, visible=True))
+#     return updates
+
 def get_prompt_from_theme(gender, theme):
     return get_prompts_by_gender(gender)[theme], gr.update(visible=False)
     
@@ -301,6 +303,16 @@ def update_suggestion_buttons(gender):
     prompts = get_prompts_by_gender(gender)
     return gr.update(visible=True), prompts 
 
+def make_click_fn(theme_name):
+    return lambda gender: get_prompt_from_theme(gender, theme_name)
+
+for i, btn in enumerate(theme_buttons):
+    theme = list(man_prompts.keys())[i]
+    btn.click(
+        fn=make_click_fn(theme),
+        inputs=gender_selector,
+        outputs=[prompt, prompt_suggestions]
+    )
     
 
 ### Description and style
@@ -486,7 +498,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Group(visible=False, elem_id="prompt_suggestions_box") as prompt_suggestions:
                 gr.Markdown("### Suggested Prompts")
                  
-                theme_buttons = []
+                theme_buttons = [gr.Button(theme, visible=False) for theme in man_prompts.keys()]
                 for theme in list(man_prompts.keys()):  # initially load man themes
                    btn = gr.Button(theme)
                    theme_buttons.append(btn)
@@ -517,9 +529,9 @@ with gr.Blocks(css=css) as demo:
             #     return updates    
             
             gender_selector.change(
-                fn=lambda gender: [gr.update(visible=True) for _ in theme_btns],
+                fn=update_theme_buttons,
                 inputs=gender_selector,
-                outputs=update_theme_buttons
+                outputs=theme_buttons
             )
             
             show_prompts_btn.click(
@@ -657,10 +669,10 @@ with gr.Blocks(css=css) as demo:
         )
 
         # Show the suggestions popup
-        # show_prompts_btn.click(
-            # lambda: gr.update(visible=True),
-            # outputs=prompt_suggestions
-        # )
+        show_prompts_btn.click(
+            fn=lambda: gr.update(visible=True),
+            outputs=prompt_suggestions
+        )
         
         # Close the popup
         # close_btn.click(