Add a refresh button to the dropdowns

app.py

@@ -74,6 +74,9 @@ def download_civitai_model(model_id, lora_id="", progress=gr.Progress(track_tqdm
     except Exception as e:
         return f"Error downloading model or LoRA: {e}"
 
+def refresh_dropdowns():
+    return gr.update(choices=models_list), gr.update(choices=loras_list)
+
 def load_model(model, lora="", use_lora=False):
     try:
         print(f"\n\nLoading {model}...")
@@ -145,6 +148,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                 with gr.Group(elem_id="input_group"):
                     model_dropdown = gr.Dropdown(choices=models_list, value=models_list[0] if models_list else None, label="Model", elem_id="model_dropdown")
                     lora_dropdown = gr.Dropdown(choices=loras_list, value=loras_list[0], label="LoRA")
+                    refresh_btn = gr.Button("Refresh Dropdowns")
                     prompt = gr.Textbox(label="Prompt", elem_id="prompt_textbox")
                     generate_btn = gr.Button("Generate Image", elem_id="generate_button")
                 with gr.Accordion("Advanced", open=False, elem_id="advanced_accordion"):
@@ -157,6 +161,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             with gr.Column(elem_id="output_column"):
                 output_gallery = gr.Gallery(label="Generated Images", height=480, scale=1, elem_id="output_gallery")
 
+            refresh_btn.click(refresh_dropdowns, outputs=[model_dropdown, lora_dropdown])
             generate_btn.click(generate_images, inputs=[model_dropdown, lora_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale, height, width, num_images], outputs=output_gallery)
 
         with gr.Tab("Download Custom Model"):