Update app.py

app.py

@@ -64,9 +64,11 @@ def extract_context_words(text, high_information_words):
 def create_context_graph(context_words):
     graph = Digraph()
     for index, (before_word, high_info_word, after_word) in enumerate(context_words):
-        graph.node(f'before{index}', before_word, shape='box') if before_word else None
+        #graph.node(f'before{index}', before_word, shape='box') if before_word else None
+        if before_word:  graph.node(f'before{index}', before_word, shape='box') # else None
         graph.node(f'high{index}', high_info_word, shape='ellipse')
-        graph.node(f'after{index}', after_word, shape='diamond') if after_word else None
+        #graph.node(f'after{index}', after_word, shape='diamond') if after_word else None
+        if after_word: graph.node(f'after{index}', after_word, shape='diamond') # else None
         if before_word:
             graph.edge(f'before{index}', f'high{index}')
         if after_word: