diff options
| author | yfszzx <yfszzx@gmail.com> | 2022-10-22 10:21:21 +0800 |
|---|---|---|
| committer | yfszzx <yfszzx@gmail.com> | 2022-10-22 10:21:21 +0800 |
| commit | d93ea5cdeb2fd3607b7265271ccab2c9bf4c1156 (patch) | |
| tree | 2c0398a8bd34214f01c098c87c10a4d1a5a93220 /modules/inspiration.py | |
| parent | 40ddb6df61564684263c7442bacf61efe3882b87 (diff) | |
inspiration perfected
Diffstat (limited to 'modules/inspiration.py')
| -rw-r--r-- | modules/inspiration.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/modules/inspiration.py b/modules/inspiration.py index 319183ab..94ff139a 100644 --- a/modules/inspiration.py +++ b/modules/inspiration.py @@ -73,8 +73,11 @@ def get_inspiration_images(source, types, keyword): image_list = [] for a in names: image_path = os.path.join(opts.inspiration_dir, a) - images = os.listdir(image_path) - image_list.append((os.path.join(image_path, random.choice(images)), a)) + images = os.listdir(image_path) + if len(images) > 0: + image_list.append((os.path.join(image_path, random.choice(images)), a)) + else: + print(image_path) return image_list, names def select_click(index, name_list): |