more --feed-from-tags

joy

@@ -448,8 +448,10 @@ def main():
     )
     parser.add_argument(
         '--feed-from-tags',
-        action='store_true',
-        help='Use .txt files with the same base filename as the images as input to the captioner'
+        type=int,
+        nargs='?',
+        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.'
     )
 
     args = parser.parse_args()
@@ -487,11 +489,18 @@ def main():
                 )
             else:
                 # Check for --feed-from-tags
-                if args.feed_from_tags:
+                if args.feed_from_tags is not None:
                     tag_file = find_tag_file(image_path)
                     if tag_file:
                         with open(tag_file, 'r', encoding='utf-8') as f:
-                            custom_prompt = f.read().strip()
+                            tags = f.read().strip().split(',')
+                        
+                        if args.feed_from_tags > 0:
+                            tags = tags[:args.feed_from_tags]
+                        
+                        tag_string = ', '.join(tags)
+                        custom_prompt = f"Write a descriptive caption for this image in a formal tone. Use these tags as context clues to construct your caption: {tag_string}"
+                        
                         caption = joy_caption_model.process_image(
                             input_image,
                             "custom",
@@ -523,7 +532,7 @@ def main():
                 if args.add_commas_to_sentence_ends:
                     caption = re.sub(r'(\.)(\s+)([A-Z])', r'\1,\2\3', caption)
 
-            print(f"Caption for {image_path}:\n{caption}\n")
+            print(f"Caption for {image_path}:\n\n{caption}\n\n")
 
             # Save the caption to a .caption file
             with open(caption_file, 'w', encoding='utf-8') as f: