import streamlit as st
import polars as pl
import numpy as np
import random

from streamlit.runtime.caching.cache_type import CacheType
from streamlit.runtime.caching.hashing import update_hash
import hashlib
import timeit


random.seed(0)
np.random.seed(0)

def my_function():
    my_hasher = hashlib.new("md5", usedforsecurity=False)

    polars_df = pl.DataFrame(
        np.random.randint(0, 100, size=(100_000, 5)),
        schema=["a", "b", "c", "d", "g"]
    )
    update_hash(polars_df, my_hasher, cache_type=CacheType.DATA)
    return my_hasher.digest()




st.write(my_function())

execution_time = timeit.timeit(
    my_function,
    number=1_000
)
st.write(f"Execution time: {execution_time:.6f} seconds")