change style

app.py

@@ -3,10 +3,18 @@ from backend import visualize_image
 
 # gradio inputs
 image_input = gr.inputs.Image(type="pil", label="Input Image")
-mode_dropdown = gr.inputs.Dropdown(["Trees", "Buildings", "Both"])
-tree_threshold_slider = gr.inputs.Slider(0, 1, 0.1, 0.7, label='Set confidence threshold % for trees')
-building_threshold_slider = gr.inputs.Slider(0, 1, 0.1, 0.7, label='Set confidence threshold % for buildings')
 color_mode_select = gr.inputs.Radio(["Black/white", "Random", "Segmentation"])
+mode_dropdown = gr.inputs.Dropdown(["Trees", "Buildings", "Both"])
+
+tree_threshold_slider = gr.inputs.Slider(0, 1, 0.1, 0.7, label='Set confidence threshold "%" for trees')
+building_threshold_slider = gr.inputs.Slider(0, 1, 0.1, 0.7, label='Set confidence threshold "%" for buildings')
+if mode_dropdown == "Trees":
+    tree_threshold_slider.enable()
+if mode_dropdown == "Buildings":
+    building_threshold_slider.enable()
+if mode_dropdown == "Both":
+    tree_threshold_slider.enable()
+    building_threshold_slider.enable()
 
 # gradio outputs
 output_image = gr.outputs.Image(type="pil", label="Output Image")

backend.py

@@ -68,7 +68,7 @@ def map_color_mode(color_mode):
     elif color_mode == "Segmentation":
         return ColorMode.SEGMENTATION
 
-def visualize_image(im, mode, tree_threshold, building_threshold, color_mode=ColorMode.SEGMENTATION):
+def visualize_image(im, mode="BOTH", tree_threshold=0.7, building_threshold=0.7, color_mode=ColorMode.SEGMENTATION):
     im = np.array(im)
     color_mode = map_color_mode(color_mode)