Update app.py

app.py

@@ -31,13 +31,21 @@ top_words = word_frequency.most_common(10)
 top_words_dict = dict(top_words)
 
 # Create a Plotly bar chart to display the top words and their frequency
-fig = px.bar(x=list(top_words_dict.keys()), y=list(top_words_dict.values()), labels={'x':'Word', 'y':'Frequency'})
-st.plotly_chart(fig)
+fig_bar = px.bar(x=list(top_words_dict.keys()), y=list(top_words_dict.values()), labels={'x':'Word', 'y':'Frequency'})
+st.plotly_chart(fig_bar)
 
 # Calculate the scores for each word based on their length
 word_scores = {word:get_word_score(word) for word in word_frequency}
 top_word_scores = dict(sorted(word_scores.items(), key=lambda item: item[1], reverse=True)[:10])
 
 # Create a Plotly bar chart to display the top words and their scores
-fig = px.bar(x=list(top_word_scores.keys()), y=list(top_word_scores.values()), labels={'x':'Word', 'y':'Score'})
-st.plotly_chart(fig)
+fig_score = px.bar(x=list(top_word_scores.keys()), y=list(top_word_scores.values()), labels={'x':'Word', 'y':'Score'})
+st.plotly_chart(fig_score)
+
+# Create a treemap chart to display the word frequency
+fig_treemap = px.treemap(values=list(top_words_dict.values()), labels=list(top_words_dict.keys()))
+st.plotly_chart(fig_treemap)
+
+# Create a sunburst chart to display the word scores
+fig_sunburst = px.sunburst(values=list(top_word_scores.values()), labels=list(top_word_scores.keys()))
+st.plotly_chart(fig_sunburst)