Update player.py

player.py

@@ -21,9 +21,10 @@ def find_dataset(author=None):
             for file in file_box:
                 if file.startswith("prompt_"):
                     prompt_file = file
-                    out_html=f'{out_html}  • {file}<br>'
-                    if prompt_file.split("prompt_",1)[1].split(".",1)[0] in file_box:
-                        print("yes")
+                    out_html=f'{out_html}<pre>  • {file}</pre><br>'
+                    for vid_file in ai_tube_box:
+                        if vid_file.split(".",1)[0] == prompt_file.split("prompt_",1)[1].split(".",1)[0]:
+                            print("yes")
                     
             #print([info.path for info in list_files_info(f'{space.id}', repo_type="dataset")])
             #print(space.id)