import gradio as gr
import json
from huggingface_hub import HfApi, ModelFilter, list_models
api = HfApi()
def model_explorer():
    # List all models
    filt = ModelFilter(task="text-classification")
    
    this = api.list_models(limit=100,filter=filt,cardData=True)
    for i,mod in enumerate(this):
        print(dir(mod))
        print(mod.id)
        print(mod.downloads)
        print(mod.likes)
        print(mod.pipeline_tag)
        out = mod.id
    return out

def tasks_json():

    with open("tasks.json", "r") as read_file:
        print("Read JSON file")
        tasks = json.load(read_file)
    
        print("Before Pretty Printing JSON Data")
        print(tasks)
def fix_json():
    new_json=[]
    with open("tasks.json", "r") as fix_file:
        for i, line in enumerate(fix_file.readlines()):

            #print (line)
            if "name:" in line:
                out = line.replace("name:",'"name":')
                new_json.append(out)
                print (out)
            elif "modality:" in line:
                out = line.replace("modality:",'"modality":')
                new_json.append(out)
                print (out)
            elif "type:" in line:
                out = line.replace("type:",'"type":')
                new_json.append(out)
                print (out)
            elif "subtasks:" in line:
                out = line.replace("subtasks:",'"subtasks":')
                new_json.append(out)
                print (out)
            elif "color:" in line:
                out = line.replace("color:",'"color":')
                new_json.append(out)
                print (out)
            elif "hideInModels:" in line:
                out = line.replace("hideInModels:",'"hideInModels":')
                new_json.append(out)
                print (out)
            elif "hideInDatasets:" in line:
                out = line.replace("hideInDatasets:",'"hideInDatasets":')
                new_json.append(out)
                print (out)
                            
            else: 
                new_json.append(line)
                print(line)            

    
    #print (json.dumps(new_json))
    #new_out=new_json.decode("utf-8")
    for i,line in enumerate(new_json):
        try:
            line.decode("utf-8")
        except Exception:
            pass   
        if line.strip():
            print (line)
fix_json()