big update

app.py

@@ -12,6 +12,7 @@ from scipy.io import wavfile
 import pydub
 import time
 import os
+import atexit
 
 # MODEL LOADING and INITIALISATION
 model = torch.jit.load("snorenetv1_small.ptl")
@@ -61,28 +62,28 @@ if uploaded_file is not None:
 
     # PERCENTAGE OF SNORING PLOT
 
-filename = uploaded_file.name
-cwdir = os.getcwd()
-file_path = os.path.join(cwdir, filename)
+    filename = uploaded_file.name
+    cwdir = os.getcwd()
+    file_path = os.path.join(cwdir, filename)
 
-if os.path.exists(file_path):
-    total = snore + other
-    snore_percentage = (snore / total) * 100
-    other_percentage = (other / total) * 100
+    if os.path.exists(file_path):
+        total = snore + other
+        snore_percentage = (snore / total) * 100
+        other_percentage = (other / total) * 100
 
-    categories = ["Snore", "Other"]
-    percentages = [snore_percentage, other_percentage]
+        categories = ["Snore", "Other"]
+        percentages = [snore_percentage, other_percentage]
 
-    plt.figure(figsize=(8, 4))
-    plt.barh(categories, percentages, color=['#ff0033', '#00ffee'])
-    plt.xlabel('Percentage')
-    plt.title('Percentage of Snoring')
-    plt.xlim(0, 100)
+        plt.figure(figsize=(8, 4))
+        plt.barh(categories, percentages, color=['#ff0033', '#00ffee'])
+        plt.xlabel('Percentage')
+        plt.title('Percentage of Snoring')
+        plt.xlim(0, 100)
 
-    for i, percentage in enumerate(percentages):
-        plt.text(percentage, i, f' {percentage:.2f}%', va='center')
-    st.write("DONE")
-    st.pyplot(plt)
+        for i, percentage in enumerate(percentages):
+            plt.text(percentage, i, f' {percentage:.2f}%', va='center')
+        st.write("DONE")
+        st.pyplot(plt)