Update app.py

app.py

@@ -228,8 +228,18 @@ def main():
             filter_set = set(tag.strip().lower() for tag in filter_tags.split(","))
             
             # Parse custom tags
-            custom_tag_set = set(tag.strip().lower() for tag in custom_tags.split(","))
+            custom_tag_list = [tag.strip().lower() for tag in custom_tags.split(",")]
             
+            # Define the fallback tags for each custom tag
+            fallback_tags = {
+                "shy, happy": "smile",
+                "happy": "joy",
+                "cute": "adorable",
+                "energetic": "vibrant",
+                "blush": "flushed",
+                # Add more custom to fallback mappings as needed
+            }
+        
             # Generate formatted output
             prompts = []
             for i, (general_tags, character_tags) in enumerate(results):
@@ -241,10 +251,12 @@ def main():
                     tag.replace('_', ' ') for tag in general_tags if tag.lower() not in filter_set
                 )
                 
-                # Check for missing custom tags
-                missing_custom_tags = custom_tag_set - set(general_tags + character_tags)
-                if missing_custom_tags:
-                    general_part += ", " + ", ".join(missing_custom_tags)
+                # Check for missing custom tags and apply corresponding fallback tags
+                for custom_tag in custom_tag_list:
+                    if custom_tag not in (general_tags + character_tags):
+                        # Append the corresponding fallback tag for the missing custom tag
+                        if custom_tag in fallback_tags:
+                            general_part += ", " + fallback_tags[custom_tag]
                 
                 # Construct the prompt based on the presence of character_part
                 if character_part:
@@ -255,7 +267,6 @@ def main():
             # Join all prompts with blank lines
             return "\n\n".join(prompts)
 
-
         submit.click(
             process_images,
             inputs=[image_files, model_repo, general_thresh, character_thresh, filter_tags, custom_tags],