first commit

app.py

@@ -6,6 +6,7 @@ whoops = load_dataset("nlphuji/whoops")['test']
 BUCKET_PATH = 'https://wmtis.s3.eu-west-1.amazonaws.com/wmtis_images'
 
 df = whoops.to_pandas()
+print(f"Got {len(df)} items in dataframe")
 df = df.sample(frac=1)
 def get_image_url(img_id):
     return f"{BUCKET_PATH}/{img_id}.png"
@@ -36,9 +37,17 @@ def display_df():
     return df_images
 
 def display_next(dataframe, end):
-    start = (end or dataframe.index[-1]) + 1
-    end = start + (LINES_NUMBER-1)
-    df_images = df.loc[start:end]
+    start = int(end or len(dataframe))
+    end = int(start) + int(LINES_NUMBER)
+    global df
+    if end >= len(df) - 1:
+        start = 0
+        end = LINES_NUMBER
+        df = df.sample(frac=1)
+        print(f"Shuffeling")
+    print(f"end: {end},start: {start}")
+    df_images = df.iloc[start:end]
+    assert len(df_images) == LINES_NUMBER
     return df_images, end
 
 # Gradio Blocks