Update app.py

app.py

@@ -14,7 +14,7 @@ import plotly.express as px
 st.set_page_config(
     page_title="FATA4 Science",
                 page_icon=":microscope:",
-                layout="wide",
+                layout="wide", #centered
                 initial_sidebar_state="auto",
                 menu_items={
                     'About': "FATA4 Science is a Natural Language Processing (NLP) that ...."
@@ -119,11 +119,23 @@ if query:
     fig = plt.gcf()
     fig.patch.set_facecolor('#CCFFFF')
     # # display the treemap in Streamlit
+
+    rank_num = list(short_table.index.tolist())
+    avg_size = sum(sizes) / len(short_table.index)
+    print(rank_num)
+    # print(sizes)
+    fig = px.treemap(short_table, path=[short_table.index], values=sizes, color=sizes, color_continuous_scale='greens',
+                     color_continuous_midpoint=avg_size)
+    fig.update(layout_coloraxis_showscale=False)
+    fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF")
+
+
     treemap1, treemap2 = st.columns(2)
     with treemap1:
         st.subheader(f"Top 10 Words closely related to {query}")
-        st.pyplot(fig)
-        plt.clf()
+        # st.pyplot(fig)
+        # plt.clf()
+        st.plotly_chart(fig, use_container_width=True)
 
         csv = table.head(100).to_csv().encode('utf-8')
         st.download_button(label="download top 100 words (csv)", data=csv, file_name=f'{database_name}_words.csv', mime='text/csv')
@@ -162,10 +174,27 @@ if query:
     fig2 = plt.gcf()
     fig2.patch.set_facecolor('#CCFFFF')
     #
+
+
+
+    # link_ref = '<a href="http://google.com" style="cursor: pointer" target="_blank" rel="noopener noreferrer">{}</a>'
+    # df10['SIMILARITY'] = df10['SIMILARITY'].apply(lambda item: link_ref.format(item, "{}"))
+    rank_num = list(df10.index.tolist())
+    avg_size = sum(sizes) / len(df10.index)
+    print(rank_num)
+    # print(sizes)
+    fig = px.treemap(path=[df10.index], values=sizes, color=sizes, color_continuous_scale='greens',
+                     color_continuous_midpoint=avg_size)
+    fig.update(layout_coloraxis_showscale=False)
+    fig.update_layout(autosize=True, paper_bgcolor="#CCFFFF", uniformtext_mode="hide", plot_bgcolor="#fff")
+    fig.update_traces(root_color='rgba(0,0,0,0)')
+
+
     # # display the treemap in Streamlit
     with treemap2:
         st.subheader(f"Top 10 Genes closely related to {query}")
-        st.pyplot(fig2)
+        # st.pyplot(fig2)
+        st.plotly_chart(fig, use_container_width=True)
 
         csv = df1.head(100).to_csv().encode('utf-8')
         st.download_button(label="download top 100 genes (csv)", data=csv, file_name=f'{database_name}_genes.csv',
@@ -208,6 +237,9 @@ if query:
            st.video("https://www.youtube.com/watch?v=" + video_ids[2])
     st.markdown("---")
 
+    # fig = plt.figure()
+
+