Update app.py

app.py

@@ -34,10 +34,12 @@ st.markdown("""
     body {
         background-color: #CCFFFF;
         # color: #ffffff;
+        # font-size: 1px
     }
     .stApp {
         background-color: #CCFFFF;
         # color: #ffffff;
+        # font-size: 1px
     }
     </style>
     """, unsafe_allow_html=True)
@@ -123,16 +125,23 @@ if query:
     # print(table.head(10)["SIMILARITY"])
     # # display the treemap in Streamlit
     table2["SIMILARITY"] = 'Similarity Score ' + table2.head(10)["SIMILARITY"].round(2).astype(str)
-    # rank_num = list(short_table.index.tolist())
+    rank_num = list(short_table.index.tolist())
     # avg_size = sum(sizes) / len(short_table.index)
-    # print(rank_num)
+    print(rank_num)
     # print(sizes)
-    fig = px.treemap(path=[short_table.index], values=sizes, hover_name=(table2.head(10)['SIMILARITY']))
+    # '{0} in {1}'.format(unicode(self.author, 'utf-8'), unicode(self.publication, 'utf-8'))
+    TEMPLATE = """
+            <br>
+            {0}: <a href='https://pubmed.ncbi.nlm.nih.gov/?term={1}%5Bmh%5D+%20%20%20%20%20NOT
+            +review%5Bpt%5D+AND+english%5Bla%5D+AND+hasabstract+AND+1990%253A2022%252F12%252F31%5Bdp%5D+AND+%22{2}%22'>google</a>
+            """.format(database_name,database_name, database_name)
+    fig = px.treemap(names=rank_num, path=[short_table.index], values=sizes, hover_name=(table2.head(10)['SIMILARITY']))
+
     fig.update(layout_coloraxis_showscale=False)
     fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF")
     fig.update_annotations(visible=False)
     fig.update_traces(marker=dict(cornerradius=5), root_color="#CCFFFF", hovertemplate=None,
-                      hoverlabel_bgcolor="lightgreen", hoverlabel_bordercolor="#000000")
+                      hoverlabel_bgcolor="lightgreen", hoverlabel_bordercolor="#000000", texttemplate=TEMPLATE)
     fig.update_layout(uniformtext=dict(minsize=15, mode='hide'), treemapcolorway=["lightgreen"])
 
 
@@ -181,19 +190,43 @@ if query:
     fig2 = plt.gcf()
     fig2.patch.set_facecolor('#CCFFFF')
     #
-    print(df10["SIMILARITY"])
-    # rank_num = list(df10.index.tolist())
-    # avg_size = sum(sizes) / len(df10.index)
-    df10["SIMILARITY"] = 'Similarity Score ' + df10["SIMILARITY"].round(2).astype(str)
-    fig = px.treemap(path=[df10.index], values=sizes, hover_name=(df10['SIMILARITY']))
+    df3 = df1.copy()
+    df3.reset_index(inplace=True)
+    df3 = df3.rename(columns={'Human Gene': 'symbol2'})
+    # Use df.query to get a subset of df1 based on ids in df2
+    subset = df3.head(10).query('symbol2 in @df2.symbol2')
+    # Use merge to join the two DataFrames on id
+    result = pd.merge(subset, df2, on='symbol2')
+    # Show the result
+    # print(result)
+
+    df = df10
+
+    # Define the `text` column for labels and `href` column for links
+    df['text'] = df10.index
+    df['href'] = [f'https://pubmed.ncbi.nlm.nih.gov/?term={database_name}%5Bmh%5D+NOT+review%5Bpt%5D' \
+                  '+AND+english%5Bla%5D+AND+hasabstract+AND+1990:2022%5Bdp%5D+AND+' + c for c in df10.index]
+    df['href2'] = [f'https://www.ncbi.nlm.nih.gov/gene/?term=' + c for c in df10.index]
+    df['name'] = [c for c in result['Approved name']]
+    df['database'] = database_name
+    # print(df['name'])
+
+    # Create the treemap using `px.treemap`
+    fig = px.treemap(df, path=[df10.index], values=sizes, hover_data=['SIMILARITY'],
+                     custom_data=['href', 'name', 'database', 'href2'])
+    # fig.update_traces(texttemplate="Gene: <b>%{label}</b><br>Gene "
+    #                                "name: %{customdata[1]}<br><a href='%{customdata[0]}'>PubMed Abstracts with %{label} "
+    #                                "in %{customdata[2]}</a><br><a href='%{customdata[3]}'>NCBI gene information for %{label} "
+    #                                "in %{customdata[2]}</a>")
     fig.update(layout_coloraxis_showscale=False)
     fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF")
     fig.update_annotations(visible=False)
     fig.update_traces(marker=dict(cornerradius=5), root_color="#CCFFFF", hovertemplate=None,
-                      hoverlabel_bgcolor="lightblue", hoverlabel_bordercolor="#000000")
-    fig.update_layout(uniformtext=dict(minsize=20, mode='hide'), treemapcolorway=["lightblue"])
-
-
+                      hoverlabel_bgcolor="lightblue", hoverlabel_bordercolor="#000000",
+                      texttemplate="<b>%{label}</b><br>%{customdata[1]}<br><a href='%{customdata[0]}'>PubMed"
+                                   "</a><br><a href='%{customdata[3]}'>NCBI"
+                                   "</a>")
+    fig.update_layout(uniformtext=dict(minsize=10, mode='hide'), treemapcolorway=["lightblue"])
     # # display the treemap in Streamlit
     # with treemap2:
     st.subheader(f"Top 10 Genes closely related to {query}")
@@ -241,7 +274,8 @@ if query:
            st.video("https://www.youtube.com/watch?v=" + video_ids[2])
     st.markdown("---")
 
-    # fig = plt.figure()
+
+