Update player.py

player.py

@@ -5,6 +5,7 @@ import json
 token = ""
 source_dataset = "jbilcke-hf/ai-tube-index"
 api = HfApi(token=token)
+
 def find_dataset(author=""):
     s_ist = (api.list_datasets(author=author,search="ai-tube"))
     #print(api.whoami())
@@ -12,65 +13,51 @@ def find_dataset(author=""):
     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]
             channels.append(space.id)
         except Exception:
             pass
     return (gr.Dropdown(label="Channels", choices=[s for s in channels],interactive=True))
 
-def find_dataset_og(author=None):
-    s_ist = (api.list_datasets(author=author,search="ai-tube"))
-    #print(api.whoami())
-    spaces=[]
-    out_html=""
-    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>')
-            out_html = f'{out_html}<b>{space.id}</b><br>'
-            file_box = [info.path for info in list_files_info(f'{space.id}', repo_type="dataset")]
-            ai_tube_box = [info.path for info in list_files_info(f'{source_dataset}', repo_type="dataset")]
-            for file in file_box:
-                if file.startswith("prompt_"):
-                    prompt_slug = file.split("prompt_",1)[1].split(".",1)[0]
-                    out_html=f'{out_html}<pre>  • {file}</pre>'
-                    print(prompt_slug)
-                    for vid_file in ai_tube_box:
+def load_vid(channel):
+    try:
+        file_box = [info.path for info in list_files_info(f'{channel}', repo_type="dataset")]
+        ai_tube_box = [info.path for info in list_files_info(f'{source_dataset}', repo_type="dataset")]
+        for file in file_box:
+            if file.startswith("prompt_"):
+                prompt_slug = file.split("prompt_",1)[1].split(".",1)[0]
+                out_html=f'{out_html}<pre>  • {file}</pre>'
+                print(prompt_slug)
+                for vid_file in ai_tube_box:
+                    try:
+                        vid_slug=vid_file.split("videos/",1)[1].split(".",1)[0]
+                        print(vid_slug)
+                        try:
+                            if vid_file.endswith(".json"): 
+                                r = requests.get(f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/raw/main/videos/{vid_slug}.json')
+                                #r = requests.get(f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/blob/main/videos/00b4bcda-7b4a-40f8-9833-e490425a7b91.mp4')
+                                print (r.content)
+                                #uid = uuid.uuid4()
+                                #file = open(f'{name}/{f_name}','wb')
+                                #file.write(r.content)
+                        except Exception as e:
+                            print (f'error:: {e}')
+                            pass
                         try:
-                            vid_slug=vid_file.split("videos/",1)[1].split(".",1)[0]
-                            print(vid_slug)
-                            try:
-                                if vid_file.endswith(".json"): 
-                                    r = requests.get(f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/raw/main/videos/{vid_slug}.json')
-                                    #r = requests.get(f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/blob/main/videos/00b4bcda-7b4a-40f8-9833-e490425a7b91.mp4')
-                                    print (r.content)
-                                    #uid = uuid.uuid4()
-                                    #file = open(f'{name}/{f_name}','wb')
-                                    #file.write(r.content)
-                            except Exception as e:
-                                print (f'error:: {e}')
-                                pass
-                            try:
-                                if vid_slug == prompt_slug:
-                                    out_html=f'{out_html}<pre>    + {vid_file}</pre>'
-                                    
-                            except Exception:
-                                pass
+                            if vid_slug == prompt_slug:
+                                out_html=f'{out_html}<pre>    + {vid_file}</pre>'
+                                
                         except Exception:
                             pass
-                    
-            #print([info.path for info in list_files_info(f'{space.id}', repo_type="dataset")])
-            #print(space.id)
-        except Exception as e:
-            print(e)
-        #s_info=api.space_info(f'{name}/{space}',files_metadata=True)
-        #print(s_info)
+                    except Exception:
+                        pass
+                
+        #print([info.path for info in list_files_info(f'{space.id}', repo_type="dataset")])
+        #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(out_html)
 with gr.Blocks() as app:
@@ -78,5 +65,5 @@ with gr.Blocks() as app:
     button = gr.Button()
     output = gr.HTML()
     app.load(find_dataset,None,author)
-    button.click(find_dataset,None,author)
+    author.change(load_vid,author,output)
 app.launch(show_api=False,share=False)