Update app.py

app.py

@@ -258,17 +258,19 @@ 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='', z=1))
+                         text=words, customdata=sims, name=''))
 
-            # Represent query as a large red diamond
+            # Represent query as a large red diamond and label
             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))
+                          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))
+                         showlegend=False))
+
+            # Set the order of the traces
+            fig.update_layout(legend=dict(traceorder='reversed'), showlegend=True)
 
             fig.update(layout_coloraxis_showscale=True)
             fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF", margin=dict(t=0, b=0, l=0, r=0))
@@ -276,6 +278,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 "