Spaces:
Sleeping
Sleeping
File size: 1,240 Bytes
692d068 d1c70f0 692d068 d1c70f0 692d068 d1c70f0 692d068 d1c70f0 692d068 d1c70f0 692d068 |
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 |
from __future__ import annotations
import base64
from io import BytesIO
from typing import TYPE_CHECKING
from urllib.parse import urlencode
import polars as pl
import zstandard as zstd
if TYPE_CHECKING:
from viewer import AxisProperties, ColorTransform
COMPRESSOR = zstd.ZstdCompressor(level=22)
DECOMPRESSOR = zstd.ZstdDecompressor()
def encode_url(
title: str,
molecule_id: str,
colors: ColorTransform,
axis_properties: AxisProperties,
data: pl.DataFrame,
description: str = "",
):
encode_dict = dict(title=title, molecule_id=molecule_id)
encode_dict.update({**colors.model_dump(), **axis_properties.model_dump()})
csv_str = data.write_csv(float_precision=4, float_scientific=True)
compressed = COMPRESSOR.compress(csv_str.encode())
base64_text = base64.b64encode(compressed).decode("utf8")
encode_dict["data"] = base64_text
if description:
encode_dict["description"] = description
return urlencode(encode_dict)
def decode_data(base64_text) -> pl.DataFrame:
decoded_bytes = base64.b64decode(base64_text)
decompressed = DECOMPRESSOR.decompress(decoded_bytes)
bio = BytesIO(decompressed)
data = pl.read_csv(bio)
bio.close()
return data
|