Update app.py

testing dropdown

app.py

@@ -116,11 +116,11 @@ def generate_image(
 
     
     #added
-    prompt_dict = get_prompts_by_gender(gender)
+    # prompt_dict = get_prompts_by_gender(gender)
 
     # added
-    actual_prompt = prompt_dict.get(prompt_name, "")
-    prompt = actual_prompt
+    actual_prompt = full_prompts[prompt]
+    # prompt = actual_prompt
     
 
     # check the trigger word
@@ -170,7 +170,8 @@ def generate_image(
         start_merge_step = 30
     print(start_merge_step)
     images = pipe(
-        prompt=prompt,
+        # prompt=prompt,
+        prompt=actual_prompt,
         width=output_w,
         height=output_h,
         input_id_images=input_id_images,
@@ -505,7 +506,11 @@ with gr.Blocks(css=css) as demo:
             # )
 
             default_gender = "woman"
-            default_prompt_keys = list(get_prompts_by_gender(default_gender).keys())
+            # default_prompt_keys = list(get_prompts_by_gender(default_gender).keys())
+            default_prompts = get_prompts_by_gender(default_gender)
+            default_prompt_keys = list(default_prompts.keys())
+
+            full_prompts = gr.State(default_prompts)
             
             prompt = gr.Dropdown(
                 label="Select Prompt",
@@ -528,8 +533,13 @@ with gr.Blocks(css=css) as demo:
             
            
             def update_prompt_dropdown(gender):
-                prompts = list(get_prompts_by_gender(gender).values())
-                return gr.update(choices=prompts, value=prompts[0])  # sets default to first
+                # prompts = list(get_prompts_by_gender(gender).values())
+                # return gr.update(choices=prompts, value=prompts[0])  # sets default to first
+                prompts = get_prompts_by_gender(gender)
+                return [
+                    gr.update(choices=list(prompts.keys()), value=list(prompts.keys())[0]),
+                    prompts  # Update the stored full prompts
+                ]
             
             gender_selector.change(
                 # fn=update_prompt_on_gender_change,
@@ -538,7 +548,7 @@ with gr.Blocks(css=css) as demo:
 
                 fn=update_prompt_dropdown,
                 inputs=gender_selector,
-                outputs=prompt
+                outputs=[prompt, full_prompts]
             )
 
           
@@ -634,6 +644,8 @@ with gr.Blocks(css=css) as demo:
         enable_doodle.select(fn=change_doodle_space, inputs=enable_doodle, outputs=doodle_space)
 
         input_list = [
+            prompt, #added
+            gender_selector, #added
             files, 
             prompt, 
             negative_prompt, 
@@ -648,6 +660,7 @@ with gr.Blocks(css=css) as demo:
             sketch_image,
             adapter_conditioning_scale,
             adapter_conditioning_factor
+            full_prompts #added
         ]
 
         submit.click(