Update app.py

app.py

@@ -257,24 +257,18 @@ 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=''))
-
-            
-
-            
+                         marker=dict(size=2, color=sims_query_top, colorscale='RdYlGn', symbol='circle'),
+                         text=words, customdata=sims, name='', layer='below'))
 
             # 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))
-            # Set layer property of scatter trace with diamond to 'above'
-            fig.update_traces(mode='markers', marker=dict(layer='above'), selector=dict(type='scatter', symbol='diamond'))
+                          showlegend=False, layer='above'))
 
             # 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))
+                         textposition='top right', textfont=dict(color='blue', size=10), hoverinfo='none',
+                         showlegend=False))
 
             fig.update(layout_coloraxis_showscale=True)
             fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF", margin=dict(t=0, b=0, l=0, r=0))
@@ -282,6 +276,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 "