Update src/streamlit_app.py

src/streamlit_app.py

@@ -48,7 +48,7 @@ with st.sidebar:
     st.code(code, language="html")
     st.text("")
     st.text("")
-    st.divider()
+    
     st.subheader("🚀 Ready to build your own AI Web App?", divider="violet")
     st.link_button("AI Web App Builder", "https://nlpblogs.com/build-your-named-entity-recognition-app/", type="primary")
 
@@ -179,6 +179,9 @@ if st.session_state.show_results:
         st.subheader("Tree map", divider="violet")
         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))
+        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)
 
         # Pie and Bar charts
@@ -190,11 +193,17 @@ if st.session_state.show_results:
             st.subheader("Pie chart", divider="violet")
             fig_pie = px.pie(grouped_counts, values='count', names='category', hover_data=['count'], labels={'count': 'count'}, title='Percentage of predicted categories')
             fig_pie.update_traces(textposition='inside', textinfo='percent+label')
+            expander = st.expander("**Download**")
+            expander.write("""You can easily download the pie chart by hovering over it. Look for the download icon that appears in the top right corner.
+            """)
             st.plotly_chart(fig_pie)
             
         with col2:
             st.subheader("Bar chart", divider="violet")
             fig_bar = px.bar(grouped_counts, x="count", y="category", color="category", text_auto=True, title='Occurrences of predicted categories')
+            expander = st.expander("**Download**")
+            expander.write("""You can easily download the bar chart by hovering over it. Look for the download icon that appears in the top right corner.
+            """)
             st.plotly_chart(fig_bar)
 
         # Most Frequent Entities
@@ -207,6 +216,9 @@ if st.session_state.show_results:
             st.dataframe(repeating_entities, use_container_width=True)
             fig_repeating_bar = px.bar(repeating_entities, x='Entity', y='Count', color='Entity')
             fig_repeating_bar.update_layout(xaxis={'categoryorder': 'total descending'})
+            expander = st.expander("**Download**")
+            expander.write("""You can easily download the bar chart by hovering over it. Look for the download icon that appears in the top right corner.
+            """)
             st.plotly_chart(fig_repeating_bar)
         else:
             st.warning("No entities were found that occur more than once.")
@@ -219,6 +231,7 @@ if st.session_state.show_results:
         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))
         
         with stylable_container(