Upload 5 files

app.py

@@ -87,45 +87,54 @@ with gr.Blocks(fill_width=True, elem_id="container", css=css, delete_cache=(60,
                     vae_model = gr.Dropdown(label="VAE Model", choices=get_vaes(), value=get_vaes()[0])
 
                 with gr.Accordion("LoRA", open=True, visible=True):
+                    def lora_dropdown(label):
+                        return gr.Dropdown(label=label, choices=get_all_lora_tupled_list(), value="", allow_custom_value=True, elem_classes="lora", min_width=320)
+
+                    def lora_scale_slider(label):
+                        return gr.Slider(minimum=-2, maximum=2, step=0.01, value=1.00, label=label)
+                    
+                    def lora_textbox():
+                        return gr.Textbox(label="", info="Example of prompt:", value="", show_copy_button=True, interactive=False, visible=False)
+                    
                     with gr.Row():
                         with gr.Column():
                             with gr.Row():
-                                lora1 = gr.Dropdown(label="LoRA 1", choices=get_all_lora_tupled_list(), value="", allow_custom_value=True, elem_classes="lora", min_width=320)
-                                lora1_wt = gr.Slider(minimum=-2, maximum=2, step=0.01, value=1.00, label="LoRA 1: weight")
+                                lora1 = lora_dropdown("LoRA 1")
+                                lora1_wt = lora_scale_slider("LoRA 1: weight")
                             with gr.Row():
-                                lora1_info = gr.Textbox(label="", info="Example of prompt:", value="", show_copy_button=True, interactive=False, visible=False)
+                                lora1_info = lora_textbox()
                                 lora1_copy = gr.Button(value="Copy example to prompt", visible=False)
                             lora1_md = gr.Markdown(value="", visible=False)
                         with gr.Column():
                             with gr.Row():
-                                lora2 = gr.Dropdown(label="LoRA 2", choices=get_all_lora_tupled_list(), value="", allow_custom_value=True, elem_classes="lora", min_width=320)
-                                lora2_wt = gr.Slider(minimum=-2, maximum=2, step=0.01, value=1.00, label="LoRA 2: weight")
+                                lora2 = lora_dropdown("LoRA 2")
+                                lora2_wt = lora_scale_slider("LoRA 2: weight")
                             with gr.Row():
-                                lora2_info = gr.Textbox(label="", info="Example of prompt:", value="", show_copy_button=True, interactive=False, visible=False)
+                                lora2_info = lora_textbox()
                                 lora2_copy = gr.Button(value="Copy example to prompt", visible=False)
                             lora2_md = gr.Markdown(value="", visible=False)
                         with gr.Column():
                             with gr.Row():
-                                lora3 = gr.Dropdown(label="LoRA 3", choices=get_all_lora_tupled_list(), value="", allow_custom_value=True, elem_classes="lora", min_width=320)
-                                lora3_wt = gr.Slider(minimum=-2, maximum=2, step=0.01, value=1.00, label="LoRA 3: weight")
+                                lora3 = lora_dropdown("LoRA 3")
+                                lora3_wt = lora_scale_slider("LoRA 3: weight")
                             with gr.Row():
-                                lora3_info = gr.Textbox(label="", info="Example of prompt:", value="", show_copy_button=True, interactive=False, visible=False)
+                                lora3_info = lora_textbox()
                                 lora3_copy = gr.Button(value="Copy example to prompt", visible=False)
                             lora3_md = gr.Markdown(value="", visible=False)
                         with gr.Column():
                             with gr.Row():
-                                lora4 = gr.Dropdown(label="LoRA 4", choices=get_all_lora_tupled_list(), value="", allow_custom_value=True, elem_classes="lora", min_width=320)
-                                lora4_wt = gr.Slider(minimum=-2, maximum=2, step=0.01, value=1.00, label="LoRA 4: weight")
+                                lora4 = lora_dropdown("LoRA 4")
+                                lora4_wt = lora_scale_slider("LoRA 4: weight")
                             with gr.Row():
-                                lora4_info = gr.Textbox(label="", info="Example of prompt:", value="", show_copy_button=True, interactive=False, visible=False)
+                                lora4_info = lora_textbox()
                                 lora4_copy = gr.Button(value="Copy example to prompt", visible=False)
                             lora4_md = gr.Markdown(value="", visible=False)
                         with gr.Column():
                             with gr.Row():
-                                lora5 = gr.Dropdown(label="LoRA 5", choices=get_all_lora_tupled_list(), value="", allow_custom_value=True, elem_classes="lora", min_width=320)
-                                lora5_wt = gr.Slider(minimum=-2, maximum=2, step=0.01, value=1.00, label="LoRA 5: weight")
+                                lora5 = lora_dropdown("LoRA 5")
+                                lora5_wt = lora_scale_slider("LoRA 5: weight")
                             with gr.Row():
-                                lora5_info = gr.Textbox(label="", info="Example of prompt:", value="", show_copy_button=True, interactive=False, visible=False)
+                                lora5_info = lora_textbox()
                                 lora5_copy = gr.Button(value="Copy example to prompt", visible=False)
                             lora5_md = gr.Markdown(value="", visible=False)
                     with gr.Accordion("From URL", open=True, visible=True):
@@ -137,7 +146,7 @@ with gr.Blocks(fill_width=True, elem_id="container", css=css, delete_cache=(60,
                             lora_search_civitai_result = gr.Dropdown(label="Search Results", choices=[("", "")], value="", allow_custom_value=True, visible=False)
                             lora_search_civitai_json = gr.JSON(value={}, visible=False)
                             lora_search_civitai_desc = gr.Markdown(value="", visible=False)
-                        lora_download_url = gr.Textbox(label="URL", placeholder="http://...my_lora_url.safetensors", lines=1)
+                        lora_download_url = gr.Textbox(label="LoRA URL", placeholder="https://civitai.com/api/download/models/28907", lines=1)
                         lora_download = gr.Button("Get and set LoRA and apply to prompt")
 
                 with gr.Row():

requirements.txt

@@ -4,7 +4,7 @@ diffusers
 invisible_watermark
 transformers
 xformers
-git+https://github.com/R3gm/stablepy.git
+git+https://github.com/R3gm/stablepy.git@flux_beta
 torch==2.2.0
 gdown
 opencv-python