Update app.py

app.py

@@ -41,7 +41,7 @@ if 'words' not in st.session_state:
     st.session_state['words'] = []
 
 if 'words_umap_df' not in st.session_state:
-    st.session_state['words_umap_df'] = pd.DataFrame({
+    words_umap_df = pd.DataFrame({
         "x": [],
         "y": [],
         "z": [],
@@ -49,8 +49,6 @@ if 'words_umap_df' not in st.session_state:
         "s": [],
         "l": [],
     })
-    words_umap_df = st.session_state['words_umap_df']
-
     secret_embedding_3d = umap_3d.transform([secret_embedding])[0]
     words_umap_df.loc[len(words_umap_df)] = {
         "x": secret_embedding_3d[0],
@@ -60,6 +58,9 @@ if 'words_umap_df' not in st.session_state:
         "s": 10,
         "l": "Secret word"
     }
+    st.session_state['words_umap_df'] = words_umap_df
+
+
 
 
 st.write('Try to guess a secret word by semantic similarity')
@@ -78,7 +79,8 @@ if st.button("Guess") or word:
         st.session_state['words'].append((str(word), similarity))
 
         pt = umap_3d.transform([word_embedding])[0]
-        st.session_state['words_umap_df'].loc[len(st.session_state['words_umap_df'])] = {
+        words_umap_df = st.session_state['words_umap_df']
+        words_umap_df.loc[len(words_umap_df)] = {
             "x": pt[0],
             "y": pt[1],
             "z": pt[2],
@@ -86,6 +88,7 @@ if st.button("Guess") or word:
             "s": 3,
             "l": str(word)
         }
+        st.session_state['words_umap_df'] = words_umap_df
 
 words_df = pd.DataFrame(
     st.session_state['words'],
@@ -94,5 +97,6 @@ words_df = pd.DataFrame(
 st.dataframe(words_df, use_container_width=True)
 
 
-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))
+words_umap_df = st.session_state['words_umap_df']
+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))
 st.plotly_chart(fig_3d, theme="streamlit", use_container_width=True)