Update app.py

app.py

@@ -28,7 +28,27 @@ def show_pdf(file_path):
     st.markdown(pdf_display, unsafe_allow_html=True)
 
 
-st.write(glob.glob(os.path.join(os.path.dirname(SigProfilerMatrixGenerator.__file__),'references/*txt')))
+@st.experimental_memo
+ def dl(file_vcf):
+     with open("output/SBS96/Suggested_Solution/COSMIC_SBS96_Decomposed_Solution/SBS96_Decomposition_Plots.pdf", "rb") as pdf_file:
+        PDFbyte1 = pdf_file.read()
+        st.download_button(label="Download image with single base substitution profiles", key='1',
+                            data=PDFbyte1,
+                            file_name="SBS.pdf",
+                            mime='application/octet-stream')
+
+
+    with open("output/ID83/Suggested_Solution/COSMIC_ID83_Decomposed_Solution/ID83_Decomposition_Plots.pdf", "rb") as pdf_file:
+        PDFbyte2 = pdf_file.read()
+        st.download_button(label="Download image with indel profiles", key='2',
+                            data=PDFbyte2,
+                            file_name="idel.pdf",
+                            mime='application/octet-stream')
+     return file_vcf
+
+
+
+#st.write(glob.glob(os.path.join(os.path.dirname(SigProfilerMatrixGenerator.__file__),'references/*txt')))
 
 if glob.glob(os.path.join(os.path.dirname(SigProfilerMatrixGenerator.__file__),'references/*txt')) == []:
     if st.button('get reference genome'):
@@ -66,20 +86,7 @@ if file_to_lookat !=[]:
         sig.sigProfilerExtractor("vcf", "output", "input", minimum_signatures=1, maximum_signatures=3)
    
   #show_pdf('output/SBS96/Suggested_Solution/COSMIC_SBS96_Decomposed_Solution/SBS96_Decomposition_Plots.pdf')
-    with open("output/SBS96/Suggested_Solution/COSMIC_SBS96_Decomposed_Solution/SBS96_Decomposition_Plots.pdf", "rb") as pdf_file:
-        PDFbyte1 = pdf_file.read()
-        st.download_button(label="Download image with single base substitution profiles", key='1',
-                            data=PDFbyte1,
-                            file_name="SBS.pdf",
-                            mime='application/octet-stream')
-
-
-    with open("output/ID83/Suggested_Solution/COSMIC_ID83_Decomposed_Solution/ID83_Decomposition_Plots.pdf", "rb") as pdf_file:
-        PDFbyte2 = pdf_file.read()
-        st.download_button(label="Download image with indel profiles", key='2',
-                            data=PDFbyte2,
-                            file_name="idel.pdf",
-                            mime='application/octet-stream')
+    dl(file_to_lookat)
     
         
     components.iframe("https://cancer.sanger.ac.uk/signatures/sbs/", height=3000,width=800)