Update app.py

app.py

@@ -89,14 +89,22 @@ def processFile(filePath):
     global gainWindow
     global minimumGain
     global maximumGain
+    print("Loading file")
     waveformList, sampleRate = su.splitIntoTimeSegments(filePath)
+    print("File loaded")
     enhancedWaveformList = []
+    print("Denoising")
     for w in waveformList:
         newW = enhance(dfModel,dfState,w,atten_lim_db=attenLimDB)#.detach().cpu()
         enhancedWaveformList.append(newW)
+    print("Audio denoised")
     waveformEnhanced = su.combineWaveforms(enhancedWaveformList)     
+    print("Equalizing Audio")
     waveform_gain_adjusted = su.equalizeVolume()(waveformEnhanced,sampleRate,gainWindow,minimumGain,maximumGain)
+    print("Audio Equalized")
+    print("Detecting speakers")
     annotations = pipeline({"waveform": waveformEnhanced, "sample_rate": sampleRate})
+    print("Speakers Detected")
     speakerList = su.annotationToSpeakerList(annotations)
     return (speakerList, annotations, int(waveformEnhanced.shape[-1]/sampleRate))
 
@@ -113,6 +121,7 @@ audio_tabs = []
 temp_dir = tempfile.mkdtemp()
 
 if uploaded_file_paths is not None:
+    print("Found file paths")
     valid_files = []
     file_paths = []
     audio_tabs = []
@@ -122,6 +131,7 @@ if uploaded_file_paths is not None:
             st.error('File must be of type: {}'.format(supported_file_types))
             uploaded_file = None
         else:
+            print(f"Valid file: {uploaded_file.name}")
             if uploaded_file not in valid_files:
                 path = os.path.join(temp_dir, uploaded_file.name)
                 with open(path, "wb") as f:
@@ -141,6 +151,7 @@ if st.button("Analyze All Audio",key=f"button_all"):
     if len(valid_files) == 0:
         st.error('Upload file(s) first!')
     else:
+        print("Start analyzing")
         start_time = time.time()
         totalFiles = len(valid_files)
         for i in range(totalFiles):