Minthy
/

ToriiGate-0.5

@@ -1,220 +1,220 @@
-import random
-prompts_b = {
-"long_thoughts_v2": """Your answer must contain 6 parts:
-<format>
-# 1. Thoughts about characters
-You need to think here and compare peoples/creatures that you see on the picture with given popular tags, or descriptions, or your memories for each characters to determine who is who.
-# 2. Key details
-Here you need to determine key details on comic and list them.
-# 3. Long description
-Here come up with a long and detailed description of image content. Be creative, mention all detailes you listed above and other important things.
-# 4. Detailed description for each character
-## Name 1
-Detailed and long description for the first character
-## Name 2
-Same for each one (if present)
-</format>
-""",
-"long_thoughts": """Your answer must contain 6 parts:
-<format>
-# 1. Thoughts about characters
-You need to think here and compare peoples/creatures that you see on the picture  with given popular tags, or descriptions, or your memories for each characters to determine who is who.
-If no characters are listed in input - just write here "No named characters"
-# 2. General description
-A one-two paragraph summary of the image. Mention all individual parts/objects/characters/positions/interactions/etc.
-# 3. Detailed description for each character
-## Character name 1 (put here the name if any)
-In very detail write about features, poses, look, used objects, interactions, and other things for character on the picture.
-## Character name 2 (put here the name if any)
-Same for each character.
-...
-# 4. Individual Parts
-List the individual things you see in the image and their relative positions to other parts. Use a numbered list of between 5 and 20 items depending on image complexity.
-# 5. Texts on image
-Mention every texts that you notice on image, including types (a speech bubble, watermark, banner, etc.) and content.
-# 6. Background and effects
-Give some info about objects on background, describe the location (if seen). Then mention effects (style, camera angle, clarity/blurrines, effects like depth of field, strange angle/forshortening, etc.)
-</format>
-""",
-"json": """Use json-style caption for given image with following structure:
-{"character" : "Description for character or object. Name (if defined), main details, features, position, pose, etc.",
-/or in case of multiple
-"character_1" : "Description for first"
-"character_2" : "Description for second ",
-"character_N"...
-/or if there are no characters
-"main content" : "long and detailed description of main content of image that might be the main focus if characters are missing",
-/
-"background" : "Detailed descritpion of background and it's content",
-"image_effects" : "If there are some visual effects like fisheye distortion, chromatic aberration, glitches, messy drawing or anything else - write about it. If it's just a general anime art - omit this field."
-"texts" : "Speech bubbles, bars, marks, signs etc. with texts if present, else None",
-"atmosphere" : "...",
-}
-In special cases you can add extra keys.
-""",
-"long": """Make a caption for given image with natural text. Use 2 to 5 paragraphs. Make your description long and vivid, mentioning all the details.
-""",
-"min_structured_md": """Your answer must contain 3 parts:
-<format>
-# 1. Thoughts about characters
-You need to think here and compare peoples/creatures that you see on the picture  with given popular tags, or descriptions, or your memories for each characters to determine who is who.
-If no characters are listed in input - just write here "No named characters"
-# 2. Key details
-Here you need to write about the key details on image, prefere using regular text.
-# 3. Structured description
-## General
-Write about general composition, content of image, background and all things that are not related to characters directly.
-## Character name 1 (put here the name if any)
-Write about datails and content related to specific character, including features, poses, look, used objects, interactions, and other things.
-## Character name 2 (put here the name if any)
-Same for each character.
-## Image effects
-Mention image effect, style, camera angle
-</format>
-In general stick to shorter descriptions.
-""",
-"json_comic": """Use json-style caption to describe to comin, stick to following structure:
-{
-"comic_format": "menation the format, for example Comic of N frames",
-"1st_frame": "Main description of the content for fist frame",
-"2nd_frame": "Same for the second",
-...
-"Nth_ftame": "...",
-"character_1": "Describe the characters in comic",
-...
-"character_N": "Separate description for each",
-"meaning": "Try to guess general mood, vibe and meaning of the comic"
-}
-""",
-"md_comic": """Use markdown format to describe to comic, 5 parts are recommended:
-<format>
-# 1. Thoughts about characters
-You need to think here and compare peoples/creatures that you see on the picture with given popular tags, or descriptions, or your memories for each characters to determine who is who.
-# 2. Key details
-Here you need to determine key details on comic and list them.
-# 3. Comic format
-In this section come up with the description of comic format, how many pages there are, horisontal/vertical orientation and other things. Optionally you can list main characters here.
-# 4. Details for each frame
-## 4.1 Frame 1 (position)
-Description for each frame, includding characters, objects, interactions, texts/speech bubbles and other things. Be detailed but not overdoo.
-## 4.2 Frame 2 (position)
-Same for each frame.
-...
-# 5. Extra comment
-Here you should write general desciption and some other info about the image.
-</format>
-""",
-"min_structured_json": """
-Use json-style caption for given image with following structure:
-{"General" : "Here you need to come up with general/common information about picture, overall composition. Stick to shorter phrases and tags instead of long purple prose. Avoid bullets and markdown, write in plain text.",
-"character_1 (put here the name if any)" : "Description of first character."
-"character_2 (if present" : "Description for second ",
-"character_N"
-...
-"image_effects" : "Mention here effects on image if there are any distinct."
-"texts" : "Speech bubbles, bars, marks, signs etc. with texts if present, else None",
-"watermarks" : "If present",
-}
-Prefere shorter description and tags.
-""",
-"chroma-style": """Your task is to describe the picture in very detail using a structure of 4 parts.
-### 1. Regular Summary:
-[A one-paragraph summary of the image. The paragraph should mention all individual parts/things/characters/etc.]
-### 2. Individual Parts:
-[List the individual things you see in the image and their relative positions to other parts. Use a numbered list of between 5 and 30 items depending on image complexity.]
-### 3. Midjourney-Style Summary:
-[A summary that has higher concept density by using comma-separated partial sentences instead of proper sentence structure.]
-### 4. DeviantArt Commission Request
-[Write a description as if you're commissioning this *exact* image via someone who is currently taking requests.]
-""",
-"short":"""The caption for image should be quite short without long purple prose and slop. Cover main objects and details.
-""",
-}
-prompts_names_only = {
-    "long_thoughts_v2":True,
-    "long_thoughts": True,
-    "json": False,
-    "long": False,
-    "json_comic": False,
-    "md_comic": True,
-    "min_structured_md": True,
-    "min_structured_json": False,
-    "chroma-style": False,
-    "short":False,
-    }
-def make_user_query(item, c_type, use_names, add_tags, add_characters, add_char_tags, add_description, underscores_replace = False):
-    tags = item.get('tags', [])
-    random.shuffle(tags)
-    if underscores_replace:
-        tags = [a.replace('_', ' ') if len(a)>3 else a for a in tags]
-        tags_string = ', '.join(tags)
-    else:
-        tags_string = ' '.join(tags)
-    user_request = '# Captioning format:\n'
-    user_request += prompts_b[c_type]
-    user_request += '\n'
-    if add_tags:
-        user_request += f"# Booru tags for the image\n[{tags_string}]\n\n"
-    if use_names: #Имена персонажей
-        if add_characters:
-            chars_tags = item.get('characters',[])
-            if underscores_replace:
-                tags = [a.replace('_', ' ') for a in chars_tags]
-                chars_string = ', '.join(chars_tags)
-            else:
-                chars_string = ' '.join(chars_tags)
-            user_request += f"# Characters on picture:\nHere are names/tags for characters from the picture, make sure to use them: [{chars_string}].\n\n"
-            chars_popular_tags = (item.get('char_p_tags',"{'chars':{},'skins':{}}"))
-            chars_description = (item.get('char_descr',"{'chars':{},'skins':{}}"))
-            if len(chars_popular_tags['chars']) > 0 and (add_char_tags or add_description):
-                user_request += "# Known traits for characters\n"
-                char_underscores = underscores_replace
-                if add_char_tags:
-                    user_request += "Here are popular tags for each characters on picture:\n"
-                    for c_name, c_tags in chars_popular_tags['chars'].items():
-                        name = c_name.replace('_',' ') if char_underscores else c_name
-                        tags_s = (', '.join([a.replace('_', ' ') if len(a)>3 else a for a in c_tags]) if char_underscores else
-                                ' '.join(c_tags))
-                        user_request += f"{name}: [{tags_s}]\n"
-                    if len(chars_popular_tags['skins']) > 0:
-                        user_request += "Extra tags for characters skins:\n"
-                        for c_name, c_tags in chars_popular_tags['skins'].items():
-                            name = c_name.replace('_',' ') if char_underscores else c_name
-                            tags_s = (', '.join([a.replace('_', ' ') if len(a)>3 else a for a in c_tags]) if char_underscores else
-                                    ' '.join(c_tags))
-                            user_request += f"{name}: [{tags_s}]\n"
-                elif add_description:
-                    user_request += "Here are general descriptions for each characters on the picture:\n"
-                    for c_name, c_descr in chars_description['chars'].items():
-                        name = c_name.replace('_',' ') if char_underscores else c_name
-                        user_request += f"## {name}\n{c_descr}\n\n"
-                    if len(chars_description['skins']) > 0:
-                        user_request += "Here are also descriptions for specific skin of characters:\n"
-                        for c_name, c_descr in chars_description['skins'].items():
-                            name = c_name.replace('_',' ') if char_underscores else c_name
-                            user_request += f"## {name}\n{c_descr}\n\n"
-        else:
-            user_request += "# Characters on picture:\nTry to recognize the characters in the picture and use their names.\n"
-        user_request += '\n'
-    else:
-        user_request += "# Characters on picture:\nAvoid to guess names for characters.\n"
-    return user_request
 system_prompt = "You are image captioning expert. Describe user's picture according to requested format and instructions."

+import random
+prompts_b = {
+"long_thoughts_v2": """Your answer must contain 6 parts:
+<format>
+# 1. Thoughts about characters
+You need to think here and compare peoples/creatures that you see on the picture with given popular tags, or descriptions, or your memories for each characters to determine who is who.
+# 2. Key details
+Here you need to determine key details on comic and list them.
+# 3. Long description
+Here come up with a long and detailed description of image content. Be creative, mention all detailes you listed above and other important things.
+# 4. Detailed description for each character
+## Name 1
+Detailed and long description for the first character
+## Name 2
+Same for each one (if present)
+</format>
+""",
+"long_thoughts": """Your answer must contain 6 parts:
+<format>
+# 1. Thoughts about characters
+You need to think here and compare peoples/creatures that you see on the picture  with given popular tags, or descriptions, or your memories for each characters to determine who is who.
+If no characters are listed in input - just write here "No named characters"
+# 2. General description
+A one-two paragraph summary of the image. Mention all individual parts/objects/characters/positions/interactions/etc.
+# 3. Detailed description for each character
+## Character name 1 (put here the name if any)
+In very detail write about features, poses, look, used objects, interactions, and other things for character on the picture.
+## Character name 2 (put here the name if any)
+Same for each character.
+...
+# 4. Individual Parts
+List the individual things you see in the image and their relative positions to other parts. Use a numbered list of between 5 and 20 items depending on image complexity.
+# 5. Texts on image
+Mention every texts that you notice on image, including types (a speech bubble, watermark, banner, etc.) and content.
+# 6. Background and effects
+Give some info about objects on background, describe the location (if seen). Then mention effects (style, camera angle, clarity/blurrines, effects like depth of field, strange angle/forshortening, etc.)
+</format>
+""",
+"json": """Use json-style caption for given image with following structure:
+{"character" : "Description for character or object. Name (if defined), main details, features, position, pose, etc.",
+/or in case of multiple
+"character_1" : "Description for first"
+"character_2" : "Description for second ",
+"character_N"...
+/or if there are no characters
+"main content" : "long and detailed description of main content of image that might be the main focus if characters are missing",
+/
+"background" : "Detailed descritpion of background and it's content",
+"image_effects" : "If there are some visual effects like fisheye distortion, chromatic aberration, glitches, messy drawing or anything else - write about it. If it's just a general anime art - omit this field."
+"texts" : "Speech bubbles, bars, marks, signs etc. with texts if present, else None",
+"atmosphere" : "...",
+}
+In special cases you can add extra keys.
+""",
+"long": """Make a caption for given image with natural text. Use 2 to 5 paragraphs. Make your description long and vivid, mentioning all the details.
+""",
+"min_structured_md": """Your answer must contain 3 parts:
+<format>
+# 1. Thoughts about characters
+You need to think here and compare peoples/creatures that you see on the picture  with given popular tags, or descriptions, or your memories for each characters to determine who is who.
+If no characters are listed in input - just write here "No named characters"
+# 2. Key details
+Here you need to write about the key details on image, prefere using regular text.
+# 3. Structured description
+## General
+Write about general composition, content of image, background and all things that are not related to characters directly.
+## Character name 1 (put here the name if any)
+Write about datails and content related to specific character, including features, poses, look, used objects, interactions, and other things.
+## Character name 2 (put here the name if any)
+Same for each character.
+## Image effects
+Mention image effect, style, camera angle
+</format>
+In general stick to shorter descriptions.
+""",
+"json_comic": """Use json-style caption to describe to comin, stick to following structure:
+{
+"comic_format": "menation the format, for example Comic of N frames",
+"1st_frame": "Main description of the content for fist frame",
+"2nd_frame": "Same for the second",
+...
+"Nth_ftame": "...",
+"character_1": "Describe the characters in comic",
+...
+"character_N": "Separate description for each",
+"meaning": "Try to guess general mood, vibe and meaning of the comic"
+}
+""",
+"md_comic": """Use markdown format to describe to comic, 5 parts are recommended:
+<format>
+# 1. Thoughts about characters
+You need to think here and compare peoples/creatures that you see on the picture with given popular tags, or descriptions, or your memories for each characters to determine who is who.
+# 2. Key details
+Here you need to determine key details on comic and list them.
+# 3. Comic format
+In this section come up with the description of comic format, how many pages there are, horisontal/vertical orientation and other things. Optionally you can list main characters here.
+# 4. Details for each frame
+## 4.1 Frame 1 (position)
+Description for each frame, includding characters, objects, interactions, texts/speech bubbles and other things. Be detailed but not overdoo.
+## 4.2 Frame 2 (position)
+Same for each frame.
+...
+# 5. Extra comment
+Here you should write general desciption and some other info about the image.
+</format>
+""",
+"min_structured_json": """
+Use json-style caption for given image with following structure:
+{"General" : "Here you need to come up with general/common information about picture, overall composition. Stick to shorter phrases and tags instead of long purple prose. Avoid bullets and markdown, write in plain text.",
+"character_1 (put here the name if any)" : "Description of first character."
+"character_2 (if present" : "Description for second ",
+"character_N"
+...
+"image_effects" : "Mention here effects on image if there are any distinct."
+"texts" : "Speech bubbles, bars, marks, signs etc. with texts if present, else None",
+"watermarks" : "If present",
+}
+Prefere shorter description and tags.
+""",
+"chroma-style": """Your task is to describe the picture in very detail using a structure of 4 parts.
+### 1. Regular Summary:
+[A one-paragraph summary of the image. The paragraph should mention all individual parts/things/characters/etc.]
+### 2. Individual Parts:
+[List the individual things you see in the image and their relative positions to other parts. Use a numbered list of between 5 and 30 items depending on image complexity.]
+### 3. Midjourney-Style Summary:
+[A summary that has higher concept density by using comma-separated partial sentences instead of proper sentence structure.]
+### 4. DeviantArt Commission Request
+[Write a description as if you're commissioning this *exact* image via someone who is currently taking requests.]
+""",
+"short":"""The caption for image should be quite short without long purple prose and slop. Cover main objects and details.
+""",
+}
+prompts_names_only = {
+    "long_thoughts_v2":True,
+    "long_thoughts": True,
+    "json": False,
+    "long": False,
+    "json_comic": False,
+    "md_comic": True,
+    "min_structured_md": True,
+    "min_structured_json": False,
+    "chroma-style": False,
+    "short":False,
+    }
+def make_user_query(item, c_type, use_names, add_tags, add_characters, add_char_tags, add_description, underscores_replace = False):
+    tags = item.get('tags', [])
+    random.shuffle(tags)
+    if underscores_replace:
+        tags = [a.replace('_', ' ') if len(a)>3 else a for a in tags]
+        tags_string = ', '.join(tags)
+    else:
+        tags_string = ' '.join(tags)
+    user_request = '# Captioning format:\n'
+    user_request += prompts_b[c_type]
+    user_request += '\n'
+    if add_tags:
+        user_request += f"# Booru tags for the image\n[{tags_string}]\n\n"
+    if use_names: #Имена персонажей
+        if add_characters:
+            chars_tags = item.get('characters',[])
+            if underscores_replace:
+                chars_tags = [a.replace('_', ' ') for a in chars_tags]
+                chars_string = ', '.join(chars_tags)
+            else:
+                chars_string = ' '.join(chars_tags)
+            user_request += f"# Characters on picture:\nHere are names/tags for characters from the picture, make sure to use them: [{chars_string}].\n\n"
+            chars_popular_tags = (item.get('char_p_tags',"{'chars':{},'skins':{}}"))
+            chars_description = (item.get('char_descr',"{'chars':{},'skins':{}}"))
+            if len(chars_popular_tags['chars']) > 0 and (add_char_tags or add_description):
+                user_request += "# Known traits for characters\n"
+                char_underscores = underscores_replace
+                if add_char_tags:
+                    user_request += "Here are popular tags for each characters on picture:\n"
+                    for c_name, c_tags in chars_popular_tags['chars'].items():
+                        name = c_name.replace('_',' ') if char_underscores else c_name
+                        tags_s = (', '.join([a.replace('_', ' ') if len(a)>3 else a for a in c_tags]) if char_underscores else
+                                ' '.join(c_tags))
+                        user_request += f"{name}: [{tags_s}]\n"
+                    if len(chars_popular_tags['skins']) > 0:
+                        user_request += "Extra tags for characters skins:\n"
+                        for c_name, c_tags in chars_popular_tags['skins'].items():
+                            name = c_name.replace('_',' ') if char_underscores else c_name
+                            tags_s = (', '.join([a.replace('_', ' ') if len(a)>3 else a for a in c_tags]) if char_underscores else
+                                    ' '.join(c_tags))
+                            user_request += f"{name}: [{tags_s}]\n"
+                elif add_description:
+                    user_request += "Here are general descriptions for each characters on the picture:\n"
+                    for c_name, c_descr in chars_description['chars'].items():
+                        name = c_name.replace('_',' ') if char_underscores else c_name
+                        user_request += f"## {name}\n{c_descr}\n\n"
+                    if len(chars_description['skins']) > 0:
+                        user_request += "Here are also descriptions for specific skin of characters:\n"
+                        for c_name, c_descr in chars_description['skins'].items():
+                            name = c_name.replace('_',' ') if char_underscores else c_name
+                            user_request += f"## {name}\n{c_descr}\n\n"
+        else:
+            user_request += "# Characters on picture:\nTry to recognize the characters in the picture and use their names.\n"
+        user_request += '\n'
+    else:
+        user_request += "# Characters on picture:\nAvoid to guess names for characters.\n"
+    return user_request
 system_prompt = "You are image captioning expert. Describe user's picture according to requested format and instructions."