Update app_merged.py

app_merged.py

@@ -1751,15 +1751,19 @@ with gr.Blocks() as app:
                         with gr.Column():
                             prompt = gr.Textbox(label="Prompt")
                             augment_prompt = gr.Button(value='Augment Prompt')
+
+                        relight_button = gr.Button(value="Relight")
+
+                        cfg = gr.Slider(label="Fidelity", minimum=1.0, maximum=32.0, value=2, step=0.01, visible=False)
                         
                         bg_source = gr.Radio(choices=[e.value for e in list(BGSource)[2:]],
                                             value=BGSource.LEFT.value,
                                             label="Lighting Preference (Initial Latent)", type='value')
+
                     
                         example_quick_subjects = gr.Dataset(samples=quick_subjects, label='Subject Quick List', samples_per_page=1000, components=[prompt])
                         example_quick_prompts = gr.Dataset(samples=quick_prompts, label='Lighting Quick List', samples_per_page=1000, components=[prompt])
-                with gr.Row():    
-                    relight_button = gr.Button(value="Relight")
+                    
 
                 with gr.Group(visible=False):
                     with gr.Row():
@@ -1772,7 +1776,6 @@ with gr.Blocks() as app:
 
                         with gr.Accordion("Advanced options", open=False):
                             steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=15, step=1)
-                            cfg = gr.Slider(label="CFG Scale", minimum=1.0, maximum=32.0, value=2, step=0.01, visible=False)
                             lowres_denoise = gr.Slider(label="Lowres Denoise (for initial latent)", minimum=0.1, maximum=1.0, value=0.9, step=0.01)
                             highres_scale = gr.Slider(label="Highres Scale", minimum=1.0, maximum=3.0, value=1.5, step=0.01)
                             highres_denoise = gr.Slider(label="Highres Denoise", minimum=0.1, maximum=1.0, value=0.5, step=0.01)
@@ -1804,7 +1807,7 @@ with gr.Blocks() as app:
             #     outputs=[result_gallery, output_bg],
             #     run_on_click=True, examples_per_page=1024
             # )
-        augment_prompt.click(generate_description, inputs=[prompt, dummy_image_for_prompt_augmentation], outputs=[prompt])
+        augment_prompt.click(generate_description, inputs=[prompt, extracted_fg], outputs=[prompt])
         ips = [extracted_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
         relight_button.click(fn=process_relight, inputs=ips, outputs=[result_gallery])
         example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
@@ -1949,4 +1952,5 @@ with gr.Blocks() as app:
         )
 
 if __name__ == "__main__":
-    app.launch(share=True)
+    app.queue(default_concurrency_limit=3)
+    app.launch(share=True)