| | |
| | |
| | """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 |
| |
|
