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 |
+
|