adding button to run inference

app.py

@@ -54,56 +54,29 @@ selected_series = st.selectbox("Select Series", series)
 
 print('selected_series: ' + str(selected_series))
 
+# Button to run the notebook - which loads the pretrained models and runs inference 
+if st.button("Run inference"):
 
-# # Button to process IDC data
-# if st.button("Process IDC data"):
-#     # Fetch data from IDC based on selection
-#     selection = index_df[
-#         (index_df["SeriesInstanceUID"] == selected_series)
-#     ]
-
-#     series_instance_uids = selection["SeriesInstanceUID"].tolist()
-
-#     #    with TemporaryDirectory() as temp_dir:
-#     download_errors = []
-#     #input_dir = os.path.join(temp_dir, "input_data")
-#     input_dir=Path("input_data/")
-#     if input_dir.exists():
-#         shutil.rmtree(input_dir)
-#     os.makedirs(input_dir, exist_ok=True)
-
-#     try:
-#         client.download_from_selection(seriesInstanceUID=series_instance_uids, downloadDir=input_dir)
-#     except Exception as e:
-#         download_errors.append(f"Error downloading data: {str(e)}")
-
-#     if download_errors:
-#         st.error("\n".join(download_errors))
-#     else:
-#         st.success("Data downloaded successfully.")
-
-#     # Process downloaded DICOM data
-#     dicom_files = [str(file) for file in input_dir.glob('**/*.dcm')]
-#     # parquet_file = 'dcm2parquet_output.parquet'
-#     # save_dicom_header_to_parquet(dicom_files, parquet_file)
-
-if os.path.exists("DICOMScanClassification_user_demo.ipynb"):
-    os.remove("DICOMScanClassification_user_demo.ipynb")
-
-subprocess.run(["wget", "https://raw.githubusercontent.com/deepakri201/DICOMScanClassification_pw41/main/DICOMScanClassification_user_demo.ipynb"])
-
-pm.execute_notebook(
-        "DICOMScanClassification_user_demo.ipynb",
-        'output.ipynb',
-        parameters = dict(SeriesInstanceUID=selected_series)
-)
-
-with open('output.ipynb', "rb") as f:
-    st.download_button(
-        label="Download the output notebook file",
-        data=f,
-        file_name="output.ipynb",
-        mime="application/json"
+    # Code to run when the button is pressed
+    st.write("Button pressed! Running inference")
+
+    if os.path.exists("DICOMScanClassification_user_demo.ipynb"):
+        os.remove("DICOMScanClassification_user_demo.ipynb")
+
+    subprocess.run(["wget", "https://raw.githubusercontent.com/deepakri201/DICOMScanClassification_pw41/main/DICOMScanClassification_user_demo.ipynb"])
+
+    pm.execute_notebook(
+            "DICOMScanClassification_user_demo.ipynb",
+            'output.ipynb',
+            parameters = dict(SeriesInstanceUID=selected_series)
     )
 
+    with open('output.ipynb', "rb") as f:
+        st.download_button(
+            label="Download the output notebook file",
+            data=f,
+            file_name="output.ipynb",
+            mime="application/json"
+        )
+