Update apps/sdg_pd.py

apps/sdg_pd.py

@@ -9,26 +9,29 @@ def app():
     st.title("SDG Embedding Visualisation")
     
     with st.spinner("👑 load data"):
-      df_osdg = pd.read_csv("sdg_umap.csv", sep = "|")
-                   
-      #labels = [_lab_dict[lab] for lab in df_osdg['label'] ]
-      keys = list(df_osdg['keys'])
-      #docs = list(df_osdg['text'])
+          df_osdg = pd.read_csv("sdg_umap.csv", sep = "|")
+                       
+          #labels = [_lab_dict[lab] for lab in df_osdg['label'] ]
+          keys = list(df_osdg['keys'])
+          #docs = list(df_osdg['text'])
+          
+          agree = st.checkbox('add labels')
     
-    with st.spinner("👑 create visualisation"):  
-      fig = px.scatter_3d(
-          df_osdg, x='coord_x', y='coord_y', z='coord_z',  
-          opacity = .5,    hover_data=[keys])
-      fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
-      fig.update_traces(marker_size=4)
-      st.plotly_chart(fig)
+          if agree:
       
-    if st.button("Embed"):
-        with st.spinner("👑 create visualisation"):  
-          fig = px.scatter_3d(
-              df_osdg, x='coord_x', y='coord_y', z='coord_z',
-        color='labels',
-              opacity = .5,    hover_data=[keys])
-          fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
-          fig.update_traces(marker_size=4)
-          st.plotly_chart(fig)
+            with st.spinner("👑 create visualisation"):  
+              fig = px.scatter_3d(
+                  df_osdg, x='coord_x', y='coord_y', z='coord_z',
+            color='labels',
+                  opacity = .5,    hover_data=[keys])
+              fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
+              fig.update_traces(marker_size=4)
+              st.plotly_chart(fig)
+          else:
+              with st.spinner("👑 create visualisation"):  
+              fig = px.scatter_3d(
+                  df_osdg, x='coord_x', y='coord_y', z='coord_z',  
+                  opacity = .5,    hover_data=[keys])
+              fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
+              fig.update_traces(marker_size=4)
+              st.plotly_chart(fig)