Update pages/2 Topic Modeling.py

pages/2 Topic Modeling.py

@@ -340,7 +340,7 @@ if uploaded_file is not None:
                            with st.spinner('Performing computations. Please wait ...'):
                                 fig1 = Vis_Topics(extype)
                                 st.write(fig1)
-                                img_bert(fig1)
+                                my_saved_image = img_bert(fig1)
                                 with open(my_saved_image, "rb") as file:
                                   btn = st.download_button(
                                        label="Download image",
@@ -353,7 +353,7 @@ if uploaded_file is not None:
                            with st.spinner('Performing computations. Please wait ...'):
                                 fig2 = Vis_Documents(extype)
                                 st.write(fig2)
-                                img_bert(fig2)
+                                my_saved_image = img_bert(fig2)
                                 with open(my_saved_image, "rb") as file:
                                   btn = st.download_button(
                                        label="Download image",
@@ -366,7 +366,7 @@ if uploaded_file is not None:
                            with st.spinner('Performing computations. Please wait ...'):
                                 fig3 = Vis_Hierarchy(extype)
                                 st.write(fig3)
-                                img_bert(fig3)
+                                my_saved_image = img_bert(fig3)
                                 with open(my_saved_image, "rb") as file:
                                   btn = st.download_button(
                                        label="Download image",
@@ -379,7 +379,7 @@ if uploaded_file is not None:
                            with st.spinner('Performing computations. Please wait ...'):
                                 fig4 = Vis_Heatmap(extype)
                                 st.write(fig4)
-                                img_bert(fig4)
+                                my_saved_image = img_bert(fig4)
                                 with open(my_saved_image, "rb") as file:
                                   btn = st.download_button(
                                        label="Download image",
@@ -392,7 +392,7 @@ if uploaded_file is not None:
                            with st.spinner('Performing computations. Please wait ...'):
                                 fig5 = Vis_Barchart(extype)
                                 st.write(fig5)
-                                img_bert(fig5)
+                                my_saved_image = img_bert(fig5)
                                 with open(my_saved_image, "rb") as file:
                                   btn = st.download_button(
                                        label="Download image",
@@ -405,7 +405,7 @@ if uploaded_file is not None:
                            with st.spinner('Performing computations. Please wait ...'):
                                 fig6 = Vis_ToT(extype)
                                 st.write(fig6)
-                                img_bert(fig6)
+                                my_saved_image = img_bert(fig6)
                                 with open(my_saved_image, "rb") as file:
                                   btn = st.download_button(
                                        label="Download image",
@@ -414,11 +414,11 @@ if uploaded_file is not None:
                                        mime="image/png"
                                        )
                     
-          #except ValueError:
-               #st.error('🙇‍♂️ Please raise the number of topics and click submit')
+          except ValueError:
+               st.error('🙇‍♂️ Please raise the number of topics and click submit')
           
-          except NameError:
-               st.warning('🖱️ Please click Submit')
+          #except NameError:
+               #st.warning('🖱️ Please click Submit')
 
         with tab2:
           st.markdown('**Grootendorst, M. (2022). BERTopic: Neural topic modeling with a class-based TF-IDF procedure. arXiv preprint arXiv:2203.05794.** https://doi.org/10.48550/arXiv.2203.05794')