Update app.py

added 2 more custom lora slots for more variance capability It will now be possable to have 4 celebrity's together in the same image, or fewer but with more styled extra options. or various combinations.

-Enjoy and remember to use responsibly. This is a public service and I will not be held accountable for what you do with it.

;p

app.py

@@ -575,7 +575,7 @@ def update_history(new_image, history):
 css = '''
 #gen_btn{height: 100%}
 #title{text-align: center}
-#title h1{font-size: 3em; display:inline-flex; align-items:center}
+#title h1{font-size: 2em; display:inline-flex; align-items:center}
 #title img{width: 100px; margin-right: 0.25em}
 #gallery .grid-wrap{height: 5vh}
 #lora_list{background: var(--block-background-fill);padding: 0 1em .3em; font-size: 90%}
@@ -594,14 +594,10 @@ css = '''
 #random_btn{font-size: 300%}
 #component-11{align-self: stretch;}
 '''
-
-with gr.Blocks(css=css, delete_cache=(240, 240)) as app:
+font=[gr.themes.GoogleFont("Source Sans Pro"), "Arial", "sans-serif"]
+with gr.Blocks(theme=gr.themes.Soft(font=font), css=css, delete_cache=(256, 256)) as app:
     title = gr.HTML(
-        """<h1><img src="https://huggingface.co/spaces/Keltezaa/Celebrity_LoRa_Mix/blob/main/solo-traveller_16875043.png" alt=" "> Celebrity LoRa Mix</h1><br><span style="
-    margin-top: -25px !important;
-    display: block;
-    margin-left: 37px;
-">SFW & NSFW FLUX LoRAs</span>""",
+        """<h1><img src="https://huggingface.co/spaces/keltezaa/Celebrity_pony_NSFW_Mix/resolve/main/solo-traveller_16875043.png" alt="LoRA"> Celebrity Pony NSFW Mix</h1>""",
         elem_id="title",
     )
     loras_state = gr.State(loras)
@@ -639,27 +635,27 @@ with gr.Blocks(css=css, delete_cache=(240, 240)) as app:
         with gr.Column(scale=1,min_width=50):
             randomize_button = gr.Button("🎲", variant="secondary", scale=1, elem_id="random_btn")
                 
-#    with gr.Row(elem_id="loaded_loras"):            
-#        with gr.Column(scale=8):
-#            with gr.Row():
-#                with gr.Column(scale=0, min_width=50):
-#                    lora_image_3 = gr.Image(label="LoRA 3 Image", interactive=False, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
-#                with gr.Column(scale=3, min_width=100):
-#                    selected_info_3 = gr.Markdown("Select a LoRA 3")
-#                with gr.Column(scale=5, min_width=50):
-#                    lora_scale_3 = gr.Slider(label="LoRA 3 Scale", minimum=0, maximum=3, step=0.05, value=0.5)
-#            with gr.Row():
-#                remove_button_3 = gr.Button("Remove", size="sm")
-#        with gr.Column(scale=8):
-#            with gr.Row():
-#                with gr.Column(scale=0, min_width=50):
-#                    lora_image_4 = gr.Image(label="LoRA 4 Image", interactive=False, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
-#                with gr.Column(scale=3, min_width=100):
-#                    selected_info_4 = gr.Markdown("Select a LoRA 4")
-#                with gr.Column(scale=5, min_width=150):
-#                    lora_scale_4 = gr.Slider(label="LoRA 4 Scale", minimum=0, maximum=3, step=0.05, value=0.5)
-#            with gr.Row():
-#                remove_button_4 = gr.Button("Remove", size="sm")
+    with gr.Row(elem_id="loaded_loras"):            
+        with gr.Column(scale=8):
+            with gr.Row():
+                with gr.Column(scale=0, min_width=50):
+                    lora_image_3 = gr.Image(label="LoRA 3 Image", interactive=False, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
+                with gr.Column(scale=3, min_width=100):
+                    selected_info_3 = gr.Markdown("Select a LoRA 3")
+                with gr.Column(scale=5, min_width=50):
+                    lora_scale_3 = gr.Slider(label="LoRA 3 Scale", minimum=0, maximum=3, step=0.05, value=0.5)
+            with gr.Row():
+                remove_button_3 = gr.Button("Remove", size="sm")
+        with gr.Column(scale=8):
+            with gr.Row():
+                with gr.Column(scale=0, min_width=50):
+                    lora_image_4 = gr.Image(label="LoRA 4 Image", interactive=False, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
+                with gr.Column(scale=3, min_width=100):
+                    selected_info_4 = gr.Markdown("Select a LoRA 4")
+                with gr.Column(scale=5, min_width=150):
+                    lora_scale_4 = gr.Slider(label="LoRA 4 Scale", minimum=0, maximum=3, step=0.05, value=0.5)
+            with gr.Row():
+                remove_button_4 = gr.Button("Remove", size="sm")
 
     with gr.Row():
         with gr.Accordion("Advanced Settings", open=True):
@@ -702,44 +698,53 @@ with gr.Blocks(css=css, delete_cache=(240, 240)) as app:
             )
         with gr.Column():
             progress_bar = gr.Markdown(elem_id="progress", visible=False)
-            result = gr.Image(label="Image", interactive=False, show_share_button=False)
-  
+            result = gr.Image(label="Generated Image", interactive=False, show_share_button=False)
 #            with gr.Accordion("History", open=False):
 #                history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)  
 
     gallery.select(
         update_selection,
         inputs=[selected_indices, loras_state, width, height],
-        outputs=[prompt, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, width, height, lora_image_1, lora_image_2])
+        outputs=[prompt, selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, width, height, lora_image_1, lora_image_2, lora_image_3, lora_image_4])
     remove_button_1.click(
         remove_lora_1,
         inputs=[selected_indices, loras_state],
-        outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
     )
     remove_button_2.click(
         remove_lora_2,
         inputs=[selected_indices, loras_state],
-        outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
+    )
+    remove_button_3.click(
+        remove_lora_3,
+        inputs=[selected_indices, loras_state],
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
+    )
+    remove_button_4.click(
+        remove_lora_4,
+        inputs=[selected_indices, loras_state],
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
     )
     randomize_button.click(
         randomize_loras,
         inputs=[selected_indices, loras_state],
-        outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2, prompt]
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4, prompt]
     )
     add_custom_lora_button.click(
         add_custom_lora,
         inputs=[custom_lora, selected_indices, loras_state, gallery],
-        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
     )
     remove_custom_lora_button.click(
         remove_custom_lora,
         inputs=[selected_indices, loras_state, gallery],
-        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
     )
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
-        inputs=[prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state],
+        inputs=[prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, randomize_seed, seed, width, height, loras_state],
         outputs=[result, seed, progress_bar]
     )#.then(
      #   fn=lambda x, history: update_history(x, history),