update app.py

app.py

@@ -189,15 +189,26 @@ audio_output_prediction, sample_rate, min_duration, frame_length, hop_length_fra
     print(audio_output_prediction)
     sf.write(audio_output_prediction , denoise_long[0, :], sample_rate)
 
-def denoise_audio(audioName):
-  sr, data = audioName
-  sf.write("temp.wav",data, sr)
-  testNo = "temp"
-  audio_dir_prediction = os.path.abspath("/")+ str(testNo) +".wav"
-  sample_rate, data = audioName[0], audioName[1]
-  len_data = len(data)  # holds length of the numpy array
-
+def denoise_audio(audioName,mic):
+  if audioName != 0:
+    sr, data = audioName
+    sf.write("temp.wav",data, sr)
+    testNo = "temp"
+    audio_dir_prediction = os.path.abspath("/")+ str(testNo) +".wav"
+    sample_rate, data = audioName[0], audioName[1]
+    len_data = len(data)  # holds length of the numpy array
+
+    
+    
+  else:
+    sr, data = mic
+    sf.write("temp.wav",data, sr)
+    testNo = "temp"
+    audio_dir_prediction = os.path.abspath("/")+ str(testNo) +".wav"
+    sample_rate, data = mic[0], mic[1]
+    len_data = len(data)
   
+
   t = len_data / sample_rate # returns duration but in floats
   print("t:",t)
   weights_path = os.path.abspath("./")
@@ -216,7 +227,7 @@ def denoise_audio(audioName):
   dim_square_spec = int(n_fft / 2) + 1
 
   prediction(weights_path, name_model, audio_dir_prediction, dir_save_prediction, audio_input_prediction,
-          audio_output_prediction, sample_rate, min_duration, frame_length, hop_length_frame, n_fft, hop_length_fft)
+            audio_output_prediction, sample_rate, min_duration, frame_length, hop_length_frame, n_fft, hop_length_fft)
   print(audio_output_prediction)
   return audio_output_prediction