Commit d93ea5cd authored by yfszzx's avatar yfszzx
Browse files

inspiration perfected

parent 40ddb6df
Loading
Loading
Loading
Loading
+5 −2
Original line number Diff line number Diff line
@@ -74,7 +74,10 @@ def get_inspiration_images(source, types, keyword):
    for a in names:
        image_path = os.path.join(opts.inspiration_dir, 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):