Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import re
 import nltk
+import os
 from nltk.corpus import stopwords
 from nltk import FreqDist
 from graphviz import Digraph
@@ -63,10 +64,26 @@ def display_context_table(context_words):
         table += f"| {before if before else ''} | {high} | {after if after else ''} |\n"
     st.markdown(table)
 
+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}"):
+        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'])
 
-if uploaded_file:
+example_text = load_example_files()
+
+if example_text:
+    file_text = example_text
+elif uploaded_file:
     file_text = uploaded_file.read().decode("utf-8")
+else:
+    file_text = ""
+
+if file_text:
     text_without_timestamps = remove_timestamps(file_text)
 
     top_words = extract_high_information_words(text_without_timestamps, 10)
@@ -79,6 +96,6 @@ if uploaded_file:
     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)