Update app.py

app.py

@@ -2,29 +2,34 @@ import streamlit as st
 import os
 import random
 
-def get_random_gifs(directory, num_gifs):
-    gif_files = [f for f in os.listdir(directory) if f.endswith('.gif')]
-    return random.sample(gif_files, num_gifs)
+def get_gifs(directory):
+    return [f for f in os.listdir(directory) if f.endswith('.gif')]
 
 def main():
     st.title('Random GIFs')
-    
+
     directory = './gifs'  # Replace with your directory of GIFs
-    gif_files = get_random_gifs(directory, 3)
+    gif_files = get_gifs(directory)
 
-    cols = st.beta_columns(3)
+    num_rows = len(gif_files) // 3
+    if len(gif_files) % 3:
+        num_rows += 1
+
+    cols = [st.columns(3) for _ in range(num_rows)]
     
-    for i in range(3):
+    for i in range(num_rows):
         for j in range(3):
-            gif_file = gif_files[i*3 + j]
-            cols[j].image(os.path.join(directory, gif_file), width=200)
-        
+            idx = i*3 + j
+            if idx < len(gif_files):
+                cols[i][j].image(os.path.join(directory, gif_files[idx]), width=200)
+
     if st.button('Randomize'):
-        gif_files = get_random_gifs(directory, 3)
-        for i in range(3):
+        random.shuffle(gif_files)
+        for i in range(num_rows):
             for j in range(3):
-                gif_file = gif_files[i*3 + j]
-                cols[j].image(os.path.join(directory, gif_file), width=200)
+                idx = i*3 + j
+                if idx < len(gif_files):
+                    cols[i][j].image(os.path.join(directory, gif_files[idx]), width=200)
 
 if __name__ == "__main__":
     main()