Update gradio_demo.py

gradio_demo.py

@@ -1251,7 +1251,8 @@ with block:
                         choices=[e.value for e in BGSource],
                         value=BGSource.UPLOAD.value,
                         label="Background Source", 
-                        type='value'
+                        type='value',
+                        visible=False
                     )
 
                 example_quick_subjects = gr.Dataset(samples=quick_subjects, label='Subject Quick List', samples_per_page=1000, components=[prompt])
@@ -1486,8 +1487,13 @@ with block:
 
         
         return result
+
+    if original_bg:
+        chosen_bg = original_bg.copy()
+    else:
+        chosen_bg = input_bg
     
-    ips_bg = [input_fg, input_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source, x_slider, y_slider, fg_scale_slider]
+    ips_bg = [input_fg, chosen_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source, x_slider, y_slider, fg_scale_slider]
 
     # Update button click events with new inputs list
     relight_button_bg.click(