Update app.py

app.py

@@ -61,7 +61,6 @@ if 'words_umap_df' not in st.session_state:
         "l": "Secret word"
     }
 
-words_umap_df = st.session_state['words_umap_df']
 
 st.write('Try to guess a secret word by semantic similarity')
 
@@ -79,7 +78,7 @@ if st.button("Guess") or word:
         st.session_state['words'].append((str(word), similarity))
 
         pt = umap_3d.transform([word_embedding])[0]
-        words_umap_df.loc[len(words_umap_df)] = {
+        st.session_state['words_umap_df'].loc[len(st.session_state['words_umap_df'])] = {
             "x": pt[0],
             "y": pt[1],
             "z": pt[2],
@@ -95,5 +94,5 @@ words_df = pd.DataFrame(
 st.dataframe(words_df, use_container_width=True)
 
 
-fig_3d = px.scatter_3d(words_umap_df, x="x", y="y", z="z", color="similarity", hover_name="l", hover_data={"x": False, "y": False, "z": False, "s": False}, size="s", size_max=10, range_color=(0,1))
+fig_3d = px.scatter_3d(st.session_state['words_umap_df'], x="x", y="y", z="z", color="similarity", hover_name="l", hover_data={"x": False, "y": False, "z": False, "s": False}, size="s", size_max=10, range_color=(0,1))
 st.plotly_chart(fig_3d, theme="streamlit", use_container_width=True)