Update app.py

app.py

@@ -258,16 +258,17 @@ if 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=''))
+                         text=words, customdata=sims, name='', z=1))
 
-            # Represent query as a large red diamond and label
+            # 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))
+                          showlegend=False, z=2))
 
+            # 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))
+                         showlegend=False, z=3))
 
             # Set the order of the traces
             fig.update_layout(legend=dict(traceorder='reversed'), showlegend=True)
@@ -279,6 +280,7 @@ if query:
             st.plotly_chart(fig, use_container_width=True)
 
 
+
             st.markdown(
             f"<b><p style='font-family: Arial; font-size: 20px; font-style: Bold;'>Top <span style='color:red; font-style: italic;'>{value_word} "
             f"</span>words contextually and semantically similar to "