--random for --feed-from-tags

joy

@@ -17,6 +17,7 @@ The main components include:
 import os
 import argparse
 import re
+import random
 from pathlib import Path
 from PIL import Image
 import pillow_jxl
@@ -453,9 +454,18 @@ def main():
         const=-1,
         help='Use .txt files with the same base filename as the images as input to the captioner. Optionally specify the number of tags to use.'
     )
+    parser.add_argument(
+        '--random-tags',
+        type=int,
+        help='Randomly select n number of tags. Only works if --feed-from-tags is enabled.'
+    )
 
     args = parser.parse_args()
 
+    # Validate random-tags usage
+    if args.random_tags is not None and args.feed_from_tags is None:
+        parser.error("--random-tags can only be used when --feed-from-tags is enabled")
+
     # Initialize and load models
     joy_caption_model = JoyCaptionModel()
     joy_caption_model.load_models()
@@ -495,7 +505,12 @@ def main():
                         with open(tag_file, 'r', encoding='utf-8') as f:
                             tags = f.read().strip().split(',')
                         
-                        if args.feed_from_tags > 0:
+                        if args.random_tags is not None:
+                            # Randomly select tags if --random-tags is specified
+                            num_tags = min(args.random_tags, len(tags))
+                            tags = random.sample(tags, num_tags)
+                        elif args.feed_from_tags > 0:
+                            # Use specified number of tags if --feed-from-tags has a positive value
                             tags = tags[:args.feed_from_tags]
                         
                         tag_string = ', '.join(tags)