deepakri201 commited on
Commit
b8d3c98
·
1 Parent(s): a88029f

adding use of papermill to run notebook

Browse files
Files changed (1) hide show
  1. app.py +24 -7
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
- # with open(parquet_file, "rb") as f:
88
- # st.download_button(
89
- # label="Download the metadata parquet file",
90
- # data=f,
91
- # file_name="dcm2parquet_output.parquet",
92
- # mime="application/parquet"
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
+