Upload app.py

app.py

@@ -341,7 +341,7 @@ def geometric_mean_given_words(target_word, context_words, co_occurrence_matrix,
     
 def create_html_tables_for_tags(tag, result, tag2count, tag2idwiki):
     # Wrap the tag part in a <span> with styles for bold and larger font
-    html_str = f"<div style='display: inline-block; margin: 20px; vertical-align: top;'><table><thead><tr><th colspan='3' style='text-align: center; padding-bottom: 10px;'>Unknown Tag: <span style='font-weight: bold; font-size: 20px;'>{tag}</span></th></tr></thead><tbody><tr style='border-bottom: 1px solid #000;'><th>Corrected Tag</th><th>Similarity</th><th>Count</th></tr>"
+    html_str = f"<div style='display: inline-block; margin: 10px; vertical-align: top;'><table><thead><tr><th colspan='3' style='text-align: center; padding-bottom: 10px;'><span style='font-weight: bold; font-size: 20px;'>{tag}</span></th></tr></thead><tbody><tr style='border-bottom: 1px solid #000;'><th>Corrected Tag</th><th>Similarity</th><th>Count</th></tr>"
     # Loop through the results and add table rows for each
     for word, sim in result:
         word_with_underscores = word.replace(' ', '_')
@@ -361,6 +361,33 @@ def create_html_tables_for_tags(tag, result, tag2count, tag2idwiki):
 
     html_str += "</tbody></table></div>"
     return html_str
+    
+    
+def create_top_artists_table(top_artists):
+    # Add a heading above the table
+    html_str = "<div style='display: inline-block; margin: 20px; text-align: center;'>"
+    html_str += "<h1>Top Artists</h1>"  # Heading for the table
+    # Start the table with increased font size and no borders between rows
+    html_str += "<table style='font-size: 20px; border-collapse: collapse;'>"
+    html_str += "<thead><tr><th>Artist</th><th>Similarity</th></tr></thead><tbody>"
+    # Loop through the top artists and add a row for each without the rank and without borders between rows
+    for artist, score in top_artists:
+        artist_name = artist[3:] if artist.startswith("by ") else artist  # Remove "by " prefix
+        similarity_percentage = "{:.1f}%".format(score * 100)  # Convert score to percentage string with one decimal
+        html_str += f"<td style='padding: 3px 20px; border: none;'>{artist_name}</td><td style='padding: 3px 20px; border: none;'>{similarity_percentage}</td></tr>"
+
+    # Close the table HTML
+    html_str += "</tbody></table></div>"
+
+    return html_str
+
+
+def create_html_placeholder(title="", placeholder_height=400, placeholder_width="100%"):
+    # Include a title in the same style as the top artists table heading
+    html_placeholder = f"<div style='text-align: center;'><h1>{title}</h1></div>"
+    # Add the placeholder div with specified height and width
+    html_placeholder += f"<div style='height: {placeholder_height}px; width: {placeholder_width}; margin: 20px; background: transparent;'></div>"
+    return html_placeholder
 
 
 def find_similar_tags(test_tags, similarity_weight, allow_nsfw_tags):
@@ -380,7 +407,9 @@ def find_similar_tags(test_tags, similarity_weight, allow_nsfw_tags):
     transformed_tags = [tag.replace(' ', '_') for tag in test_tags]
 
     # Find similar tags and prepare data for tables
-    html_content = ""
+    html_content = "<div style='display: inline-block; margin: 20px; text-align: center;'>"
+    html_content += "<h1>Unknown Tags</h1>"  # Heading for the table
+    tags_added = False
     for tag in test_tags:
         if tag in special_tags:
             continue
@@ -422,9 +451,10 @@ 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(tag, result, find_similar_tags.tag2count, find_similar_tags.tag2idwiki)
+        tags_added=True
     # If no tags were processed, add a message
-    if not html_content:
-        html_content = "<p>No Unknown Tags Found</p>"
+    if not tags_added:
+        html_content = create_html_placeholder(title="Unknown Tags")
 
     return html_content  # Return list of lists for Dataframe
 
@@ -446,10 +476,11 @@ def find_similar_artists(new_tags_string, top_n, similarity_weight, allow_nsfw_t
         X_new_image = vectorizer.transform([','.join(new_image_tags + removed_tags)])
         similarities = cosine_similarity(X_new_image, X_artist)[0]
 
-        top_artist_indices = np.argsort(similarities)[-top_n:][::-1]
-        top_artists = [(artist_names[i], similarities[i]) for i in top_artist_indices]
-
-        top_artists_str = "\n".join([f"{rank+1}. {artist[3:]} ({score:.4f})" for rank, (artist, score) in enumerate(top_artists)])
+        top_artist_indices = np.argsort(similarities)[-(top_n + 1):][::-1]
+        top_artists = [(artist_names[i], similarities[i]) for i in top_artist_indices if artist_names[i].lower() != "by conditional dnp"][:top_n]
+        
+        #top_artists_str = "\n".join([f"{rank+1}. {artist[3:]} ({score:.4f})" for rank, (artist, score) in enumerate(top_artists)])
+        top_artists_str = create_top_artists_table(top_artists)
         dynamic_prompts_formatted_artists = "{" + "|".join([artist for artist, _ in top_artists]) + "}"
 
         image_galleries = []
@@ -474,12 +505,11 @@ with gr.Blocks() as app:
             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():
-        with gr.Column():
-            top_artists = gr.Textbox(label="Top Artists", info="These are the artists most strongly associated with your tags. The number in parentheses is a similarity score between 0 and 1, with higher numbers indicating greater similarity.")
-        with gr.Column():
+        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.Row():
-        unseen_tags = gr.HTML(label="Unseen Tags")
+        with gr.Column(scale=3):
+           unseen_tags = gr.HTML(label="Unknown Tags", value=create_html_placeholder(title="Unknown Tags"))
     galleries = []
     for root, dirs, files in os.walk(sample_images_directory_path):
         for name in dirs: