Update app.py

app.py

@@ -8,6 +8,16 @@ def process_metadata(file):
     with open(file.name, "r") as f:
         data = json.load(f)
 
+    # First pass: collect all unique categories
+    unique_categories = set()
+    for item in data["media_items"]:
+        for metadata in item["metadata_items"]:
+            if metadata["type"] == "object_label":
+                unique_categories.add(metadata["properties"]["category_name"])
+
+    # Create category mapping dynamically
+    category_mapping = {cat: idx for idx, cat in enumerate(sorted(unique_categories))}
+
     results = []
     # Process each media item
     for item in data["media_items"]:
@@ -21,10 +31,6 @@ def process_metadata(file):
                 bbox = metadata["properties"]["bbox"]
                 bboxes.append(bbox)
 
-                category_mapping = {
-                    "vest": 0,
-                    # Add other categories as needed
-                }
                 category = category_mapping.get(
                     metadata["properties"]["category_name"], 0
                 )