Update app.py

share button test

app.py

@@ -353,7 +353,7 @@ article = r""" """
 # <br>
 # If our work is useful for your research, please consider citing:
 
-#bibtex
+# bibtex
 # @article{li2023photomaker,
 #   title={PhotoMaker: Customizing Realistic Human Photos via Stacked ID Embedding},
 #   author={Li, Zhen and Cao, Mingdeng and Wang, Xintao and Qi, Zhongang and Cheng, Ming-Ming and Shan, Ying},
@@ -472,41 +472,19 @@ with gr.Blocks(css=css) as demo:
             with gr.Column(visible=False, elem_id="reupload_wrapper") as clear_button:
                 remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm")
 
+            prompt = gr.Dropdown(
+                label="Select Prompt",
+                choices=list(get_prompts_by_gender("woman").values()),  # default to woman
+                # value=list(get_prompts_by_gender("woman").values())[0],
+                value = "Select Your Prompt",
+                interactive=True
+            )
 
             gender_selector = gr.Radio(
                 label="Choose Gender",
                 choices=["woman", "man"],
                 value="woman"
             )
-            
-            prompt_selector = gr.Dropdown(
-                label="Select Prompt",
-                choices=list(get_prompts_by_gender("woman").keys()),
-                value="Select Prompt Theme",  # default name
-                interactive=True
-            )
-
-
-            def get_full_prompt(gender, prompt_name):
-                return get_prompts_by_gender(gender)[prompt_name]
-
-            prompt_selector.change(
-                fn=get_full_prompt,
-                inputs=[gender_selector, prompt_selector],
-                outputs=prompt  # `prompt` here would be your gr.Textbox holding the full text
-            )
-
-
-            # prompt = gr.Dropdown(
-            #     label="Select Prompt",
-            #     # choices=list(get_prompts_by_gender("woman").values()),  # default to woman
-            #     choices=list(get_prompts_by_gender("woman").keys()),
-            #     # value=list(get_prompts_by_gender("woman").values())[0],
-            #     value = "Select Your Prompt",
-            #     interactive=True
-            # )
-
-            
 
 
             with gr.Row():
@@ -515,8 +493,8 @@ with gr.Blocks(css=css) as demo:
             
            
             def update_prompt_dropdown(gender):
-                prompt_names = list(get_prompts_by_gender(gender).keys())
-                return gr.update(choices=prompt_names, 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
             
             gender_selector.change(
                 # fn=update_prompt_on_gender_change,
@@ -525,7 +503,7 @@ with gr.Blocks(css=css) as demo:
 
                 fn=update_prompt_dropdown,
                 inputs=gender_selector,
-                outputs=prompt_selector
+                outputs=prompt
             )
 
           
@@ -611,7 +589,8 @@ with gr.Blocks(css=css) as demo:
                 )
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
         with gr.Column():
-            gallery = gr.Gallery(label="Generated Images")
+            # gallery = gr.Gallery(label="Generated Images")
+            gallery = gr.Gallery(label="Generated Images", show_share_button=False)
             usage_tips = gr.Markdown(label="Usage tips of PhotoMaker", value=tips ,visible=False, elem_id="usage_tips_section")
 
             # usage_tips = gr.Markdown(label="Usage tips of PhotoMaker", value=tips ,visible=False)