Update apps/intro.py

apps/intro.py

@@ -9,9 +9,13 @@ import joblib
 
 def app():
     word_to_embed_list =  st.session_state['embed_list']
+    cat_list =  st.session_state['cat_list']
+    
     with st.container():
         word_to_embed= st.text_input("Please enter your text here and we will embed it for you.",
                                          value="Woman",)
+        cat= st.selectbox('Categorie',  ('1', '2', '3', '4', '5'))
+
         
         if st.button("Embed"):
             with st.spinner("👑 load language model (sentence transformer)"):
@@ -23,6 +27,8 @@ def app():
                 
                 word_to_embed_list.append(word_to_embed)
                 st.session_state['embed_list'] = word_to_embed_list
+                cat_list .append(cat)
+                st.session_state['cat_list '] = cat_list 
                 
                 examples_embeddings = model.encode(word_to_embed_list)
  
@@ -33,7 +39,7 @@ def app():
                 with st.spinner("👑 create visualisation"):  
                       fig = px.scatter_3d(
                           examples_umap[1:] , x=0, y=1, z=2,
-                           # color='labels',
+                           color=cat_list[1:] ,
                           opacity = .7,    hover_data=[word_to_embed_list[1:]])
                       fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
                       fig.update_traces(marker_size=4)