resize 1000x1000

app.py

@@ -1,3 +1,5 @@
+from ast import Interactive
+from xml.sax.xmlreader import InputSource
 import gradio as gr
 import numpy as np
 import torch
@@ -11,6 +13,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 from predictor import Predictor
 
+display_height = 600
 H = 256
 W = 256
 
@@ -96,7 +99,13 @@ def image_overlay(img, mask=None, scribbles=None, contour=False, alpha=0.5):
     return output
     
 
-def viz_pred_mask(img, mask=None, point_coords=None, point_labels=None, bbox_coords=None, seperate_scribble_masks=None, binary=True):
+def viz_pred_mask(img, 
+                  mask=None, 
+                  point_coords=None, 
+                  point_labels=None, 
+                  bbox_coords=None, 
+                  seperate_scribble_masks=None, 
+                  binary=True):
     """
     Visualize image with clicks, scribbles, predicted mask overlaid
     """
@@ -266,7 +275,7 @@ def get_select_coords(predictor, input_img, brush_label, bbox_label, best_mask,
         # Don't update output image if waiting for additional bounding box click
         return click_input_viz, scribble_input_viz, output_img, best_mask, low_res_mask, img_features, click_coords, click_labels, bbox_coords, seperate_scribble_masks, last_scribble_mask   
     
-
+    
 def undo_click(predictor, input_img, brush_label, bbox_label, best_mask, low_res_mask, click_coords, click_labels, bbox_coords,
                seperate_scribble_masks, last_scribble_mask, scribble_img, img_features,
                 output_img, binary_checkbox, multimask_mode, autopredict_checkbox):
@@ -373,10 +382,9 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
             multimask_mode = gr.Checkbox(value=True, label="Multi-mask mode", visible=False)
 
     with gr.Row():
-        display_height = 500
-
-        green_brush = gr.Brush(colors=["#00FF00"], color_mode="fixed", default_size=2)
-        red_brush = gr.Brush(colors=["#FF0000"], color_mode="fixed", default_size=2)
+        
+        green_brush = gr.Brush(colors=["#00FF00"], color_mode="fixed", default_size=3)
+        red_brush = gr.Brush(colors=["#FF0000"], color_mode="fixed", default_size=3)
 
         with gr.Column(scale=1):
             with gr.Tab("Scribbles"):
@@ -388,8 +396,9 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
                     value=default_example,  
                     transforms=(),
                     sources=(),
+                    container=True,
                     show_download_button=True,
-                    # height=display_height
+                    height=display_height+60
                 )
 
             with gr.Tab("Clicks/Boxes") as click_tab:
@@ -398,9 +407,8 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
                     type='numpy',
                     value=default_example, 
                     show_download_button=True,
-                    sources=(),
                     container=True,
-                    # height=display_height-50
+                    height=display_height
                 )
                 with gr.Row():
                     undo_click_button = gr.Button("Undo Last Click")
@@ -411,8 +419,9 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
                     label="Input",
                     image_mode="L",
                     value=default_example, 
-                    container=True
-                    # height=display_height
+                    show_download_button=True,
+                    container=True,
+                    height=display_height
                 ) 
                 gr.Markdown("To upload your own image: click the `x` in the top right corner to clear the current image, then drag & drop")
 
@@ -424,7 +433,7 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
                     elem_id="gallery",
                     preview=True, 
                     object_fit="scale-down",
-                    # height=display_height,
+                    height=display_height+60,
                     container=True
                 )
     
@@ -451,6 +460,16 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
                 label='Examples from datasets unseen during training'
                 )
 
+    # with gr.Accordion():
+    #     height_scribble = gr.Number(label="Scribble Panel Height", 
+    #                                 value=display_height, Interactive=True)
+
+    # height_scribble.change(
+    #     fn=lambda x: gr.update(height=x),
+    #     inputs=[scribble_img],
+    #     outputs=[scribble_img],
+    # )
+
     # When clear clicks button is clicked
     def clear_click_history(input_img):
         return input_img, input_img, [], [], [], None, None