Upload app.py

Added checking for duplicate tags. Shuffled components around a bit so that the settings that affect the tag suggestions are above the suggestions, and the settings that affect artist suggestions are above the artist suggestions.

app.py

@@ -414,6 +414,7 @@ def find_similar_tags(test_tags, similarity_weight, allow_nsfw_tags):
     html_content += "<h1>Unknown Tags</h1>"  # Heading for the table
     tags_added = False
     bad_entities = []
+    encountered_modified_tags = set()
     for tag_info in test_tags:
         original_tag = tag_info['original_tag']
         modified_tag = tag_info['modified_tag']
@@ -423,8 +424,15 @@ def find_similar_tags(test_tags, similarity_weight, allow_nsfw_tags):
         #print(original_tag, modified_tag, start_pos, end_pos)
         
         if modified_tag in special_tags:
+            bad_entities.append({"entity":"Special", "start":start_pos, "end":end_pos})
             continue
             
+        if modified_tag in encountered_modified_tags:
+            bad_entities.append({"entity":"Duplicate", "start":start_pos, "end":end_pos})
+            continue
+        encountered_modified_tags.add(modified_tag)
+        
+            
         modified_tag_for_search = modified_tag.replace(' ','_')
         similar_words = find_similar_tags.fasttext_small_model.most_similar(modified_tag_for_search, topn = 100)
         result, seen = [], set(transformed_tags)
@@ -463,7 +471,7 @@ def find_similar_tags(test_tags, similarity_weight, allow_nsfw_tags):
         result = sorted(result, key=lambda x: x[1], reverse=True)[:10]
         html_content += create_html_tables_for_tags(modified_tag, result, find_similar_tags.tag2count, find_similar_tags.tag2idwiki)
         
-        bad_entities.append({"entity":"*", "start":start_pos, "end":end_pos})
+        bad_entities.append({"entity":"Unknown", "start":start_pos, "end":end_pos})
         
         tags_added=True
     # If no tags were processed, add a message
@@ -538,20 +546,24 @@ def find_similar_artists(original_tags_string, top_n, similarity_weight, allow_n
 with gr.Blocks() as app:
     with gr.Group():
         with gr.Row():
-            image_tags = gr.Textbox(label="Enter image tags", placeholder="e.g. fox, outside, detailed background, ...")
-        with gr.Row():
-            submit_button = gr.Button("Submit")
-            similarity_weight = gr.Slider(minimum=0, maximum=1, value=0.5, step=0.1, label="Similarity weight")
-            num_artists = gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Number of artists")
-            allow_nsfw = gr.Checkbox(label="Allow NSFW Tags", value=False)
-    with gr.Row():
-        bad_tags_illustrated_string = gr.HighlightedText(label="Visual depiction of bad tags. Character offsets may be buggy.")
+            with gr.Column(scale=3):
+                image_tags = gr.Textbox(label="Enter Prompt", placeholder="e.g. fox, outside, detailed background, ...")
+                bad_tags_illustrated_string = gr.HighlightedText(show_legend=True,label="Annotated Prompt")
+            with gr.Column(scale=1):
+                gr.HTML("<br>" * 2)  # Adjust the number of line breaks ("<br>") as needed to push the button down
+                submit_button = gr.Button("Submit")
     with gr.Row():
+        with gr.Column(scale=3):    
+            with gr.Group():
+                with gr.Row():
+                    similarity_weight = gr.Slider(minimum=0, maximum=1, value=0.5, step=0.1, label="Similarity weight")
+                    allow_nsfw = gr.Checkbox(label="Allow NSFW Tags", value=False)
+                unseen_tags = gr.HTML(label="Unknown Tags", value=create_html_placeholder(title="Unknown Tags"))
         with gr.Column(scale=1):
-            top_artists = gr.HTML(label="Top Artists", value=create_html_placeholder(title="Top Artists"))
-            dynamic_prompts = gr.Textbox(label="Dynamic Prompts Format", info="For if you're using the Automatic1111 webui (https://github.com/AUTOMATIC1111/stable-diffusion-webui) with the Dynamic Prompts extension activated (https://github.com/adieyal/sd-dynamic-prompts) and want to try them all individually.")
-        with gr.Column(scale=3):
-           unseen_tags = gr.HTML(label="Unknown Tags", value=create_html_placeholder(title="Unknown Tags"))
+            with gr.Group():
+                num_artists = gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Number of artists")
+                top_artists = gr.HTML(label="Top Artists", value=create_html_placeholder(title="Top Artists"))
+                dynamic_prompts = gr.Textbox(label="Dynamic Prompts Format", info="For if you're using the Automatic1111 webui (https://github.com/AUTOMATIC1111/stable-diffusion-webui) with the Dynamic Prompts extension activated (https://github.com/adieyal/sd-dynamic-prompts) and want to try them all individually.")
     galleries = []
     for root, dirs, files in os.walk(sample_images_directory_path):
         for name in dirs: