Update app.py

app.py

@@ -71,6 +71,9 @@ if uploaded_zip_file is not None:
             plt = display_dicom_image(selected_slice, dicom_files)
             st.pyplot(plt)
 
+    except Exception as e:
+        st.error(f"Error: {str(e)}")
+
 if st.button("Analyze"):
     command = "chmod +x inference.sh"
     
@@ -120,14 +123,13 @@ if st.button("Analyze"):
         if os.path.exists(image_path):
             st.title("Diameter Graph")
             st.image(image_path, use_column_width=True)
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+except Exception as e:
+    st.error(f"Error: {str(e)}")
 
 
 
 
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+    
     # finally:
     #     shutil.rmtree(temp_dir)