Update app.py

app.py

@@ -102,8 +102,17 @@ def update_segmentation_options(options):
 #def update_confidence_threshold(thres_val):
 #    img_seg.confidence_threshold = thres_val/100
 
+#def update_confidence_threshold(thres_val, img_seg_instance):
+#    """Update confidence threshold in ImageSegmenter"""
+#    img_seg_instance.confidence_threshold = thres_val
+#    print(f"Confidence threshold updated to: {thres_val}")
+
+
 def update_confidence_threshold(thres_val, img_seg_instance):
     """Update confidence threshold in ImageSegmenter"""
+    # For Gradio UI (0-100), convert to 0.0-1.0; for API (0.0-1.0), use directly
+    if thres_val > 1.0:  # Assume Gradio slider value
+        thres_val = thres_val / 100.0
     img_seg_instance.confidence_threshold = thres_val
     print(f"Confidence threshold updated to: {thres_val}")
 
@@ -473,16 +482,29 @@ if __name__ == "__main__":
             
 
         # image tab logic
+        #submit_btn_img.click(process_image, inputs=img_input, outputs=[segmentation_img_output, depth_img_output, dist_img_output, pcd_img_output])
+        #options_checkbox_img.change(update_segmentation_options, options_checkbox_img, [])
+        #conf_thres_img.change(update_confidence_threshold, conf_thres_img, [])
+        #model_type_img.change(model_selector, model_type_img, [])
+
+        # video tab logic
+        #submit_btn_vid.click(process_video, inputs=vid_input, outputs=[segmentation_vid_output, depth_vid_output, dist_vid_output])
+        #model_type_vid.change(model_selector, model_type_vid, [])
+        #cancel_btn.click(cancel, inputs=[], outputs=[])
+        #options_checkbox_vid.change(update_segmentation_options, options_checkbox_vid, [])
+        #conf_thres_vid.change(update_confidence_threshold, conf_thres_vid, []) 
+
+        # Image tab logic
         submit_btn_img.click(process_image, inputs=img_input, outputs=[segmentation_img_output, depth_img_output, dist_img_output, pcd_img_output])
         options_checkbox_img.change(update_segmentation_options, options_checkbox_img, [])
-        conf_thres_img.change(update_confidence_threshold, conf_thres_img, [])
-        model_type_img.change(model_selector, model_type_img, [])
+        conf_thres_img.change(lambda x: update_confidence_threshold(x, img_seg), conf_thres_img, [])  # Pass img_seg explicitly
+        model_type_img.change(lambda x: model_selector(x, img_seg, depth_estimator), model_type_img, [])
 
-        # video tab logic
+        # Video tab logic
         submit_btn_vid.click(process_video, inputs=vid_input, outputs=[segmentation_vid_output, depth_vid_output, dist_vid_output])
-        model_type_vid.change(model_selector, model_type_vid, [])
+        model_type_vid.change(lambda x: model_selector(x, img_seg, depth_estimator), model_type_vid, [])
         cancel_btn.click(cancel, inputs=[], outputs=[])
         options_checkbox_vid.change(update_segmentation_options, options_checkbox_vid, [])
-        conf_thres_vid.change(update_confidence_threshold, conf_thres_vid, [])      
+        conf_thres_vid.change(lambda x: update_confidence_threshold(x, img_seg), conf_thres_vid, [])  # Pass img_seg explicitly
 
     my_app.queue(max_size=20).launch(share=True)  # Add share=True here