Update app.py

app.py

@@ -253,6 +253,13 @@ if query:
                 # font=dict(color='black')))
             fig.update_coloraxes(colorbar_title=f"Similarity with {query}")
 
+            
+
+            # Add circles for the top 50 similar words
+            fig.add_trace(go.Scatter(x=X_top[:, 0], y=X_top[:, 1], mode='markers',
+                                     marker=dict(size=2, color=sims_query_top, colorscale='RdYlGn', symbol='circle'),
+                                     text=words, customdata=sims, name=''))
+
             # Represent query as a large red diamond
             fig.add_trace(go.Scatter(x=[model.wv[query][0]], y=[model.wv[query][1]], mode='markers',
                                      marker=dict(size=12, color='black', symbol='diamond'), name=query, hovertext=query,
@@ -261,12 +268,7 @@ if query:
             # Add label for the query above the diamond
             fig.add_trace(go.Scatter(x=[model.wv[query][0]], y=[model.wv[query][1]], mode='text', text=[query],
                                      textposition='top right', textfont=dict(color='blue', size=10), hoverinfo='none',
-                                     showlegend=False))
-
-            # Add circles for the top 50 similar words
-            fig.add_trace(go.Scatter(x=X_top[:, 0], y=X_top[:, 1], mode='markers',
-                                     marker=dict(size=2, color=sims_query_top, colorscale='RdYlGn', symbol='circle'),
-                                     text=words, customdata=sims, name=''))
+                                     showlegend=False, layer='above traces'))
 
             fig.update(layout_coloraxis_showscale=True)
             fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF", margin=dict(t=0, b=0, l=0, r=0))