Update src/streamlit_app.py

src/streamlit_app.py

@@ -209,6 +209,10 @@ with tab1:
                     st.subheader("Candidate Card", divider="green")
                     fig_treemap = px.treemap(df, path=[px.Constant("all"), 'category', 'label', 'text'], values='score', color='category')
                     fig_treemap.update_layout(margin=dict(t=50, l=25, r=25, b=25), paper_bgcolor='#F5FFFA', plot_bgcolor='#F5FFFA')
+                    expander = st.expander("**Download**")
+                    expander.write("""
+                    You can easily download the tree map by hovering over it. Look for the download icon that appears in the top right corner.
+                    """)
                     st.plotly_chart(fig_treemap)
                     col1, col2 = st.columns(2)
                     with col1:
@@ -248,6 +252,7 @@ with tab1:
                     buf = io.BytesIO()
                     with zipfile.ZipFile(buf, "w") as myzip:
                         myzip.writestr("Summary of the results.csv", df.to_csv(index=False))
+                        myzip.writestr("Most Frequent Entities.csv", repeating_entities.to_csv(index=False))
                         myzip.writestr("Glossary of tags.csv", dfa.to_csv(index=False))
                     st.download_button(
                         label="Download results and glossary (zip)",