fixed gradio logic

app.py

@@ -18,11 +18,11 @@ with gr.Blocks() as demo:
         with gr.Column():
             image_input = gr.components.Image(type="pil", label="Input Image")
         with gr.Column():
-            mode_dropdown = gr.Dropdown(choices=["Trees", "Buildings", "Both"], label="Detection Mode", value="Both")
+            mode_dropdown = gr.Dropdown(choices=["Trees", "Buildings", "Trees & Buildings", "LCZ"], label="Detection Mode", value="Trees & Buildings")
             color_mode_select = gr.components.Radio(choices=["Black/white", "Random", "Segmentation"], label="Color Mode", value="Segmentation")
 
     # split tree and building into two rows side by side
-    tree_row, building_row , lcz_row = gr.Row(), gr.Row(), gr.Row()
+    tree_row, building_row , lcz_row = gr.Row(), gr.Row(), gr.Row(visible=False)
     # tree_col, building_col = gr.Column(elem_id="tree_col"), gr.Column(elem_id="building_col")
     with tree_row as tree_options:
         tree_version_dropdown = gr.Dropdown(choices=list_cfg_file_versions("tree_model_weights"), label="Tree Detection Version", value="treev2", visible=True, interactive=True)
@@ -48,5 +48,4 @@ with gr.Blocks() as demo:
     output_image = gr.components.Image(type="pil", label="Output Image")
     run_model = gr.Button("Upload Image and Run Model")
     
-    run_model.click(visualize_image, inputs=[image_input, mode_dropdown, tree_threshold_slider, building_threshold_slider, lcz_threshold_slider, color_mode_select, tree_version_dropdown, tree_pth_dropdown, building_version_dropdown, building_pth_dropdown, lcz_version_dropdown, lcz_pth_dropdown], outputs=[output_image])
     demo.launch()

backend.py

@@ -51,18 +51,21 @@ def get_version_cfg_yml(path):
 
 def update_row_visibility(mode):
     visibility = {
-        "tree": mode in ["Trees", "Both"],
-        "building": mode in ["Buildings", "Both"]
+        "tree": mode in ["Trees", "Trees & Buildings"],
+        "building": mode in ["Buildings", "Trees & Buildings"],
+        "lcz": mode in ["LCZ"]
     }
-    tree_row, building_row = gr.Row(visible=visibility["tree"]), gr.Row(visible=visibility["building"])
-    
-    return tree_row, building_row
+    tree_row, building_row, lcz_row = gr.Row(visible=visibility["tree"]), gr.Row(visible=visibility["building"]), gr.Row(visible=visibility["lcz"])
+    print(visibility)
+    return tree_row, building_row, lcz_row
 
 def update_path_options(version):
     if "tree" in version:
         directory = "tree_model_weights"
-    else:
+    elif "building" in version:
         directory = "building_model_weights"
+    elif "lcz" in version:
+        directory = "lcz_model_weights"
     return gr.Dropdown(choices=list_pth_files_in_directory(directory, version), label=f"Select a {version.split('v')[0]} model file", visible=True, interactive=True)
 
 # Model for trees