Enhance CSV parsing utility to support optional shuffling; update HTML table generation for improved playback handling

app.py

@@ -6,6 +6,7 @@ import logging
 import pandas as pd
 from pytube import Search, YouTube
 from flask import Flask, request, send_from_directory
+from sklearn.utils import shuffle as shuffle_fn
 
 # https://github.com/pytube/pytube/issues/1270#issuecomment-2100372834
 pytube_logger = logging.getLogger('pytube')
@@ -48,7 +49,7 @@ def parse_csv_test():
     try:
         # Construct the path to the CSV file
         csv_path = Path(__file__).parent / 'shazamlibrary.test.csv'
-        return parse_csv_util(pd.read_csv(csv_path, header=1))
+        return parse_csv_util(pd.read_csv(csv_path, header=1), True)
     except Exception as e:
         return str(e)
 
@@ -57,9 +58,11 @@ def get_youtube_song(title: str, artist: str) -> Optional[YouTube]:
     search_result = Search(f'{title} by {artist}')
     return search_result.results[0] if search_result.results else None
 
-def parse_csv_util(shazamlibrary_df):
+def parse_csv_util(df: pd.DataFrame, shuffle = False):
     try:
-        shazamlibrary_df = shazamlibrary_df.drop_duplicates(subset=['TrackKey'])[['Title', 'Artist']]
-        return shazamlibrary_df.to_json(orient="records")
+        df = df.drop_duplicates(subset=['TrackKey'])[['Title', 'Artist']]
+        if shuffle:
+            df = shuffle_fn(df)
+        return df.to_json(orient="records")
     except Exception as e:
         return str(e)

index.html

@@ -136,7 +136,7 @@
         function generateTable(playlist) {
             try {
                 songsPlaylist = playlist
-                tableBody = songsPlaylist.map((i, index) => `
+                const tableBody = songsPlaylist.map((i, index) => `
                             <tr data-index="${index}">
                                 <th>${index + 1}</th>
                                 <th>${i.Title}</th>
@@ -154,6 +154,11 @@
                     </thead>
                     <tbody>${tableBody}</tbody>
                 </table>`
+
+                tryToPlay(() => {
+                    videoIndex = -1;
+                    onContinue();
+                }, 100)
             } catch (error) {
                 playlistTable.innerHTML = error;
             }
@@ -207,16 +212,21 @@
                 }, newIndex);
             }
         }
-        const setIntervalId = setInterval(() => {
-            if (YT.Player) {
-                if (!youtubePlayer) {
-                    initiatePlayer();
-                } else if (songsPlaylist.length > 0) {
-                    clearInterval(setIntervalId);
-                    onContinue();
+
+        function tryToPlay(playCallback, timeout) {
+            setTimeout(() => {
+                if (YT.Player) {
+                    if (!youtubePlayer) {
+                        initiatePlayer();
+                        tryToPlay?.();
+                    } else if (songsPlaylist.length > 0) {
+                        playCallback();
+                    }
                 }
-            }
-        }, 300);
+            }, timeout);
+        }
+
+        tryToPlay(null, 300);
     </script>
 </body>
 

requirements.txt

@@ -1,4 +1,5 @@
 pytube==15.0.0
 pandas==2.2.2
 flask==3.0.3
-gunicorn==22.0.0
+gunicorn==22.0.0
+scikit-learn==1.5.2