import pickle
import datasets
from renumics import spotlight
import os
import pandas as pd
from renumics.spotlight.layouts import debug_classification



if __name__ == "__main__":

    cache_file_issues="metadata_issues.pkl"
   
    # Load dataset using datasets.load_dataset()
    ds = datasets.load_dataset("renumics/emodb-enriched", split="all")
    print("Dataset loaded using datasets.load_dataset().")

     
    with open(cache_file_issues, "rb") as issue_file:
        issues = pickle.load(issue_file)


    while True:
        layout = debug_classification(label='gender', prediction='m1_gender_prediction', embedding='m1_embedding', features=['age', 'emotion'], inspect={'audio': spotlight.Audio})
        
        view = spotlight.show(ds, issues=issues, port=7860, host="0.0.0.0", layout=layout,
                    allow_filebrowsing=False)
        view.close()