Update app.py

app.py

@@ -58,6 +58,30 @@ def display_context_graph(context_words):
     graph = create_context_graph(context_words)
     st.graphviz_chart(graph)
 
+def create_combined_context_graph(context_words, high_information_words):
+    graph = Digraph()
+
+    # Create nodes for high information words
+    for word in high_information_words:
+        graph.node(word, shape='ellipse')
+
+    for before_word, high_info_word, after_word in context_words:
+        # Connect before_word to high_info_word
+        if before_word:
+            graph.node(before_word, shape='box' if before_word not in high_information_words else 'ellipse')
+            graph.edge(before_word, high_info_word)
+
+        # Connect high_info_word to after_word
+        if after_word:
+            graph.node(after_word, shape='box' if after_word not in high_information_words else 'ellipse')
+            graph.edge(high_info_word, after_word)
+
+    return graph
+
+def display_combined_context_graph(context_words, high_information_words):
+    graph = create_combined_context_graph(context_words, high_information_words)
+    st.graphviz_chart(graph)
+
 def display_context_table(context_words):
     table = "| Before | High Info Word | After |\n|--------|----------------|-------|\n"
     for before, high, after in context_words:
@@ -93,9 +117,4 @@ if file_text:
     st.markdown("**Relationship Graph:**")
     display_relationship_graph(top_words)
 
-    context_words = extract_context_words(text_without_timestamps, top_words)
-    st.markdown("**Context Graph:**")
-    display_context_graph(context_words)
-
-    st.markdown("**Context Table:**")
-    display_context_table(context_words)
+    context_words = extract_context_words(text_without_timestamps, top_words