import plotly.express as px
import streamlit as st
from sentence_transformers import SentenceTransformer
import umap.umap_ as umap
import pandas as pd
import os

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'])
    
    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)