Merge pull request #534 from superannotateai/1510_objects_json

fixing __objects.json and freestyle attributes

Author: VaghinakDev

src/superannotate/lib/app/analytics/common.py

@@ -4,7 +4,6 @@
 import pandas as pd
 import plotly.express as px
 from lib.app.exceptions import AppException
-from lib.core import DEPRICATED_DOCUMENT_VIDEO_MESSAGE
 from superannotate.logger import get_default_logger
 
 
@@ -44,14 +43,6 @@ def aggregate_image_annotations_as_df(
     :rtype: pandas DataFrame
     """
 
-    json_paths = list(Path(str(project_root)).glob("*.json"))
-    if (
-        json_paths
-        and "___pixel.json" not in json_paths[0].name
-        and "___objects.json" not in json_paths[0].name
-    ):
-        raise AppException(DEPRICATED_DOCUMENT_VIDEO_MESSAGE)
-
     logger.info("Aggregating annotations from %s as pandas DataFrame", project_root)
 
     annotation_data = {
@@ -101,12 +92,15 @@ def aggregate_image_annotations_as_df(
     classes_json = json.load(open(classes_path))
     class_name_to_color = {}
     class_group_name_to_values = {}
+    freestyle_attributes = set()
     for annotation_class in classes_json:
         name = annotation_class["name"]
         color = annotation_class["color"]
         class_name_to_color[name] = color
         class_group_name_to_values[name] = {}
         for attribute_group in annotation_class["attribute_groups"]:
+            if attribute_group["group_type"] in ["text", "numeric"]:
+                freestyle_attributes.add(attribute_group["name"])
             class_group_name_to_values[name][attribute_group["name"]] = []
             for attribute in attribute_group["attributes"]:
                 class_group_name_to_values[name][attribute_group["name"]].append(
@@ -175,10 +169,14 @@ def __get_user_metadata(annotation):
 
     if not annotations_paths:
         logger.warning(f"Could not find annotations in {project_root}.")
-    if len(list(Path(project_root).rglob("*___objects.json"))) > 0:
+
+    if "___objects.json" in annotations_paths[0].name:
         type_postfix = "___objects.json"
-    else:
+    elif "___pixel.json" in annotations_paths[0].name:
         type_postfix = "___pixel.json"
+    else:
+        type_postfix = ".json"
+
     for annotation_path in annotations_paths:
         annotation_json = json.load(open(annotation_path))
         parts = annotation_path.name.split(type_postfix)
@@ -294,6 +292,7 @@ def __get_user_metadata(annotation):
                         not in class_group_name_to_values[annotation_class_name][
                             attribute_group
                         ]
+                        and attribute_group not in freestyle_attributes
                     ):
                         logger.warning(
                             "Annotation class group value %s not in classes json. Skipping.",