sample images

app.py

@@ -13,7 +13,12 @@ from collections import OrderedDict
 from lark import Lark
 from lark import Token
 from lark.exceptions import ParseError
-
+import json
+import zipfile
+from PIL import Image
+import io
+import os
+    
 
 
 
@@ -178,6 +183,24 @@ with open("word_rating_probabilities.csv", 'r', newline='', encoding='utf-8') as
             nsfw_tags.add(word)
     
     
+soyjak_directory_path = 'artistsoyjaks'
+soyjak_json_file_path = os.path.join(soyjak_directory_path, 'artistsoyjaks.json')
+with open(soyjak_json_file_path, 'r') as json_file:
+    soyjak_artist_to_file_map = json.load(json_file)
+
+    
+def generate_artist_image_tuples(top_artists, image_directory=soyjak_directory_path):
+    artist_image_tuples = []
+    for artist in top_artists:
+        filename = soyjak_artist_to_file_map.get(artist)
+        if filename:
+            image_path = os.path.join(image_directory, filename)
+            if os.path.exists(image_path):
+                artist_image_tuples.append((image_path, artist if artist else "No Artist"))
+
+    return artist_image_tuples
+    
+    
 def clean_tag(tag):
     return ''.join(char for char in tag if ord(char) < 128)
     
@@ -412,27 +435,42 @@ def find_similar_artists(new_tags_string, top_n, similarity_weight, allow_nsfw_t
         top_artists_str = "\n".join([f"{rank+1}. {artist[3:]} ({score:.4f})" for rank, (artist, score) in enumerate(top_artists)])
         dynamic_prompts_formatted_artists = "{" + "|".join([artist for artist, _ in top_artists]) + "}"
 
-        return unseen_tags_data, top_artists_str, dynamic_prompts_formatted_artists
+        image_gallery = generate_artist_image_tuples([''] + [name[3:] for name, _ in top_artists])
+
+        return unseen_tags_data, top_artists_str, dynamic_prompts_formatted_artists, image_gallery
     except ParseError as e:
-        return [], "Parse Error: Check for mismatched parentheses or something", ""
-
-
-iface = gr.Interface(
-    fn=find_similar_artists,
-    inputs=[
-        gr.Textbox(label="Enter image tags", placeholder="e.g. fox, outside, detailed background, ..."),
-        gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Number of artists"),
-        gr.Slider(minimum=0, maximum=1, value=0.5, step=0.1, label="Similarity weight"), 
-        gr.Checkbox(label="Allow NSFW Tags", value=False)
-    ],
-    outputs=[
-        gr.HTML(label="Unseen Tags"),
-        gr.Textbox(label="Top Artists", info="These are the artists most strongly associated with your tags.  The number in parenthes is a similarity score between 0 and 1, with higher numbers indicating greater similarity."),
-        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.") 
-    ],
-    title="Tagset Completer",
-    description="Enter a list of comma-separated e6 tags",
-    article=faq_content 
-)
-
-iface.launch()
+        return [], "Parse Error: Check for mismatched parentheses or something", "", None
+
+
+with gr.Blocks() as app:
+    with gr.Group():
+        with gr.Column(scale=2):
+            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")
+                allow_nsfw = gr.Checkbox(label="Allow NSFW Tags", value=False)
+        with gr.Column(scale=1):
+            unseen_tags = gr.HTML(label="Unseen Tags")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            num_artists = gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Number of artists")
+            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.")
+            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):
+            styles = gr.Gallery(label="Styles", info="This is what Fluffyrock Unleashed, an e621 model present in many mixes, thinks the recommended artists' styles look like.", allow_preview=True, preview=True)
+    
+    
+    submit_button.click(
+        find_similar_artists, 
+        inputs=[image_tags, num_artists, similarity_weight, allow_nsfw], 
+        outputs=[unseen_tags, top_artists, dynamic_prompts, styles]
+    )
+    
+    gr.Markdown(faq_content)
+    
+
+app.launch()
+