Spaces:
Runtime error
Runtime error
File size: 1,858 Bytes
e875957 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
"""FilePicker for streamlit.
Still doesn't seem to be a good solution for a way to select files to process from the server Streamlit is running on.
Here's a pretty functional solution.
Usage:
```
import streamlit as st
from filepicker import st_file_selector
tif_file = st_file_selector(st, key = 'tif', label = 'Choose tif file')
```
"""
import os
import streamlit as st
def update_dir(key):
choice = st.session_state[key]
if os.path.isdir(os.path.join(st.session_state[key+'curr_dir'], choice)):
st.session_state[key+'curr_dir'] = os.path.normpath(os.path.join(st.session_state[key+'curr_dir'], choice))
files = sorted(os.listdir(st.session_state[key+'curr_dir']))
if "images" in files:
files.remove("images")
st.session_state[key+'files'] = files
def st_file_selector(st_placeholder, path='.', label='Select a file/folder', key = 'selected'):
if key+'curr_dir' not in st.session_state:
base_path = '.' if path is None or path == '' else path
base_path = base_path if os.path.isdir(base_path) else os.path.dirname(base_path)
base_path = '.' if base_path is None or base_path == '' else base_path
files = sorted(os.listdir(base_path))
files.insert(0, 'Choose a file...')
if "images" in files:
files.remove("images")
st.session_state[key+'files'] = files
st.session_state[key+'curr_dir'] = base_path
else:
base_path = st.session_state[key+'curr_dir']
selected_file = st_placeholder.selectbox(label=label,
options=st.session_state[key+'files'],
key=key,
on_change = lambda: update_dir(key))
if selected_file == "Choose a file...":
return None
return selected_file |