Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| import pickle | |
| import pandas as pd | |
| import json | |
| import base64 | |
| import uuid | |
| import re | |
| import importlib.util | |
| def import_from_file(module_name: str, filepath: str): | |
| """ | |
| Imports a module from file. | |
| Args: | |
| module_name (str): Assigned to the module's __name__ parameter (does not | |
| influence how the module is named outside of this function) | |
| filepath (str): Path to the .py file | |
| Returns: | |
| The module | |
| """ | |
| spec = importlib.util.spec_from_file_location(module_name, filepath) | |
| module = importlib.util.module_from_spec(spec) | |
| spec.loader.exec_module(module) | |
| return module | |
| def notebook_header(text): | |
| """ | |
| Insert section header into a jinja file, formatted as notebook cell. | |
| Leave 2 blank lines before the header. | |
| """ | |
| return f"""# # {text} | |
| """ | |
| def code_header(text): | |
| """ | |
| Insert section header into a jinja file, formatted as Python comment. | |
| Leave 2 blank lines before the header. | |
| """ | |
| seperator_len = (75 - len(text)) / 2 | |
| seperator_len_left = math.floor(seperator_len) | |
| seperator_len_right = math.ceil(seperator_len) | |
| return f"# {'-' * seperator_len_left} {text} {'-' * seperator_len_right}" | |
| def to_notebook(code): | |
| """Converts Python code to Jupyter notebook format.""" | |
| notebook = jupytext.reads(code, fmt="py") | |
| return jupytext.writes(notebook, fmt="ipynb") | |
| def open_link(url, new_tab=True): | |
| """Dirty hack to open a new web page with a streamlit button.""" | |
| # From: https://discuss.streamlit.io/t/how-to-link-a-button-to-a-webpage/1661/3 | |
| if new_tab: | |
| js = f"window.open('{url}')" # New tab or window | |
| else: | |
| js = f"window.location.href = '{url}'" # Current tab | |
| html = '<img src onerror="{}">'.format(js) | |
| div = Div(text=html) | |
| st.bokeh_chart(div) | |
| def download_button(object_to_download, download_filename, button_text): | |
| """ | |
| Generates a link to download the given object_to_download. | |
| From: https://discuss.streamlit.io/t/a-download-button-with-custom-css/4220 | |
| 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: | |
| if isinstance(object_to_download, bytes): | |
| pass | |
| elif isinstance(object_to_download, pd.DataFrame): | |
| object_to_download = object_to_download.to_csv(index=False) | |
| # 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(object_to_download).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:file/txt;base64,{b64}">{button_text}</a><br><br>' | |
| ) | |
| st.markdown(dl_link, unsafe_allow_html=True) | |