Commit
·
b8d3c98
1
Parent(s):
a88029f
adding use of papermill to run notebook
Browse files
app.py
CHANGED
@@ -11,6 +11,8 @@ from pathlib import Path
|
|
11 |
import polars
|
12 |
import pydicom.datadict as dd
|
13 |
import shutil
|
|
|
|
|
14 |
|
15 |
|
16 |
# Main Streamlit app code
|
@@ -84,10 +86,25 @@ if st.button("Process IDC data"):
|
|
84 |
# save_dicom_header_to_parquet(dicom_files, parquet_file)
|
85 |
|
86 |
st.success("Processing complete.")
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
import polars
|
12 |
import pydicom.datadict as dd
|
13 |
import shutil
|
14 |
+
import papermill as pm
|
15 |
+
import subprocess
|
16 |
|
17 |
|
18 |
# Main Streamlit app code
|
|
|
86 |
# save_dicom_header_to_parquet(dicom_files, parquet_file)
|
87 |
|
88 |
st.success("Processing complete.")
|
89 |
+
|
90 |
+
|
91 |
+
if os.path.exists("DICOMScanClassification_user_demo.ipynb"):
|
92 |
+
os.remove("DICOMScanClassification_user_demo.ipynb")
|
93 |
+
|
94 |
+
subprocess.run(["wget https://raw.githubusercontent.com/deepakri201/DICOMScanClassification_pw41/main/DICOMScanClassification_user_demo.ipynb"])
|
95 |
+
|
96 |
+
pm.execute_notebook(
|
97 |
+
"DICOMScanClassification_user_demo.ipynb",
|
98 |
+
'output.ipynb',
|
99 |
+
parameters = dict(SeriesInstanceUID=series_instance_uids)
|
100 |
+
)
|
101 |
+
|
102 |
+
with open('output.ipynb', "rb") as f:
|
103 |
+
st.download_button(
|
104 |
+
label="Download the output notebook file",
|
105 |
+
data=f,
|
106 |
+
file_name="output.ipynb",
|
107 |
+
mime="application/json"
|
108 |
+
)
|
109 |
+
|
110 |
+
|