From 661e702de1082d4e7dbaa5a4dbf98b7533c29c4a Mon Sep 17 00:00:00 2001
From: Tahiya Salam <tahiyasalam@gmail.com>
Date: Thu, 27 Apr 2023 08:49:56 -0400
Subject: [PATCH] Add bbox to filter cli

---
 cli/viam/main.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/cli/viam/main.go b/cli/viam/main.go
index 5dbcd15f1bc..b94370cf8b2 100644
--- a/cli/viam/main.go
+++ b/cli/viam/main.go
@@ -36,6 +36,7 @@ const (
 	dataFlagEnd               = "end"
 	dataFlagParallelDownloads = "parallel"
 	dataFlagTags              = "tags"
+	dataFlagBboxLabels        = "bbox_labels"
 
 	dataTypeBinary  = "binary"
 	dataTypeTabular = "tabular"
@@ -326,6 +327,12 @@ func main() {
 								Usage: "tags filter. " +
 									"accepts tagged for all tagged data, untagged for all untagged data, or a list of tags for all data matching any of the tags",
 							},
+							&cli.StringSliceFlag{
+								Name:     dataFlagBboxLabels,
+								Required: false,
+								Usage: "bbox labels filter. " +
+									"accepts string labels corresponding to bounding boxes within images",
+							},
 						},
 						Action: DataCommand,
 					},
@@ -923,7 +930,9 @@ func createDataFilter(c *cli.Context) (*datapb.Filter, error) {
 			}
 		}
 	}
-
+	if len(c.StringSlice(dataFlagBboxLabels)) != 0 {
+		filter.BboxLabels = c.StringSlice(dataFlagBboxLabels)
+	}
 	var start *timestamppb.Timestamp
 	var end *timestamppb.Timestamp
 	timeLayout := time.RFC3339