Update app.py

app.py

@@ -74,20 +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):
-    # First try to get the team name directly
+    # Try to get the team name directly
     data_field = object_data['data']['stream']['object']['data']
     try:
         nested_data_str = data_field['data']
         nested_data = json.loads(nested_data_str)
-        return nested_data.get('teamName')
+        team_name = nested_data.get('teamName')
+        if team_name:
+            return team_name
     except (KeyError, json.JSONDecodeError):
         pass
 
-    # If not found, check for referencedId
+    # 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:
-            return referenced_id  # Returning the ID to fetch further if needed
+            referenced_object_data = fetch_referenced_object(speckle_token, api_url, stream_id, referenced_id)
+            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')
+
     return None
 
 @webhook_endpoint