import pickle
import pandas as pd
from renumics import spotlight


if __name__ == "__main__":
    df = pd.read_json("df.json")
    df[
        "GitHub"
    ] = '<a href="https://github.com/Renumics/spotlight" target="_blank">⭐ Check Out Spotlight on GitHub</a>'

    with open("issues.pkl", "rb") as f:
        issues = pickle.load(f)

    while True:
        dtypes = {
            "embedding": spotlight.Embedding,
            "filename": spotlight.Image,
            "filename_nearest": spotlight.Image,
        }
        view = spotlight.show(
            df,
            dtype=dtypes,
            port=7860,
            host="0.0.0.0",
            allow_filebrowsing=False,
            issues=issues,
            layout="spotlight-layout.json",
        )

        view.close()