Update app.py

app.py

@@ -58,30 +58,6 @@ 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:
@@ -118,12 +94,8 @@ if file_text:
     display_relationship_graph(top_words)
 
     context_words = extract_context_words(text_without_timestamps, top_words)
-    if context_words and all(context_words[0]):
-        st.markdown("**Context Graph:**")
-        display_context_graph(context_words)
-
-        st.markdown("**Combined Context Graph:**")
-        display_combined_context_graph(context_words, top_words)  # New combined context graph
+    st.markdown("**Context Graph:**")
+    display_context_graph(context_words)
 
-        st.markdown("**Context Table:**")
-        display_context_table(context_words)
+    st.markdown("**Context Table:**")
+    display_context_table(context_words)