czyoung commited on
Commit
6d79beb
·
verified ·
1 Parent(s): ef17bdc

Save uploaded files in temp directory

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -9,6 +9,8 @@ from pathlib import Path
9
  from typing import Any, Dict, List, Optional, Union
10
  import copy
11
  import datetime
 
 
12
 
13
  PARQUET_DATASET_DIR = Path("parquet_dataset")
14
  PARQUET_DATASET_DIR.mkdir(parents=True,exist_ok=True)
@@ -46,7 +48,9 @@ uploaded_file_paths = st.file_uploader("Upload an audio of classroom activity to
46
  supported_file_types = ('.wav','.mp3','.mp4','.txt')
47
 
48
  valid_files = []
 
49
  audio_tabs = []
 
50
 
51
  if uploaded_file_paths is not None:
52
  # Reset valid_files?
@@ -56,7 +60,11 @@ if uploaded_file_paths is not None:
56
  uploaded_file = None
57
  else:
58
  if uploaded_file not in valid_files:
 
 
 
59
  valid_files.append(uploaded_file)
 
60
  if len(valid_files) > 0:
61
  audio_tabs = st.tabs([f.name for f in valid_files])
62
 
@@ -198,7 +206,7 @@ radio = st.radio('Pick one:', ['Left','Right'])
198
  selection = st.selectbox('Select', [1,2,3])
199
  if st.button("Upload Files to Dataset"):
200
  save_data({"color":colorPref,"direction":radio,"number":selection},
201
- valid_files,
202
  userid)
203
  st.success('I think it worked!')
204
 
 
9
  from typing import Any, Dict, List, Optional, Union
10
  import copy
11
  import datetime
12
+ import tempfile
13
+ import os
14
 
15
  PARQUET_DATASET_DIR = Path("parquet_dataset")
16
  PARQUET_DATASET_DIR.mkdir(parents=True,exist_ok=True)
 
48
  supported_file_types = ('.wav','.mp3','.mp4','.txt')
49
 
50
  valid_files = []
51
+ file_paths = []
52
  audio_tabs = []
53
+ temp_dir = tempfile.mkdtemp()
54
 
55
  if uploaded_file_paths is not None:
56
  # Reset valid_files?
 
60
  uploaded_file = None
61
  else:
62
  if uploaded_file not in valid_files:
63
+ path = os.path.join(temp_dir, uploaded_file.name)
64
+ with open(path, "wb") as f:
65
+ f.write(uploaded_file.getvalue())
66
  valid_files.append(uploaded_file)
67
+ file_paths.append(path)
68
  if len(valid_files) > 0:
69
  audio_tabs = st.tabs([f.name for f in valid_files])
70
 
 
206
  selection = st.selectbox('Select', [1,2,3])
207
  if st.button("Upload Files to Dataset"):
208
  save_data({"color":colorPref,"direction":radio,"number":selection},
209
+ file_paths,
210
  userid)
211
  st.success('I think it worked!')
212