Umang-Bansal commited on
Commit
17217b4
1 Parent(s): 9956f18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -12,17 +12,23 @@ global_data = None
12
  def get_data_preview(file):
13
  global global_data
14
  global_data = pd.read_csv(file.name)
 
 
15
  return global_data.head()
16
 
17
  def label_data(ranges):
18
  global global_data
19
- for start, end, label in ranges.values:
 
 
20
  global_data.loc[start:end, 'label'] = label
 
21
  return global_data.tail()
22
 
23
  def preprocess_data():
24
  global global_data
25
- global_data.drop(columns=global_data.columns[0], axis=1, inplace=True)
 
26
  global_data.columns = ['raw_eeg', 'label']
27
  raw_data = global_data['raw_eeg']
28
  labels_old = global_data['label']
 
12
  def get_data_preview(file):
13
  global global_data
14
  global_data = pd.read_csv(file.name)
15
+ global_data['label'] = np.nan # Initialize a label column
16
+ print("Data preview:\n", global_data.head())
17
  return global_data.head()
18
 
19
  def label_data(ranges):
20
  global global_data
21
+ print("Ranges received for labeling:", ranges)
22
+ for i, (start, end, label) in enumerate(ranges.values):
23
+ print(f"Processing range {i}: start={start}, end={end}, label={label}")
24
  global_data.loc[start:end, 'label'] = label
25
+ print("Data after labeling:\n", global_data.tail())
26
  return global_data.tail()
27
 
28
  def preprocess_data():
29
  global global_data
30
+ if 'Unnamed: 0' in global_data.columns:
31
+ global_data.drop(columns='Unnamed: 0', axis=1, inplace=True)
32
  global_data.columns = ['raw_eeg', 'label']
33
  raw_data = global_data['raw_eeg']
34
  labels_old = global_data['label']