Update app.py

app.py

@@ -58,7 +58,9 @@ class DataViewerApp:
         #self.sdata = df['audio'].to_list()  # Separate audio data storage
         #self.current_page = 0
         #self.current_selected = -1
-    
+    def set1(self,df):
+        self.data =df[['text','speaker_id','secs','flag']]
+        self.sdata =df['audio'].to_list()
     def settt(self,df):
         self.df=pd.DataFrame()
         self.data =pd.DataFrame()
@@ -115,6 +117,9 @@ class DataViewerApp:
                       
         
                 self.sdata[self.current_selected] = audio
+                self.df['text'][self.current_selected] =text
+                self.df['audio'][self.current_selected] = audio
+                self.df['flag'][self.current_selected] =1
         return self.get_page_data(self.current_page),None,""
     def GetDataset_2(self,filename,ds=1.5):
         audios_data = []
@@ -214,15 +219,16 @@ class DataViewerApp:
             # Perform actions based on the selected column
 
     def on_column_dropdown_change(self,selected_column):
-        data=self.df1
+        data=self.df
         if selected_column=="all":
 
-            return self.settt(data)
-        else:
-            data=data[data['speaker_id'] ==selected_column]
-            return self.settt(data)
-
-             # Perform actions based on the selected column
+            return self.set1(data),len(data)
+        elif selected_column=="0":
+             data=data[data['flag'] ==0]
+             return self.set1(data),len(data)
+        else :
+            data=data[data['flag'] ==1]
+            return self.set1(data),len(data)
 
     def on_select(self,evt:gr.SelectData):
         index_now = evt.index[0]
@@ -246,7 +252,7 @@ class DataViewerApp:
     def get_output_audio(self):
         return self.sdata[self.current_selected] if self.current_selected >= 0 else None
     def Convert_DataFreme_To_DataSet(self,namedata):
-           df=self.finsh_data()
+           df=self.df
 
            df['audio'] = df['audio'].apply(lambda x: np.array(x, dtype=np.float32))
            if "__index_level_0__" in df.columns:
@@ -378,7 +384,7 @@ class DataViewerApp:
                              columns = []
                              columns1 = []
 
-                             columns =unique_speaker_ids =self.df['speaker_id'].unique().tolist()
+                             columns =["all","0","1"]
                              columns.append("all")
                              self.labell=gr.Label("count:")
                              self.column_dropdown = gr.Dropdown(choices=columns, label="speaker_id")
@@ -462,7 +468,7 @@ class DataViewerApp:
          self.prev_button.click(lambda page: self.update_page(page - 1), [self.page_number], [self.data_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.next_button.click(lambda page: self.update_page(page + 1), [self.page_number], [self.data_table, self.prev_button, self.next_button, self.page_number])
-         self.column_dropdown.change(self.on_column_dropdown_change,[self.column_dropdown], [self.data_table])
+         self.column_dropdown.change(self.on_column_dropdown_change,[self.column_dropdown], [self.data_table,self.labell])
          self.column_dropdown11.change(self.on_column_dropdown_change_operater,[self.column_dropdown1,self.column_dropdown11], [self.data_table])
          self.btn_convertDataset.click(self.Convert_DataFreme_To_DataSet,[self.txt_dataset],[self.label_dataset])
          self.totext.click(lambda:self.get_text_from_audio(self.get_output_audio()), [], self.txt_audio)