Update merged_files3.py

merged_files3.py

@@ -153,6 +153,11 @@ def download_models():
 ensure_directories()
 download_models()
 
+def clear_memory():
+       if torch.cuda.is_available():
+           torch.cuda.empty_cache()
+           torch.cuda.synchronize()
+
 
 hf_hub_download(repo_id="black-forest-labs/FLUX.1-Redux-dev", filename="flux1-redux-dev.safetensors", local_dir="models/style_models")
 hf_hub_download(repo_id="black-forest-labs/FLUX.1-Depth-dev", filename="flux1-depth-dev.safetensors", local_dir="models/diffusion_models")
@@ -1698,10 +1703,10 @@ with gr.Blocks() as app:
                         with gr.Group():
                             gr.Markdown("Outpaint")
                             with gr.Row():
-                                with gr.Column(scale=2):
-                                    prompt_fill = gr.Textbox(label="Prompt (Optional)")
-                                with gr.Column(scale=1):
-                                    fill_button = gr.Button("Generate")
+                                # with gr.Column(scale=2):
+                                #     prompt_fill = gr.Textbox(label="Prompt (Optional)")
+                                # with gr.Column(scale=1):
+                                fill_button = gr.Button("Generate")
                             target_ratio = gr.Radio(
                                 label="Image Ratio",
                                 choices=["9:16", "16:9", "1:1", "Custom"],
@@ -1777,7 +1782,11 @@ with gr.Blocks() as app:
                         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.Column(scale=4):
+                        relight_button = gr.Button(value="Relight")
+                    with gr.Column(scale=1):
+                        augment_prompt = gr.Button(value='+')
+
 
                 with gr.Group(visible=False):
                     with gr.Row():
@@ -1822,6 +1831,9 @@ 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], 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]).then(clear_memory, inputs=[], outputs=[])
         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)
@@ -1871,7 +1883,7 @@ with gr.Blocks() as app:
         ).then(
             fn=inpaint,
             inputs=[dummy_image_for_outputs, width_slider, height_slider, overlap_percentage, num_inference_steps,
-                    resize_option, custom_resize_percentage, prompt_fill, alignment_dropdown,
+                    resize_option, custom_resize_percentage, prompt, alignment_dropdown,
                     overlap_left, overlap_right, overlap_top, overlap_bottom],
             outputs=[fill_result]).then(clear_memory, inputs=[], outputs=[])
         # ).then(
@@ -1880,16 +1892,16 @@ with gr.Blocks() as app:
         #     outputs=use_as_input_button,
         # )
 
-        prompt_fill.submit(
-            fn=clear_result,
-            inputs=None,
-            outputs=fill_result,
-        ).then(
-            fn=inpaint,
-            inputs=[dummy_image_for_outputs, width_slider, height_slider, overlap_percentage, num_inference_steps,
-                    resize_option, custom_resize_percentage, prompt_fill, alignment_dropdown,
-                    overlap_left, overlap_right, overlap_top, overlap_bottom],
-            outputs=[fill_result]).then(clear_memory, inputs=[], outputs=[])
+        # prompt_fill.submit(
+        #     fn=clear_result,
+        #     inputs=None,
+        #     outputs=fill_result,
+        # ).then(
+        #     fn=inpaint,
+        #     inputs=[dummy_image_for_outputs, width_slider, height_slider, overlap_percentage, num_inference_steps,
+        #             resize_option, custom_resize_percentage, prompt_fill, alignment_dropdown,
+        #             overlap_left, overlap_right, overlap_top, overlap_bottom],
+        #     outputs=[fill_result]).then(clear_memory, inputs=[], outputs=[])
 
         def convert_to_pil(image):
             try: