Update app.py

app.py

@@ -5,6 +5,7 @@ import zipfile
 import os
 import subprocess
 from datetime import datetime
+import shutil
 
 # Flag to track if subprocess commands have been executed
 subprocess_executed = False
@@ -128,5 +129,7 @@ if uploaded_zip_file is not None:
 
     except Exception as e:
         st.error(f"Error: {str(e)}")
+    finally:
+        shutil.rmtree(temp_dir)
 
 st.write("Upload a ZIP file containing DICOM slices to view the images.")