| import os |
| import streamlit as st |
| from views.utils import * |
| from PIL import Image |
|
|
| class Sidebar: |
| def __init__(self): |
| pass |
| |
| def choose_folder_view(self,ct): |
| with ct: |
| |
| |
| current_dir = os.path.dirname(os.path.abspath(__file__)) |
| image_path = os.path.join(current_dir, 'assets/drilldocs_logo.jpeg') |
| image = Image.open(image_path) |
| new_height = 200 |
| aspect_ratio = image.width / image.height |
| new_width = int(new_height * aspect_ratio) |
| resized_image = image.resize((new_width, new_height)) |
| ct.image(resized_image,use_column_width=False) |
|
|
| ct.header('⚙️ Parameters') |
| ct.subheader('Choose the folder to store the images') |
|
|
| def file_selector_view(self,ct): |
| folder_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../..', 'uploads')) |
| folders = [] |
| files = os.listdir(folder_path) |
| new_folder = '➕ new client' |
| for folder in files: |
| flags =['.streamlit','__pycache__'] |
| if os.path.isdir(folder) and folder not in flags: |
| folders.append(folder) |
| |
| folders.append(new_folder) |
| selected_filename = ct.selectbox('📁 Select a folder for saving', folders, key="folder_selectbox") |
| if selected_filename==folders[-1]: |
| fn,fb=ct.columns(2) |
| folder_name = fn.text_input('Client name',placeholder="Client name", label_visibility='collapsed') |
| add_btn = fb.button('Add') |
| if add_btn: |
| try: |
| on_click_btn(folder_name=folder_name) |
| st.experimental_rerun() |
| except: |
| ct.error("Error with folder name ") |
| |
| elif is_any_directory_not_empty(selected_filename): |
| ct.warning("This folder is not empty! it's content will be erased ") |
| |
| return os.path.join(folder_path, selected_filename) |
| |
