download visualization for pyLDA

pages/2 Topic Modeling.py

@@ -32,6 +32,8 @@ import sys
 import spacy
 import en_core_web_sm
 import pipeline
+import plotly.graph_objects as go
+from html2image import Html2Image
 
 
 #===config===
@@ -180,6 +182,26 @@ if uploaded_file is not None:
                         st.write('Coherence: ', (coherence_lda))
                         st.components.v1.html(py_lda_vis_html, width=1700, height=800)
                         st.markdown('Copyright (c) 2015, Ben Mabey. https://github.com/bmabey/pyLDAvis')
+                       
+                        @st.cache_data(ttl=3600, show_spinner=False)
+                        def img_lda(vis):
+                             pyLDAvis.save_html(vis, 'output.html')
+                             hti = Html2Image()
+                             hti.browser.flags = ['--default-background-color=ffffff', '--hide-scrollbars']
+                             css = "body {background: white;}"
+                             hti.screenshot(
+                                  other_file='output.html', css_str=css, size=(1500, 800),
+                                  save_as='ldavis_img.png'
+                             )
+                             with open("ldavis_img.png", "rb") as file:
+                              btn = st.download_button(
+                                  label="Download image",
+                                  data=file,
+                                  file_name="ldavis_img.png",
+                                  mime="image/png"
+                                  )
+                        img_lda(vis)    
+                       
                    except NameError:
                         st.warning('🖱️ Please click Submit')