Update app.py

app.py

@@ -52,8 +52,6 @@ def main():
     st.sidebar.subheader("Actions")
     if st.sidebar.button("Home"):
         display_home()
-    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,38 +85,35 @@ def display_table(csv_df):
     )
     st.dataframe(styled_df)
 
-def display_word_cloud():
-    st.title("Word Cloud")
+    if st.button("Word Count"):
+        display_word_count(filtered_df)
+
+def display_word_count(df):
+    st.title("Word Count")
     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]
-    selected_file = st.sidebar.selectbox("Select a File", selected_files, key="word_count_file_select")
-    
-    file_url = f"https://raw.githubusercontent.com/halimbahae/Hadith/main/{selected_book}/{selected_file}"
-    csv_df = pd.read_csv(file_url, header=None) 
-    
-    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 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]
-        
-        wordcloud_data = {'word': top_words, 'count': [word_counts[word] for word in top_words]}
-        wordcloud_df = pd.DataFrame(wordcloud_data)
-        
-        fig = px.bar(wordcloud_df, x='word', y='count', title="Word Cloud")
-        st.plotly_chart(fig)
+    word_count = st.slider("Select Number of Words", min_value=10, max_value=50, value=10, key="word_count_slider")
+
+    words_to_exclude = st.text_input("Words to Exclude (comma-separated)", "", key="exclude_words_input")
+    words_to_exclude = [word.strip() for word in words_to_exclude.split(",")]
+
+    word_counts = {}
+    for row in df.itertuples(index=False):
+        for word in str(row[1]).split():
+            if word not in words_to_exclude:
+                if word in word_counts:
+                    word_counts[word] += 1
+                else:
+                    word_counts[word] = 1
+
+    sorted_word_counts = sorted(word_counts.items(), key=lambda x: x[1], reverse=True)[:word_count]
+    words, counts = zip(*sorted_word_counts)
+
+    wordcloud_data = {'word': words, 'count': counts}
+    wordcloud_df = pd.DataFrame(wordcloud_data)
+
+    fig = px.bar(wordcloud_df, x='word', y='count', title="Word Cloud")
+    st.plotly_chart(fig)
 
 if __name__ == "__main__":
     main()