from renumics import spotlight
from renumics.spotlight.analysis.typing import DataIssue
import pandas as pd
import numpy as np


if __name__ == "__main__":
    df = pd.read_json("sd_dataset.json")

    data_issues = sorted(df["selection"].unique())
    spotlight_data_issues = []
    for issue in data_issues:
        if issue == -1:
            continue
        issue_rows = np.where(df["selection"] == issue)[0].tolist()
        new_issue = DataIssue(
                severity="medium",
                title=f"Best Images Cluster {issue}",
                description=f"Best Images Cluster {issue}",
                rows=issue_rows)
        spotlight_data_issues.append(new_issue)

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

        view.close()