Update app.py

app.py

@@ -96,7 +96,7 @@ if query:
     # table.head(10).to_csv("clotting_sim1.csv", index=True)
     # short_table = table.head(50)
     # print(table)
-    st.subheader(f"Top 10 Words closely related to {query}")
+
 
     # calculate the sizes of the squares in the treemap
     short_table = table.head(10)
@@ -115,11 +115,14 @@ if query:
     fig = plt.gcf()
     fig.patch.set_facecolor('#CCFFFF')
     # # display the treemap in Streamlit
-    st.pyplot(fig)
-    plt.clf()
+    treemap1, treemap2 = st.columns(2)
+    with treemap1:
+        st.subheader(f"Top 10 Words closely related to {query}")
+        st.pyplot(fig)
+        plt.clf()
 
-    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')
+        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')
 
     # st.write(short_table)
     #
@@ -136,7 +139,7 @@ if query:
     print()
     # df1.head(50).to_csv("clotting_sim2.csv", index=True, header=False)
     # time.sleep(2)
-    st.subheader(f"Top 10 Genes closely related to {query}")
+
 
     df10 = df1.head(10)
     df10.index = 1 / df10.index
@@ -157,12 +160,15 @@ if query:
     # plt.show()
     #
     # # display the treemap in Streamlit
-    st.pyplot(fig2)
+    with treemap2:
+        st.subheader(f"Top 10 Genes closely related to {query}")
+        st.pyplot(fig2)
 
-    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',
+        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',
                        mime='text/csv')
-
+    st.markdown("---")
+    st.subheader("Cancer-related videos")
     if query:
         idlist=[]
         search_keyword = {query}
@@ -189,27 +195,12 @@ if query:
         random.shuffle(video_ids)
 
         c1, c2, c3 = st.columns(3)
+
+
         with c1:
            st.video("https://www.youtube.com/watch?v=" + video_ids[0])
         with c2:
            st.video("https://www.youtube.com/watch?v=" + video_ids[1])
         with c3:
            st.video("https://www.youtube.com/watch?v=" + video_ids[2])
-            # width = 40
-            # side = 30  # (100-width)/2
-            # _, container, _ = st.columns([side, width, side])
-            # container.video(data=VIDEO_DATA)
-
-            # VIDEO_DATA = "https://www.youtube.com/watch?v=" + video_ids[1]
-            # width = 40
-            # side = 30  # (100-width)/2
-            # _, container, _ = st.columns([side, width, side])
-            # container.video(data=VIDEO_DATA)
-
-            # VIDEO_DATA = "https://www.youtube.com/watch?v=" + video_ids[2]
-            # width = 40
-            # side = 30  # (100-width)/2
-            # _, container, _ = st.columns([side, width, side])
-            # container.video(data=VIDEO_DATA)
-
-
+    st.markdown("---")