File size: 2,315 Bytes
3e423f6 |
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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
# taken from https://gist.github.com/benlansdell/44000c264d1b373c77497c0ea73f0ef2
# slightly modified
"""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):
global i_will_regret_this, i_will_regret_this2
choice = st.session_state[key]
if os.path.isdir(os.path.join(st.session_state[key + "curr_dir"], choice)):
st.session_state[key + "index"] = 0
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"]))
files.insert(0, "..")
files.insert(0, ".")
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, "..")
files.insert(0, ".")
st.session_state[key + "files"] = files
st.session_state[key + "curr_dir"] = base_path
st.session_state[key + "index"] = (
st.session_state[key + "files"].index(os.path.basename(path))
if os.path.isfile(path) and path[-4:] == '.pth'
else 0
)
else:
base_path = st.session_state[key + "curr_dir"]
selected_file = st_placeholder.selectbox(
label=label,
options=st.session_state[key + "files"],
index=st.session_state[key + "index"],
key=key,
on_change=lambda: update_dir(key),
)
selected_path = os.path.normpath(os.path.join(base_path, selected_file))
st_placeholder.write(os.path.abspath(selected_path))
return selected_path
|