Update app.py

app.py

@@ -74,27 +74,27 @@ def fetch_referenced_object(speckleToken, apiUrl, stream_id, object_id):
     return send_graphql_query(speckleToken, apiUrl, query)
 
 def extract_team_name(object_data):
-    # Try to get the team name directly
     data_field = object_data['data']['stream']['object']['data']
+
+    # Check for the first structure with direct 'data'
     try:
-        nested_data_str = data_field['data']
-        nested_data = json.loads(nested_data_str)
-        team_name = nested_data.get('teamName')
-        if team_name:
-            return team_name
+        nested_data_str = data_field.get('data')
+        if nested_data_str:
+            nested_data = json.loads(nested_data_str)
+            team_name = nested_data.get('teamName')
+            if team_name:
+                return team_name
     except (KeyError, json.JSONDecodeError):
         pass
 
-    # If not found, check for referencedId and fetch further data
-    if '@Data' in data_field:
-        referenced_id = data_field['@Data'].get('referencedId')
-        if referenced_id:
-            referenced_object_data = fetch_referenced_object(speckle_token, api_url, stream_id, referenced_id)
-            print ("__________", referenced_object_data)
-            if referenced_object_data:
-                nested_data_str = referenced_object_data['data']['stream']['object']['data']['data']
-                nested_data = json.loads(nested_data_str)
-                return nested_data.get('teamName')
+    # Check for the second structure under 'Data' -> '@{0}'
+    try:
+        data_object = data_field.get('Data', {})
+        team_name_list = data_object.get('@{0}')
+        if team_name_list and isinstance(team_name_list, list):
+            return team_name_list[0]
+    except (KeyError, json.JSONDecodeError) as e:
+        print(f"Error extracting team name: {e}")
 
     return None