From 2947cc3ff477c5f270d854837eeb8d08ab71d481 Mon Sep 17 00:00:00 2001
From: iris-liu0312 <iris.liu0312@gmail.com>
Date: Sun, 5 Dec 2021 12:18:57 -0700
Subject: [PATCH] all subtrees viewable

---
 app.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index ea63657..d018b87 100644
--- a/app.py
+++ b/app.py
@@ -12,7 +12,7 @@
 
 # SDF version 1.2
 schema_key_dict = {
-    'root': ['@id', 'name', 'description', 'comment', '@type', 'repeatable'],
+    'root': ['@id', 'name', 'description', 'comment', '@type', 'repeatable', 'TA1explanation'],
     'participant': ['@id', 'roleName', 'entity'],
     'child': ['child', 'comment', 'outlinks', 'outlink_gate', 'optional']
 }
@@ -239,6 +239,9 @@ def get_connected_nodes(selected_node):
     for id in id_set:
         for edge in edges:
             if edge['data']['source'] == id and edge['data']['_edge_type'] == 'child_outlink':
+                # check if node was created previously
+                if edge['data']['target'] not in id_set:
+                    n.append(nodes[edge['data']['target']])
                 e.append(edge)
 
     return {