Update player.py

player.py

@@ -5,9 +5,25 @@ import json
 token = ""
 source_dataset = "jbilcke-hf/ai-tube-index"
 api = HfApi(token=token)
-
-
 def find_dataset(author=None):
+    s_ist = (api.list_datasets(author=author,search="ai-tube"))
+    #print(api.whoami())
+    authors=[]
+    videos=[]
+    out_html=""
+    for space in s_ist:
+    #for i,space in enumerate(s_ist):
+        try:
+            space_loc=space.id
+            space_auth=space_loc.split("/",1)[0]
+            space_name=space_loc.split("/",1)[0]
+            #space_ea = space.id.split("/",1)[1]
+            authors.append(space.id)
+        except Exception:
+            pass
+    return (gr.Dropdown.update(label="Spaces", choices=[s for s in spaces])
+
+def find_dataset_og(author=None):
     s_ist = (api.list_datasets(author=author,search="ai-tube"))
     #print(api.whoami())
     spaces=[]
@@ -58,8 +74,8 @@ def find_dataset(author=None):
     print(spaces)
     return(out_html)
 with gr.Blocks() as app:
-    author = gr.Textbox(label="Author")
+    author = gr.Dropdown(label="Author")
     button = gr.Button()
     output = gr.HTML()
-    button.click(find_dataset,author,output)
+    button.click(find_dataset,None,author)
 app.launch(show_api=False,share=False)