Spaces:
Running
Running
| #!/usr/bin/env python | |
| # coding: utf-8 | |
| import pandas as pd | |
| import base64 | |
| import uuid | |
| import io | |
| import pickle | |
| import string | |
| import simplejson as json | |
| import regex as re | |
| from functools import reduce | |
| import streamlit as st | |
| chars = string.digits + string.ascii_uppercase + string.ascii_lowercase + '_$' | |
| selected_sheet = None | |
| def compress(g): | |
| bs = [int(g[i:i + 2], 16) for i in range(0, len(g), 2)] | |
| def b64(v, l=4): | |
| return ''.join([chars[(v // (64 ** i)) % 64] for i in range(l)][::-1]) | |
| return ''.join([b64(bs[0], 2)] + [b64((bs[i] << 16) + (bs[i + 1] << 8) + bs[i + 2]) for i in range(1, 16, 3)]) | |
| def download_button(object_to_download, download_filename, button_text, pickle_it=False): | |
| """ | |
| Generates a link to download the given object_to_download. | |
| Params: | |
| ------ | |
| object_to_download: The object to be downloaded. | |
| download_filename (str): filename and extension of file. e.g. mydata.csv, | |
| some_txt_output.txt download_link_text (str): Text to display for download | |
| link. | |
| button_text (str): Text to display on download button (e.g. 'click here to download file') | |
| pickle_it (bool): If True, pickle file. | |
| Returns: | |
| ------- | |
| (str): the anchor tag to download object_to_download | |
| Examples: | |
| -------- | |
| download_link(your_df, 'YOUR_DF.csv', 'Click to download data!') | |
| download_link(your_str, 'YOUR_STRING.txt', 'Click to download text!') | |
| """ | |
| if pickle_it: | |
| try: | |
| object_to_download = pickle.dumps(object_to_download) | |
| except pickle.PicklingError as e: | |
| st.write(e) | |
| return None | |
| else: | |
| if isinstance(object_to_download, bytes): | |
| pass | |
| elif isinstance(object_to_download, pd.DataFrame): | |
| #object_to_download = object_to_download.to_csv(index=False) | |
| towrite = io.BytesIO() | |
| object_to_download = object_to_download.to_excel( | |
| towrite, | |
| encoding='utf-8', | |
| index=False, | |
| header=True, | |
| na_rep='' | |
| ) | |
| towrite.seek(0) | |
| # Try JSON encode for everything else | |
| else: | |
| object_to_download = json.dumps(object_to_download) | |
| try: | |
| # some strings <-> bytes conversions necessary here | |
| b64 = base64.b64encode(object_to_download.encode()).decode() | |
| except AttributeError as e: | |
| b64 = base64.b64encode(towrite.read()).decode() | |
| button_uuid = str(uuid.uuid4()).replace('-', '') | |
| button_id = re.sub('\d+', '', button_uuid) | |
| custom_css = f""" | |
| <style> | |
| #{button_id} {{ | |
| display: inline-flex; | |
| align-items: center; | |
| justify-content: center; | |
| background-color: rgb(255, 255, 255); | |
| color: rgb(38, 39, 48); | |
| padding: .25rem .75rem; | |
| position: relative; | |
| text-decoration: none; | |
| border-radius: 4px; | |
| border-width: 1px; | |
| border-style: solid; | |
| border-color: rgb(230, 234, 241); | |
| border-image: initial; | |
| }} | |
| #{button_id}:hover {{ | |
| border-color: rgb(246, 51, 102); | |
| color: rgb(246, 51, 102); | |
| }} | |
| #{button_id}:active {{ | |
| box-shadow: none; | |
| background-color: rgb(246, 51, 102); | |
| color: white; | |
| }} | |
| </style> """ | |
| dl_link = custom_css + f'<a download="{download_filename}" id="{button_id}" href="data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{b64}">{button_text}</a><br></br>' | |
| return dl_link | |
| cobie_file_button = st.text_input("Dropbox link to COBie file", key="cobie_file_button") | |
| # In[ ]: | |
| if cobie_file_button: | |
| cobie_file_path = st.session_state.cobie_file_button | |
| if '=0' in cobie_file_path: | |
| cobie_file_path = cobie_file_path.replace('=0', '=1') | |
| cobie_file = pd.ExcelFile(cobie_file_path) | |
| tabs = pd.ExcelFile("path").sheet_names | |
| selected_sheet = st.selectbox( | |
| 'Select sheet for changing revit ids to COBie ids', | |
| tabs, | |
| ) | |
| if selected_sheet: | |
| df = cobie_file.parse(selected_sheet) | |
| st.write(df) | |
| # st.markdown(download_button(df, 'COBieIDs.xlsx', 'Download COBieIDs.xlsx'), unsafe_allow_html=True) | |