Update app_3.py

app_3.py

@@ -613,6 +613,13 @@ def extract_foreground(image):
     logging.info(f"RGBA shape: {rgba.shape}, dtype: {rgba.dtype}")
     return result, gr.update(visible=True), gr.update(visible=True)
 
+def update_extracted_fg_height(selected_image: gr.SelectData):
+    if selected_image:
+        # Get the height of the selected image
+        height = selected_image.value['image']['shape'][0]  # Assuming the image is in numpy format
+        return gr.update(height=height)  # Update the height of extracted_fg
+    return gr.update(height=480)  # Default height if no image is selected
+
 @torch.inference_mode()
 def process_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):
     clear_memory()
@@ -923,6 +930,7 @@ def compress_image(image):
 
 def use_orientation(selected_image:gr.SelectData):
     return selected_image.value['image']['path']
+    
 
 @spaces.GPU(duration=60)
 @torch.inference_mode
@@ -1188,9 +1196,9 @@ with block:
                     angles_fg = gr.Image(type="pil", label="Converted Foreground", height=480, visible=False)
 
                 with gr.Row():
-                    run_button = gr.Button("Generate alternative angles")
-                    
-                    orientation_result = gr.Gallery(
+                    with gr.Group():
+                        run_button = gr.Button("Generate alternative angles")
+                        orientation_result = gr.Gallery(
                                     label="Result",
                                     show_label=False,
                                     columns=[3],
@@ -1201,8 +1209,12 @@ with block:
                                 )                
 
                     if orientation_result:
-                        selected = gr.Number(visible=True)
-                        orientation_result.select(use_orientation, inputs=None, outputs=extracted_fg)
+                        #selected = gr.Number(visible=False)
+                        orientation_result.select(
+                            update_extracted_fg_height,  # Function to call on selection
+                            inputs=None,  # No inputs needed
+                            outputs=extracted_fg  # Output to update the extracted_fg component
+                            ).then(use_orientation, inputs=None, outputs=extracted_fg)
                     
                     # output_bg = gr.Image(type="numpy", label="Preprocessed Foreground", height=480)
                 with gr.Group():