Update player.py

player.py

@@ -20,15 +20,16 @@ def find_dataset(author=None):
             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_file = file
+                    prompt_slug = file.split("prompt_",1)[1].split(".",1)[0]
                     out_html=f'{out_html}<pre>  • {file}</pre>'
                     print(prompt_file)
                     for vid_file in ai_tube_box:
+                        vid_slug=vid_file.split("videos/",1)[1].split(".",1)[0]
                         print(vid_file)
                         try:
-                            if vid_file.split("videos/",1)[1].split(".",1)[0] == prompt_file.split("prompt_",1)[1].split(".",1)[0]:
-                                print("yes")
+                            if vid_slug == prompt_slug:
                                 out_html=f'{out_html}<pre>    + {vid_file}</pre>'
+                                
                         except Exception:
                             pass