Update app.py

app.py

@@ -132,8 +132,12 @@ def construct_both_graph(docs):
     nodes = [token.text for token in doc]
     nodes.append("Sentence")
     senders = [token.i for token in doc][:-1]
+    senders.extend([token.i for token in doc][1:])
     receivers = [token.i for token in doc][1:]
+    receivers.extend([token.i for token in doc][:-1])
     edge_labels = {(token.i, token.i + 1): "next" for token in doc[:-1]}
+    for token in doc[:-1]:
+        edge_labels[(token.i + 1, token.i)] = "previous"
     for node in range(len(nodes) - 1):
       senders.append(node)
       receivers.append(len(nodes) - 1)
@@ -152,8 +156,12 @@ def construct_structural_graph(docs):
     nodes = [token.text for token in doc]
     nodes.append("Sentence")
     senders = [token.i for token in doc][:-1]
+    senders.extend([token.i for token in doc][1:])
     receivers = [token.i for token in doc][1:]
+    receivers.extend([token.i for token in doc][:-1])
     edge_labels = {(token.i, token.i + 1): "next" for token in doc[:-1]}
+    for token in doc[:-1]:
+        edge_labels[(token.i + 1, token.i)] = "previous"
     for node in range(len(nodes) - 1):
       senders.append(node)
       receivers.append(len(nodes) - 1)