Update app.py

app.py

@@ -278,13 +278,12 @@ def update_theme_buttons(gender):
     updates = []
     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]
-    )
+    for i, theme in enumerate(themes):
+        if i < len(theme_buttons):
+            updates.append(gr.update(value=theme, visible=True))
+    # hide extra buttons if any
+    for j in range(len(themes), len(theme_buttons)):
+        updates.append(gr.update(visible=False))
     return updates
 
 
@@ -453,7 +452,7 @@ def update_prompt_buttons_visibility(gender):
     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())]
+theme_btns = []
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown(logo)
@@ -498,9 +497,9 @@ 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 = [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 = [gr.Button(theme, visible=False) for theme in man_prompts.keys()]
+                for theme in man_prompts.keys():  # initially load man themes
+                   btn = gr.Button(theme, visible=False)
                    theme_buttons.append(btn)
                 #     theme_buttons.append(gr.Button(theme))
                 # prompt_buttons = []