Update app.py

app.py

@@ -1,3 +1,4 @@
+# Import necessary libraries
 import streamlit as st
 import re
 import nltk
@@ -6,6 +7,7 @@ from nltk.corpus import stopwords
 from nltk import FreqDist
 from graphviz import Digraph
 
+# Download NLTK resources
 nltk.download('punkt')
 nltk.download('stopwords')
 
@@ -64,15 +66,17 @@ def display_context_table(context_words):
         table += f"| {before if before else ''} | {high} | {after if after else ''} |\n"
     st.markdown(table)
 
+# Load example files
 def load_example_files():
     example_files = [f for f in os.listdir() if f.endswith('.txt')]
-    selected_file = st.selectbox("Select an example file:", example_files)
-    if st.button(f"Load {selected_file}"):
+    selected_file = st.selectbox("📄 Select an example file:", example_files)
+    if st.button(f"📂 Load {selected_file}"):
         with open(selected_file, 'r', encoding="utf-8") as file:
             return file.read()
     return None
 
-uploaded_file = st.file_uploader("Choose a .txt file", type=['txt'])
+# Main code for UI
+uploaded_file = st.file_uploader("📁 Choose a .txt file", type=['txt'])
 
 example_text = load_example_files()
 
@@ -85,17 +89,18 @@ else:
 
 if file_text:
     text_without_timestamps = remove_timestamps(file_text)
-
     top_words = extract_high_information_words(text_without_timestamps, 10)
-    st.markdown("**Top 10 High Information Words:**")
-    st.write(top_words)
 
-    st.markdown("**Relationship Graph:**")
-    display_relationship_graph(top_words)
+    with st.expander("📊 Top 10 High Information Words"):
+        st.write(top_words)
+
+    with st.expander("📈 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)
+    with st.expander("🔗 Context Graph"):
+        display_context_graph(context_words)
+
+    with st.expander("📑 Context Table"):
+        display_context_table(context_words)