Update app.py

testing dropdown

app.py

@@ -83,8 +83,8 @@ pipe.to(device)
 
 @spaces.GPU()
 def generate_image(
-    prompt_name,
-    gender,
+    prompt_theme, #added
+    gender, #added
     upload_images, 
     prompt, 
     negative_prompt, 
@@ -119,8 +119,10 @@ def generate_image(
     # prompt_dict = get_prompts_by_gender(gender)
 
     # added
-    actual_prompt = full_prompts[prompt]
+    # actual_prompt = full_prompts[prompt]
     # prompt = actual_prompt
+
+    full_prompt = current_prompts[prompt_theme]
     
 
     # check the trigger word
@@ -171,7 +173,8 @@ def generate_image(
     print(start_merge_step)
     images = pipe(
         # prompt=prompt,
-        prompt=actual_prompt,
+        # prompt=actual_prompt,
+        prompt=full_prompt,
         width=output_w,
         height=output_h,
         input_id_images=input_id_images,
@@ -507,15 +510,15 @@ with gr.Blocks(css=css) as demo:
 
             default_gender = "woman"
             # 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())
+            # default_prompts = get_prompts_by_gender(default_gender)
+            # default_prompt_keys = list(default_prompts.keys())
 
-            full_prompts = gr.State(default_prompts)
+            curren_prompts = gr.State(default_prompts)
             
-            prompt = gr.Dropdown(
+            prompt_dropdown = gr.Dropdown(
                 label="Select Prompt",
-                choices=default_prompt_keys,
-                value=default_prompt_keys[0],
+                choices=list(default_prompts.keys()),
+                value=list(default_prompts.keys())[0],
                 interactive=True
             )
            
@@ -532,23 +535,31 @@ with gr.Blocks(css=css) as demo:
                 pass
             
            
-            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 = get_prompts_by_gender(gender)
+            # 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 = get_prompts_by_gender(gender)
+            #     return [
+            #         gr.update(choices=list(prompts.keys()), value=list(prompts.keys())[0]),
+            #         prompts  # Update the stored full prompts
+            #     ]
+
+
+            def update_prompts_on_gender_change(gender):
+                prompts = woman_prompts if gender == "woman" else man_prompts
                 return [
-                    gr.update(choices=list(prompts.keys()), value=list(prompts.keys())[0]),
-                    prompts  # Update the stored full prompts
-                ]
+                    gr.update(choices=list(prompts.keys()), 
+                    prompts  # Update the stored prompts
+    ]
             
             gender_selector.change(
                 # fn=update_prompt_on_gender_change,
                 # inputs=[gender_selector, selected_theme, prompt],
                 # outputs=[prompt] + theme_buttons
 
-                fn=update_prompt_dropdown,
+                fn=update_prompts_on_gender_change,
                 inputs=gender_selector,
-                outputs=[prompt, full_prompts]
+                outputs=[prompt_dropdown, current_prompts]
             )
 
           
@@ -644,10 +655,10 @@ with gr.Blocks(css=css) as demo:
         enable_doodle.select(fn=change_doodle_space, inputs=enable_doodle, outputs=doodle_space)
 
         input_list = [
-            prompt, #added
+            prompt_dropdown, #added
             gender_selector, #added
             files, 
-            prompt, 
+            # prompt, 
             negative_prompt, 
             aspect_ratio, 
             style, 
@@ -660,7 +671,7 @@ with gr.Blocks(css=css) as demo:
             sketch_image,
             adapter_conditioning_scale,
             adapter_conditioning_factor,
-            full_prompts #added
+            current_prompts #added
         ]
 
         submit.click(