aa

tour-comfy/naming.py

@@ -0,0 +1,96 @@
+import random
+import re
+
+def clean_tag(tag):
+    return tag.replace("_", " ").strip()
+
+def generate_associative_name(tags):
+    """
+    Generate a 'real-alike' associative name based on WD tagger tags.
+    tags: list of dicts {'tag': str, 'score': float, 'cat': int}
+    """
+    if not tags:
+        return None
+
+    # Filter by score
+    high_score_tags = [t for t in tags if t['score'] > 0.4]
+    if not high_score_tags:
+        high_score_tags = tags[:5]
+
+    # Categories: 0=general, 4=character
+    characters = [clean_tag(t['tag']) for t in high_score_tags if t['cat'] == 4]
+    general = [clean_tag(t['tag']) for t in high_score_tags if t['cat'] == 0]
+
+    # Key attributes
+    subject = None
+    if characters:
+        subject = characters[0].title()
+    elif "1girl" in [t['tag'] for t in high_score_tags]:
+        subject = "Maiden"
+    elif "1boy" in [t['tag'] for t in high_score_tags]:
+        subject = "Youth"
+    else:
+        subject = "Subject"
+
+    # Actions/Poses
+    actions = ["standing", "sitting", "lying", "running", "walking", "dancing", "sleeping"]
+    found_action = next((clean_tag(t['tag']) for t in high_score_tags if t['tag'] in actions), None)
+
+    # Setting/Background
+    settings = ["forest", "beach", "city", "space", "room", "garden", "ocean", "mountain", "sky", "underwater", "street"]
+    found_setting = next((clean_tag(t['tag']) for t in high_score_tags if t['tag'] in settings), None)
+
+    # Appearance
+    colors = ["red", "blue", "green", "white", "black", "gold", "silver", "pink", "purple", "yellow"]
+    found_color = next((clean_tag(t['tag']) for t in high_score_tags if clean_tag(t['tag']).split()[0] in colors), None)
+    if not found_color:
+         found_color = next((clean_tag(t['tag']) for t in high_score_tags if any(c in t['tag'] for c in colors)), None)
+
+    # Styles/Atmosphere
+    styles = ["cyberpunk", "fantasy", "realistic", "ethereal", "dark", "bright", "sketch", "oil painting"]
+    found_style = next((clean_tag(t['tag']) for t in high_score_tags if t['tag'] in styles), None)
+
+    # Build the name
+    templates = []
+    
+    if found_style and subject:
+        templates.append(f"{found_style.title()} {subject}")
+    
+    if found_color and subject:
+        templates.append(f"The {found_color.title()} {subject}")
+
+    if subject and found_action:
+        if found_action.endswith("ing"):
+            templates.append(f"{subject} {found_action.title()}")
+        else:
+            # Basic attempt at present participle
+            action_ing = found_action
+            if action_ing.endswith("e"):
+                action_ing = action_ing[:-1] + "ing"
+            else:
+                action_ing += "ing"
+            templates.append(f"{subject} {action_ing.title()}")
+
+    if subject and found_setting:
+        templates.append(f"{subject} in the {found_setting.title()}")
+
+    if found_style and found_setting:
+        templates.append(f"{found_style.title()} {found_setting.title()}")
+
+    if not templates:
+        # Fallback: combine two random general tags
+        if len(general) >= 2:
+            return f"{general[0].title()} {general[1].title()}"
+        elif general:
+            return general[0].title()
+        else:
+            return "Untitled Artwork"
+
+    # Return a random template result
+    return random.choice(templates)
+
+
+def get_base_name(name: str) -> str:
+    """Remove timestamp prefixes from filename."""
+    # Matches YYYYMMDD_HHMMSS_
+    return re.sub(r'^(\d{8}_\d{6}_)+', '', name)