improved random wallpaper picker

1 files changed, 8 insertions, 0 deletions

diff --git a/src/user_interface.py b/src/user_interface.py
index 99eb524..f3d068a 100644
--- a/src/user_interface.py
+++ b/src/user_interface.py
@@ -38,8 +38,16 @@ def colorPickerUI(img_path: str) -> tuple:
 
 def pickRandomWallpaper() -> str:
     confirmed = False
+    history = []
+    num_wallpapers = len(os.listdir(Paths['wallpapers']))
     while not confirmed:
+        if len(history) == num_wallpapers:
+            print('[bold blue] Wallpapers exhausted. Resetting history...')
+            history.clear()
         wallpaper = Paths['wallpapers'] + random.choice(os.listdir(Paths['wallpapers']))
+        if wallpaper in history:
+            continue
+        history.append(wallpaper)
         # TODO: Replace image preview with something native to python
         # (Currently borrowing viu module from Rust)
         os.system(f'viu {wallpaper}')