Update app.py

app.py

@@ -55,7 +55,7 @@ if query:
     print("Similarity to " + str(query))
     pd.set_option('display.max_rows', None)
     print(table.head(50))
-    table.head(10).to_csv("clotting_sim1.csv", index=True)
+    # table.head(10).to_csv("clotting_sim1.csv", index=True)
     # short_table = table.head(50)
     # print(table)
     st.subheader(f"Similar Words to {query}")
@@ -79,6 +79,13 @@ if query:
     st.pyplot(fig)
     plt.clf()
 
+    csv = table.head(100)
+    st.download_button(
+        label="download top 100 words (csv)",
+        data=csv,
+        file_name='clotting_words.csv',
+        mime='text/csv')
+
     # st.write(short_table)
     #
 
@@ -92,13 +99,13 @@ if query:
     df1["Human Gene"] = df1["Human Gene"].str.upper()
     print(df1.head(50))
     print()
-    df1.head(50).to_csv("clotting_sim2.csv", index=True, header=False)
+    # df1.head(50).to_csv("clotting_sim2.csv", index=True, header=False)
     # time.sleep(2)
     st.subheader(f"Similar Genes to {query}")
 
-    df1 = df1.head(10)
-    df1.index = 1/df1.index
-    sizes = df1.index.tolist()
+    df10 = df1.head(10)
+    df10.index = 1/df10.index
+    sizes = df10.index.tolist()
 
     cmap2 = plt.cm.Blues(np.linspace(0.05, .5, len(sizes)))
     color2 = [cmap2[i] for i in range(len(sizes))]
@@ -116,6 +123,13 @@ if query:
     # # display the treemap in Streamlit
     st.pyplot(fig2)
 
+    csv = df1.head(100)
+    st.download_button(
+        label="download top 100 genes (csv)",
+        data=csv,
+        file_name='clotting_genes.csv',
+        mime='text/csv')
+