Add download buttons

app.py

@@ -447,6 +447,7 @@ if currFile is None and len(st.session_state.results) > 0 and len(st.session_sta
 try:
     st.session_state.resetResult = False
     currFileIndex = file_names.index(currFile)
+    currPlainName = currFile.split('.')[0]
     if len(st.session_state.results) > currFileIndex and len(st.session_state.summaries) > currFileIndex and len(st.session_state.results[currFileIndex]) > 0:
         st.header(f"Analysis of file {currFile}")
         graphNames = ["Data","Voice Categories","Speaker Percentage","Speakers with Categories","Treemap","Timeline","Time Spoken"]
@@ -503,7 +504,25 @@ try:
         df4.name = "df4"
         st.session_state.summaries[currFileIndex]["df4"] = df4
 
+        # Save files for potential download
+        
+        fig1.write_image("ascn_pie1.pdf")
+        fig2.write_image("ascn_pie2.pdf")
+        fig3_1.write_image("ascn_sunburst.pdf")
+        fig3.write_image("ascn_treemap.pdf")
+        fig_la.write_image("ascn_timeline.pdf")
+        fig2_la.write_image("ascn_bar.pdf")
+        
         with dataTab:
+            csv = convert_df(currDF)
+            
+            st.download_button(
+               "Press to Download analysis data",
+               csv,
+               "sonogram-analysis.csv",
+               "text/csv",
+               key='download-csv'
+            )
             st.dataframe(currDF)
         with pie1:
             df3 = st.session_state.summaries[currFileIndex]["df3"]
@@ -515,6 +534,24 @@ try:
             paper_bgcolor='rgba(0, 0, 0, 0)',
             )
             fig1.add_trace(go.Pie(values=df3["values"],labels=df3["names"],sort=False))
+            
+            col1, col2 = st.columns(2)
+            fig1.write_image("ascn_pie1.pdf")
+            fig1.write_image("ascn_pie1.svg")
+            with open('ascn_pie1.pdf') as f:
+                col1.dowload_botton(
+                    "Save As PDF",
+                    f,
+                    'sonogram-voice-category-'+currPlainName+'.pdf',
+                    key='download-pdf1'
+                )
+            with open('ascn_pie1.svg') as f:
+                col2.dowload_botton(
+                    "Save As SVG",
+                    f,
+                    'sonogram-voice-category-'+currPlainName+'.svg',
+                    key='download-svg1'
+                )
             st.plotly_chart(fig1, use_container_width=True)
 
         with pie2:
@@ -533,6 +570,24 @@ try:
                 paper_bgcolor='rgba(0, 0, 0, 0)',
             )
             fig2.add_trace(go.Pie(values=df4["values"],labels=df4["names"],sort=False))
+
+            col1, col2 = st.columns(2)
+            fig2.write_image("ascn_pie2.pdf")
+            fig2.write_image("ascn_pie2.svg")
+            with open('ascn_pie2.pdf') as f:
+                col1.dowload_botton(
+                    "Save As PDF",
+                    f,
+                    'sonogram-speaker-percent-'+currPlainName+'.pdf',
+                    key='download-pdf2'
+                )
+            with open('ascn_pie2.svg') as f:
+                col2.dowload_botton(
+                    "Save As SVG",
+                    f,
+                    'sonogram-speaker-percent-'+currPlainName+'.svg',
+                    key='download-svg2'
+                )
             st.plotly_chart(fig2, use_container_width=True)
 
         with sunburst1:
@@ -561,6 +616,24 @@ try:
                 plot_bgcolor='rgba(0, 0, 0, 0)',
                 paper_bgcolor='rgba(0, 0, 0, 0)',
             )
+            
+            col1, col2 = st.columns(2)
+            fig3_1.write_image("ascn_sunburst.pdf")
+            fig3_1.write_image("ascn_sunburst.svg")
+            with open('ascn_sunburst.pdf') as f:
+                col1.dowload_botton(
+                    "Save As PDF",
+                    f,
+                    'sonogram-speaker-categories-'+currPlainName+'.pdf',
+                    key='download-pdf3'
+                )
+            with open('ascn_sunburst.svg') as f:
+                col2.dowload_botton(
+                    "Save As SVG",
+                    f,
+                    'sonogram-speaker-categories-'+currPlainName+'.svg',
+                    key='download-svg3'
+                )
             st.plotly_chart(fig3_1, use_container_width=True)
     
         with treemap1:
@@ -589,6 +662,24 @@ try:
                 plot_bgcolor='rgba(0, 0, 0, 0)',
                 paper_bgcolor='rgba(0, 0, 0, 0)',
             )
+                        
+            col1, col2 = st.columns(2)
+            fig3.write_image("ascn_treemap.pdf")
+            fig3.write_image("ascn_treemap.svg")
+            with open('ascn_treemap.pdf') as f:
+                col1.dowload_botton(
+                    "Save As PDF",
+                    f,
+                    'sonogram-treemap-'+currPlainName+'.pdf',
+                    key='download-pdf4'
+                )
+            with open('ascn_treemap.svg') as f:
+                col2.dowload_botton(
+                    "Save As SVG",
+                    f,
+                    'sonogram-treemap-'+currPlainName+'.svg',
+                    key='download-svg4'
+                )
             st.plotly_chart(fig3, use_container_width=True)
        
         # generate plotting window
@@ -619,7 +710,24 @@ try:
                 plot_bgcolor='rgba(0, 0, 0, 0)',
                 paper_bgcolor='rgba(0, 0, 0, 0)',
             )
-            
+                                    
+            col1, col2 = st.columns(2)
+            fig_la.write_image("ascn_timeline.pdf")
+            fig_la.write_image("ascn_timeline.svg")
+            with open('ascn_timeline.pdf') as f:
+                col1.dowload_botton(
+                    "Save As PDF",
+                    f,
+                    'sonogram-timeline-'+currPlainName+'.pdf',
+                    key='download-pdf5'
+                )
+            with open('ascn_timeline.svg') as f:
+                col2.dowload_botton(
+                    "Save As SVG",
+                    f,
+                    'sonogram-timeline-'+currPlainName+'.svg',
+                    key='download-svg5'
+                )
             st.plotly_chart(fig_la, use_container_width=True)
     
         with bar1:
@@ -643,6 +751,24 @@ try:
                     'Percentage of Time: %{customdata[1]:.2f}%'
                 ])
             )
+                                                
+            col1, col2 = st.columns(2)
+            fig_la.write_image("ascn_bar.pdf")
+            fig_la.write_image("ascn_bar.svg")
+            with open('ascn_bar.pdf') as f:
+                col1.dowload_botton(
+                    "Save As PDF",
+                    f,
+                    'sonogram-speaker-time-'+currPlainName+'.pdf',
+                    key='download-pdf6'
+                )
+            with open('ascn_bar.svg') as f:
+                col2.dowload_botton(
+                    "Save As SVG",
+                    f,
+                    'sonogram-speaker-time-'+currPlainName+'.svg',
+                    key='download-svg6'
+                )
             st.plotly_chart(fig2_la, use_container_width=True)
 
 except ValueError:
@@ -729,13 +855,3 @@ def convert_df(df):
    return df.to_csv(index=False).encode('utf-8')
 
 
-if currDF is not None:
-    csv = convert_df(currDF)
-    
-    st.download_button(
-       "Press to Download analysis data",
-       csv,
-       "file.csv",
-       "text/csv",
-       key='download-csv'
-    )