Update app.py

app.py

@@ -395,6 +395,7 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
     # Get input dimensions
     input_height, input_width = input_fg.shape[:2]
 
+    if bg_source is not ""
     bg_source = BGSource(bg_source)
 
 
@@ -402,7 +403,7 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
         pass
     elif bg_source == BGSource.UPLOAD_FLIP:
         input_bg = np.fliplr(input_bg)
-    elif bg_source == BGSource.GREY:
+    if bg_source == BGSource.GREY:
         input_bg = np.zeros(shape=(input_height, input_width, 3), dtype=np.uint8) + 64
     elif bg_source == BGSource.LEFT:
         gradient = np.linspace(255, 0, input_width)
@@ -620,6 +621,8 @@ def process_relight(input_fg, prompt, image_width, image_height, num_samples, se
 @torch.inference_mode()
 def process_relight_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):
     bg_source = BGSource(bg_source)
+
+    # bg_source = "Use Background Image"
     
     # Convert numerical inputs to appropriate types
     image_width = int(image_width)
@@ -697,8 +700,8 @@ quick_subjects = [[x] for x in quick_subjects]
 
 
 class BGSource(Enum):
-    # UPLOAD = "Use Background Image"
-    # UPLOAD_FLIP = "Use Flipped Background Image"
+    UPLOAD = "Use Background Image"
+    UPLOAD_FLIP = "Use Flipped Background Image"
     LEFT = "Left Light"
     RIGHT = "Right Light"
     TOP = "Top Light"
@@ -984,7 +987,7 @@ with block:
                     # output_bg = gr.Image(type="numpy", label="Preprocessed Foreground", height=480)
                 with gr.Group():
                     prompt = gr.Textbox(label="Prompt")
-                    bg_source = gr.Radio(choices=[e.value for e in BGSource],
+                    bg_source = gr.Radio(choices=[e.value for e in BGSource[2:]],
                                         value=BGSource.GREY.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])
@@ -1106,9 +1109,10 @@ with block:
                     prompt = gr.Textbox(label="Prompt")
                     bg_source = gr.Radio(
                         choices=[e.value for e in BGSource],
-                        #value=BGSource.UPLOAD.value,
+                        value=BGSource.UPLOAD.value,
                         label="Background Source", 
-                        type='value'
+                        type='value',
+                        visible=False
                     )
 
                     example_prompts = gr.Dataset(