Update app.py

app.py

@@ -52,8 +52,8 @@ def main():
     st.sidebar.subheader("Actions")
     if st.sidebar.button("Home"):
         display_home()
-    if st.sidebar.button("Word Count"):
-        display_word_count()
+    if st.sidebar.button("Word Cloud"):
+        display_word_cloud()
 
     st.sidebar.title("Books")
     selected_book = st.sidebar.selectbox("Select a Book", list(book_files.keys()))
@@ -87,9 +87,9 @@ def display_table(csv_df):
     )
     st.dataframe(styled_df)
 
-def display_word_count():
-    st.title("Word count")
-    st.write("Select the number of words to display.")
+def display_word_cloud():
+    st.title("Word Cloud")
+    st.write("Select the number of words to display and exclude words.")
 
     selected_book = st.sidebar.selectbox("Select a Book", list(book_files.keys()), key="word_count_book_select")
     selected_files = book_files[selected_book]
@@ -100,20 +100,24 @@ def display_word_count():
     
     if csv_df is not None:
         word_counts = {}
+        exclude_words = st.sidebar.text_input("Exclude Words (comma-separated)", "", key="exclude_words_input")
+        exclude_words = [word.strip() for word in exclude_words.split(",")]
+        
         for row in csv_df.itertuples(index=False):
             for word in str(row[1]).split():
-                if word in word_counts:
-                    word_counts[word] += 1
-                else:
-                    word_counts[word] = 1
+                if word not in exclude_words:
+                    if word in word_counts:
+                        word_counts[word] += 1
+                    else:
+                        word_counts[word] = 1
         
         num_words = st.slider("Select Number of Words", min_value=10, max_value=50, value=10, key="word_count_slider")
         top_words = sorted(word_counts, key=word_counts.get, reverse=True)[:num_words]
         
-        wordcount_data = {'word': top_words, 'count': [word_counts[word] for word in top_words]}
-        wordcount_df = pd.DataFrame(wordcount_data)
+        wordcloud_data = {'word': top_words, 'count': [word_counts[word] for word in top_words]}
+        wordcloud_df = pd.DataFrame(wordcloud_data)
         
-        fig = px.bar(wordcount_df, x='word', y='count', title="Word count")
+        fig = px.bar(wordcloud_df, x='word', y='count', title="Word Cloud")
         st.plotly_chart(fig)
 
 if __name__ == "__main__":