Update app.py

app.py

@@ -313,13 +313,16 @@ def shuffle_gallery(sdxl_loras):
     random.shuffle(sdxl_loras)
     return [(item["image"], item["title"]) for item in sdxl_loras], sdxl_loras
 
+def classify_gallery(sdxl_loras):
+    sorted_gallery = sorted(sdxl_loras, key=lambda x: x.get("likes", 0), reverse=True)
+    return [(item["image"], item["title"]) for item in sorted_gallery], sorted_gallery
+
 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], sorted_gallery
-
+        return classify_gallery(sdxl_loras)
+        
 def deselect():
   return gr.Gallery(selected_index=None)
 
@@ -335,7 +338,7 @@ with gr.Blocks(css="custom.css") as demo:
             with gr.Group(elem_id="gallery_box"):
                 photo = gr.Image(label="Upload a picture of yourself", interactive=True, type="pil", height=300)
                 selected_loras = gr.Gallery(label="Selected LoRAs", height=80, show_share_button=False, visible=False, elem_id="gallery_selected", )
-                order_gallery = gr.Radio(choices=["random", "likes"], value="random", label="Order by", elem_id="order_radio")
+                #order_gallery = gr.Radio(choices=["random", "likes"], value="random", label="Order by", elem_id="order_radio")
                 #new_gallery = gr.Gallery(
                 #    label="New LoRAs",
                 #    elem_id="gallery_new",
@@ -348,8 +351,9 @@ with gr.Blocks(css="custom.css") as demo:
                     columns=4,
                     elem_id="gallery",
                     show_share_button=False,
-                    height=784
+                    height=450
                 )
+                custom_model = gr.Textbox(label="Enter a custom Hugging Face or CivitAI SDXL LoRA", disabled=True, info="Coming soon...")
         with gr.Column(scale=5):
             with gr.Row():
                 prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, info="Describe your subject (optional)", value="A person", elem_id="prompt")
@@ -373,12 +377,12 @@ with gr.Blocks(css="custom.css") as demo:
                 visible=True,
                 elem_id="selected_lora",
             )
-    order_gallery.change(
-        fn=swap_gallery,
-        inputs=[order_gallery, gr_sdxl_loras],
-        outputs=[gallery, gr_sdxl_loras],
-        queue=False
-    )
+    #order_gallery.change(
+    #    fn=swap_gallery,
+    #    inputs=[order_gallery, gr_sdxl_loras],
+    #    outputs=[gallery, gr_sdxl_loras],
+    #    queue=False
+    #)
     gallery.select(
         fn=update_selection,
         inputs=[gr_sdxl_loras, face_strength, image_strength, weight, depth_control_scale, negative],
@@ -414,6 +418,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, js=js)
+    demo.load(fn=classify_gallery, inputs=[gr_sdxl_loras], outputs=[gallery, gr_sdxl_loras], queue=False, js=js)
 demo.queue(max_size=20)
 demo.launch(share=True)