save_image.py

from hashlib import sha256
import streamlit as st
from datetime import datetime, timezone
from scheduler import ParquetScheduler


scheduler = ParquetScheduler(
    repo_id=st.secrets.IMAGES_REPO,
    token=st.secrets.HF_TOKEN,
    every=15,
)


@st.cache_data(ttl=3600)
def save_image(data, prediction):
    scheduler.append(
        {
            "date": datetime.now(timezone.utc).date(),
            "image": data,
            "prediction": prediction,
            "sha256": sha256(data).hexdigest(),
        }
    )