Update app.py

app.py

@@ -10,7 +10,7 @@ from time import sleep
 import copy
 import json
 import gc
-
+import random
 with open("sdxl_loras.json", "r") as file:
     data = json.load(file)
     sdxl_loras = [
@@ -23,6 +23,8 @@ with open("sdxl_loras.json", "r") as file:
             "is_compatible": item["is_compatible"],
             "is_pivotal": item.get("is_pivotal", False),
             "text_embedding_weights": item.get("text_embedding_weights", None),
+            "likes": item.get("likes", 0),
+            "downloads": item.get("downloads", 0),
             "is_nc": item.get("is_nc", False)
         }
         for item in data
@@ -41,6 +43,8 @@ for item in sdxl_loras:
     item["saved_name"] = saved_name
     item["state_dict"] = state_dict #{k: v.to(device=device, dtype=torch.float16) for k, v in state_dict.items() if torch.is_tensor(v)}
 
+gr_sdxl_loras = gr.State(value=sdxl_loras)
+
 vae = AutoencoderKL.from_pretrained(
     "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
 )
@@ -55,7 +59,7 @@ pipe.to(device)
 last_lora = ""
 last_merged = False
 last_fused = False
-def update_selection(selected_state: gr.SelectData):
+def update_selection(selected_state: gr.SelectData, sdxl_loras):
     lora_repo = sdxl_loras[selected_state.index]["repo"]
     instance_prompt = sdxl_loras[selected_state.index]["trigger_word"]
     new_placeholder = "Type a prompt. This LoRA applies for all prompts, no need for a trigger word" if instance_prompt == "" else "Type a prompt to use your selected LoRA"
@@ -135,7 +139,7 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
                 del lora_model
                 gc.collect()
 
-def run_lora(prompt, negative, lora_scale, selected_state, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, negative, lora_scale, selected_state, sdxl_loras, progress=gr.Progress(track_tqdm=True)):
     global last_lora, last_merged, last_fused, pipe
 
     if negative == "":
@@ -195,6 +199,17 @@ def run_lora(prompt, negative, lora_scale, selected_state, progress=gr.Progress(
     gc.collect()
     return image, gr.update(visible=True)
 
+def shuffle_gallery(sdxl_loras):
+    random.shuffle(sdxl_loras)
+    return [(item["image"], item["title"]) for item in sdxl_loras], sdxl_loras
+
+def swap_gallery(order, sdxl_loras):
+    if(order == "random"):
+        return shuffle_gallery(sdxl_loras)
+    else:
+        sorted_gallery = sorted(sdxl_loras, key=lambda x: x.get(order, 0), reverse=True)
+        return [(item["image"], item["title"]) for item in sorted_gallery], sdxl_loras
+
 
 with gr.Blocks(css="custom.css") as demo:
     title = gr.HTML(
@@ -203,14 +218,16 @@ with gr.Blocks(css="custom.css") as demo:
     )
     selected_state = gr.State()
     with gr.Row():
-        gallery = gr.Gallery(
-            value=[(item["image"], item["title"]) for item in sdxl_loras],
-            label="SDXL LoRA Gallery",
-            allow_preview=False,
-            columns=3,
-            elem_id="gallery",
-            show_share_button=False
-        )
+        with gr.Box():
+            order_gallery = gr.Radio(choices=["random", "likes", "downloads"], value="random", label="Order by")
+            gallery = gr.Gallery(
+                value=[(item["image"], item["title"]) for item in sdxl_loras],
+                label="SDXL LoRA Gallery",
+                allow_preview=False,
+                columns=3,
+                elem_id="gallery",
+                show_share_button=False
+            )
         with gr.Column():
             prompt_title = gr.Markdown(
                 value="### Click on a LoRA in the gallery to select it",
@@ -268,9 +285,15 @@ with gr.Blocks(css="custom.css") as demo:
                 submit_disclaimer = gr.Markdown(
                     "This is a curated gallery by me, [apolinário (multimodal.art)](https://twitter.com/multimodalart). I'll try to include as many cool LoRAs as they are submitted! You can [duplicate this Space](https://huggingface.co/spaces/multimodalart/LoraTheExplorer?duplicate=true) to use it privately, and add your own LoRAs by editing `sdxl_loras.json` in the Files tab of your private space."
                 )
-
+    order_gallery.change(
+        fn=swap_gallery,
+        inputs=[order_gallery, gr_sdxl_loras],
+        outputs=[gallery, gr_sdxl_loras],
+        queue=False
+    )
     gallery.select(
         update_selection,
+        inputs=[gr_sdxl_loras],
         outputs=[prompt_title, prompt, prompt, selected_state, use_diffusers, use_uis],
         queue=False,
         show_progress=False,
@@ -282,7 +305,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[prompt, negative, weight, selected_state],
+        inputs=[prompt, negative, weight, selected_state, gr_sdxl_loras],
         outputs=[result, share_group],
     )
     button.click(
@@ -296,6 +319,6 @@ with gr.Blocks(css="custom.css") as demo:
         outputs=[result, share_group],
     )
     share_button.click(None, [], [], _js=share_js)
-
+    demo.load(fn=shuffle_gallery, inputs=[gr_sdxl_loras], outputs=[gallery, gr_sdxl_loras], queue=False)
 demo.queue(max_size=20)
-demo.launch()
+demo.launch(share=True)