Update app.py

app.py

@@ -5,7 +5,21 @@ import os
 import numpy as np
 from datasets import Dataset, DatasetDict
 from datasets import load_dataset
-
+from df.enhance import enhance, init_df, load_audio, save_audio
+def remove_nn(wav, sample_rate=16000):
+    
+    audio=librosa.resample(wav,orig_sr=sample_rate,target_sr=df_state.sr(),)
+    
+    audio=torch.tensor([audio])
+  #  audio, _ = load_audio('full_generation.wav', sr=df_state.sr())
+  
+    
+    enhanced = enhance(model_enhance, df_state, audio)
+    
+   # save_audio("enhanced.wav", enhanced, df_state.sr())
+    audiodata=librosa.resample(enhanced[0].numpy(),orig_sr=df_state.sr(),target_sr=sample_rate)
+   
+    return 44100, audiodata/np.max(audiodata)
 class DataEditor:
     def __init__(self,df):
         self.df=df
@@ -25,6 +39,8 @@ class DataEditor:
         self.current_page = 0
         self.current_selected =1
         return self.data
+    def get_output_audio(self):
+        return self.sdata[self.current_selected] if self.current_selected >= 0 else None
     def get_prev_page(self,pagenumber):
         if self.next_prveo>=0:
             self.next_prveo-=1
@@ -176,6 +192,7 @@ class DataEditor:
                      with gr.Row():
                           self.btn_save = gr.Button("Save")
                           self.btn_saveAs = gr.Button("SaveAs")
+                          self.btn_enhance = gr.Button("enhance ", size="sm",variant="primary",min_width=50)
                          # self.buttonnext=gr.Button("Next")
                          # self.buttonprev=gr.Button("Prev")
                           self.btn_delete = gr.Button("Delete")
@@ -191,6 +208,7 @@ class DataEditor:
             self.btn_savedataset.click(self.convert_dataframe_to_dataset, [self.txtsaveurl], [self.label])
             self.prev_button.click(lambda page: self.update_page(page - 1), [self.page_number], [self.table, self.prev_button, self.next_button, self.page_number])
                    #self.btn_save.click(self.save_row, [self.txt_audio,self.audio_player], [self.data_table])
+            self.btn_enhance.click(lambda: remove_nn(self.get_output_audio()), [],self.self.btn_record)
             self.next_button.click(lambda page: self.update_page(page + 1), [self.page_number], [self.table, self.prev_button, self.next_button, self.page_number])
         #demo.launch()
 df1=pd.DataFrame(columns=['text','flag','audio'])