import gradio as gr
from huggingface_hub import HfApi
token = ""
api = HfApi(token=token)



def find_dataset(author=None):
    s_ist = (api.list_datasets(author=author,search="ai-tube"))
    #print(api.whoami())
    spaces=[]
    
    for space in s_ist:
    #for i,space in enumerate(s_ist):
        try:
            #space_ea = space.id.split("/",1)[1]
            
            spaces.append(f'{space.id}<br>')
            
            print(space.id)
        except Exception as e:
            print(e)
        #s_info=api.space_info(f'{name}/{space}',files_metadata=True)
        #print(s_info)
    print(spaces)
    return(spaces)
with gr.Blocks() as app:
    author = gr.Textbox(label="Author")
    button = gr.Button()
    output = gr.HTML()
    button.click(find_dataset,author,output)
app.launch(show_api=False,share=False)