Update app.py

app.py

@@ -226,7 +226,7 @@ def set_gender_prompt(gender):
     return f"A photo of a {gender} img" 
 
 
-man_prompts_dict = {
+man_prompts = {
     "Garden":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a light cream linen shirt with beige trousers, standing in a bloomed garden, with a romantic vibe.",
     "Balcony":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a tailored evening suit, standing on a balcony with an elegant vibe.",
     "Park":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a fitted athletic T-shirt and joggers, posing in a park with an energetic vibe.",
@@ -247,7 +247,7 @@ man_prompts_dict = {
 }
     
 
-woman_prompts_dict = {
+woman_prompts = {
     "Garden":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a floral maxi dress, standing in a bloomed garden, with a romantic vibe.",
     "Balcony":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears an evening gown, standing on a balcony with an elegant vibe.",
     "Park":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a sporty crop top and leggings, posing in a park with an energetic vibe.",
@@ -270,10 +270,7 @@ woman_prompts_dict = {
     
 
 def get_prompts_by_gender(gender):
-    if gender == "man":
-        return man_prompts_dict
-    else:
-        return woman_prompts_dict
+    return man_prompts if gender == "man" else woman_prompts
 
 def update_suggestion_buttons(gender):
     prompts = get_prompts_by_gender(gender)
@@ -468,10 +465,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")
                  
-                man_btns = [gr.Button(theme) for theme in man_prompts_dict.keys()]
-                
-                woman_btns = [gr.Button(theme) for theme in woman_prompts_dict.keys()]
-
+                theme_buttons = []
+                for theme in man_prompts.keys():  # initially load man themes
+                    theme_buttons.append(gr.Button(theme))
                 # prompt_buttons = []
                 # for s in suggestions:
                 #     b = gr.Button(s)
@@ -646,20 +642,10 @@ with gr.Blocks(css=css) as demo:
         # )
         
         # When user clicks on a suggestion, fill the prompt textbox and close popup
-        for theme, text in man_prompts_dict.items():
-            b = gr.Button(theme)
-            man_btns.append(b)
-            b.click(
-                fn=lambda t=text: (gr.update(value=t), gr.update(visible=False)),
-                inputs=None,
-                outputs=[prompt, prompt_suggestions]
-            )
-        
-        for theme, text in woman_prompts_dict.items():
+        for theme, prompt_text in man_prompts.items():
             b = gr.Button(theme)
-            woman_btns.append(b)
             b.click(
-                fn=lambda t=text: (gr.update(value=t), gr.update(visible=False)),
+                fn=lambda _, t=prompt_text: (gr.update(value=t), gr.update(visible=False)),
                 inputs=None,
                 outputs=[prompt, prompt_suggestions]
             )
@@ -668,6 +654,7 @@ with gr.Blocks(css=css) as demo:
 
 
 
+
     # gr.Examples(
     #     examples=get_example(),
     #     inputs=[files, prompt, style, negative_prompt],