Update app.py

app.py

@@ -470,6 +470,9 @@ with gr.Blocks(css=css) as demo:
     gr.HTML(title_html)
     #gr.Markdown(description)
 
+    selected_theme = gr.State("")
+
+
     # current_suggestions = gr.State(woman_prompts)
     
     # gr.DuplicateButton(
@@ -510,9 +513,15 @@ with gr.Blocks(css=css) as demo:
                     gr.Markdown("### Suggested Prompts")
 
                 # 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)
+                    for theme in man_prompts.keys():
+                        btn = gr.Button(theme, visible=False)
+                        theme_buttons.append(btn)
+                        btn.click(
+                            fn=lambda gender, t=theme: (get_prompts_by_gender(gender)[t], t, gr.update(visible=False)),
+                            inputs=gender_selector,
+                            outputs=[prompt, selected_theme, prompt_suggestions]
+                        )
+
                 #     theme_buttons.append(gr.Button(theme))
                 # prompt_buttons = []
                 # for s in suggestions:
@@ -539,11 +548,20 @@ with gr.Blocks(css=css) as demo:
             #     updates[0] = gr.update(visible=True)    # force popup visible
             #     return updates    
             
+            def update_prompt_on_gender_change(gender, theme_name, current_prompt):
+                if theme_name and theme_name in get_prompts_by_gender(gender):
+                    # regenerate the prompt for the same theme but with new gender
+                    new_prompt = get_prompts_by_gender(gender)[theme_name]
+                    return new_prompt, update_theme_buttons_only(gender)
+                # if no theme selected, keep existing prompt
+                return current_prompt, update_theme_buttons_only(gender)
+            
             gender_selector.change(
-                fn=update_theme_buttons,
-                inputs=gender_selector,
-                outputs=theme_buttons
+                fn=update_prompt_on_gender_change,
+                inputs=[gender_selector, selected_theme, prompt],
+                outputs=[prompt] + theme_buttons
             )
+
             
             show_prompts_btn.click(
                 # fn=lambda: gr.update(visible=True),