Update src/streamlit_app.py

src/streamlit_app.py

@@ -218,11 +218,19 @@ with tab1:
                         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')
                         fig_pie.update_layout(paper_bgcolor='#F5FFFA', plot_bgcolor='#F5FFFA')
+                        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="green")
                         fig_bar = px.bar(grouped_counts, x="count", y="category", color="category", text_auto=True, title='Occurrences of predicted categories')
                         fig_bar.update_layout(paper_bgcolor='#F5FFFA', plot_bgcolor='#F5FFFA')
+                        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)
                     st.subheader("Most Frequent Entities", divider="green")
                     word_counts = df['text'].value_counts().reset_index()
@@ -325,7 +333,7 @@ with tab2:
                         st.subheader("Extracted Answers", divider="green")
                         expander = st.expander("**Download**")
                         expander.write("""
-                         To download the data, simply hover your cursor over the table. A download icon will appear on the right side.
+                        To download the data, simply hover your cursor over the table. A download icon will appear in the top right corner.
                         """)
                         st.dataframe(df, use_container_width=True)
                         st.subheader("Tree map", divider="green")
@@ -333,6 +341,10 @@ with tab2:
                         label_color_map = {label: get_stable_color(label) for label in all_labels}
                         fig_treemap = px.treemap(df, path=[px.Constant("all"), 'question', 'answer'], values='score', color='question', color_discrete_map=label_color_map)
                         fig_treemap.update_layout(margin=dict(t=50, l=25, r=25, b=25), paper_bgcolor='#F3E5F5', plot_bgcolor='#F3E5F5')
+                        expander = st.expander("**Download**")
+                        expander.write("""
+                        You can easily download the treemap by hovering over it. Look for the download icon that appears in the top right corner.
+                        """)
                         st.plotly_chart(fig_treemap)
                         if comet_initialized:
                             experiment.log_metric("processing_time_seconds", elapsed_time)