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 main():
    st.title('Random GIFs')
    
    directory = './gifs'  # Replace with your directory of GIFs
    gif_files = get_random_gifs(directory, 3)

    cols = st.beta_columns(3)
    
    for i in range(3):
        for j in range(3):
            gif_file = gif_files[i*3 + j]
            cols[j].image(os.path.join(directory, gif_file), width=200)
        
    if st.button('Randomize'):
        gif_files = get_random_gifs(directory, 3)
        for i in range(3):
            for j in range(3):
                gif_file = gif_files[i*3 + j]
                cols[j].image(os.path.join(directory, gif_file), width=200)

if __name__ == "__main__":
    main()