Update app.py

app.py

@@ -253,25 +253,20 @@ 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='', z=1))
-
             # 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,
-                          showlegend=False, z=2))
+                                     marker=dict(size=12, color='black', symbol='diamond'), name=query, hovertext=query,
+                                     showlegend=False))
 
             # 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, z=3))
+                                     textposition='top right', textfont=dict(color='blue', size=10), hoverinfo='none',
+                                     showlegend=False))
 
-            # Set the order of the traces
-            fig.update_layout(legend=dict(traceorder='reversed'), showlegend=True)
+            # 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=''))
 
             fig.update(layout_coloraxis_showscale=True)
             fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF", margin=dict(t=0, b=0, l=0, r=0))