Update app.py

app.py

@@ -198,11 +198,11 @@ if query:
             pd.set_option('display.max_rows', None)
             table2 = table.copy()
 
-            # st.markdown(f"<b><p style='font-family: Arial; font-size: 20px;'>Populate a treemap to visualize "
-            #             f"<span style='color:red; font-style: italic;'>words</span> contextually "
-            #             f"and semantically similar to <span style='color:red; font-style: italic;'>{query}</span> "
-            #             f"within the <span style='color:red; font-style: italic;'>{database_name}</span> corpus.</p></b>",
-            #     unsafe_allow_html=True)
+            st.markdown(
+                f"<h2 style='text-align: center; font-family: Arial; font-size: 20px; font-weight: bold;'>"
+                f"Top <span style='color:red; font-style: italic;'>10000</span> words in an interactive embedding map most similar to <span style='color:red; font-style: italic;'>{query}</span> in <span style='color:red; font-style: italic;'>{database_name}</span> "
+                f"PubMed corpus: Zoom in to the black diamond to find <span style='color:red; font-style: italic;'>{query}</span></h2>",
+                unsafe_allow_html=True)
 
             # Set the max number of words to display
             value_word = min(100, len(table2))
@@ -232,10 +232,10 @@ if query:
                               yaxis=dict(gridcolor='#CCFFFF', color='blue'))
 
             # fig.update_traces(hovertemplate='<b>%{hovertext}</b><br>Similarity score: %{customdata[0]:.2f}<extra></extra>')
-            fig.update_layout(title=dict(
-                text=f"Top 10000 words in an interactive embedding map for {query} in {database_name} PubMed corpus"
-                     f": Zoom in to the black diamond to find {query}", x=0.5, y=1, xanchor='center', yanchor='top',
-                font=dict(color='black')))
+            # fig.update_layout(title=dict(
+                # text=f"Top 10000 words in an interactive embedding map for {query} in {database_name} PubMed corpus"
+                #      f": Zoom in to the black diamond to find {query}", x=0.5, y=1, xanchor='center', yanchor='top',
+                # font=dict(color='black')))
             fig.update_coloraxes(colorbar_title="Similarity with query")
 
             # Represent query as a large red diamond
@@ -465,6 +465,12 @@ if query:
             st.warning(
                 f"This selection exceeds the number of similar proteins related to {query} within the {database_name} corpus, please choose a lower number")
 
+        st.markdown(
+            f"<h2 style='text-align: center; font-family: Arial; font-size: 20px; font-weight: bold;'>3D interactive "
+            f"gene embedding map for <span style='color:red; font-style: italic;'>{value_gene}</span> genes most similar "
+            f"with <span style='color:red; font-style: italic;'>{query}</span> in <span style='color:red; font-style: italic;'>{database_name}</span> PubMed corpus</h2>",
+            unsafe_allow_html=True)
+
         try:
             # Get the top 50 similar genes to the query
             value_gene = min(df_len, 50)
@@ -499,7 +505,7 @@ if query:
         fig2.update_traces(
                 hovertemplate='<b>%{hovertext}</b><br>Similarity score: %{customdata[0]:.2f}<extra></extra>')
         fig2.update_layout(
-                title=dict(text=f"Word embedding map for {query} in {database_name} PubMed corpus", x=0.5, y=0.95,
+                title=dict(text=f"", x=0.5, y=0.95,
                            xanchor='center', yanchor='top', font=dict(color='black')),
                 scene=dict(xaxis_title="Dimension 1", yaxis_title="Dimension 2", zaxis_title="Dimension 3"))
         fig2.update_coloraxes(colorbar_title="Similarity with query")