Update app.py

app.py

@@ -2,6 +2,13 @@ import gradio as gr
 import os
 import modules.wdtagger
 
+from modules.model import get_model_and_vae_options
+from modules.text2img import generate_image_wrapper
+
+# Mendapatkan daftar model dan VAE untuk dropdown
+all_models, all_vaes = get_model_and_vae_options()
+
+
 # Daftar model dan ControlNet
 models = ["Model A", "Model B", "Model C"]
 vae = ["VAE A", "VAE B", "VAE C"]
@@ -43,7 +50,7 @@ def intpaint_func (image, controlnet_type, model):
 with gr.Blocks(css= "style.css") as app:
     # Dropdown untuk memilih model di luar tab dengan lebar kecil
     with gr.Row():
-        model_dropdown = gr.Dropdown(choices=models, label="Model", value="Model B")
+        checkpoint_dropdown = gr.Dropdown(choices=models, label="Model", value="Model B")
         vae_dropdown = gr.Dropdown(choices=vae, label="VAE", value="VAE C")
     
     # Tab untuk Text-to-Image
@@ -92,7 +99,7 @@ with gr.Blocks(css= "style.css") as app:
             # Update fungsi sesuai kebutuhan
             return generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, cfg_scale, seed, model)
         
-        generate_button.click(fn=update_images, inputs=[prompt_input, neg_prompt_input, width_input, height_input, scheduler_input, num_steps_input, batch_size, batch_count, cfg_scale_input, seed_input, model_dropdown, vae_dropdown], outputs=[output_gallery, output_text])
+        generate_button.click(fn=update_images, inputs=[prompt_input, neg_prompt_input, width_input, height_input, scheduler_input, num_steps_input, batch_size, batch_count, cfg_scale_input, seed_input, checkpoint_dropdown, vae_dropdown], outputs=[output_gallery, output_text])
 
     # Tab untuk Image-to-Image
     with gr.Tab("Image-to-Image"):
@@ -124,7 +131,7 @@ with gr.Blocks(css= "style.css") as app:
             # Update fungsi sesuai kebutuhan
             return process_image(image, prompt, neg_prompt, model)
         
-        generate_button_i2i.click(fn=process_image_func, inputs=[image_input, prompt_input_i2i, neg_prompt_input_i2i, model_dropdown, vae_dropdown], outputs=output_image_i2i)
+        generate_button_i2i.click(fn=process_image_func, inputs=[image_input, prompt_input_i2i, neg_prompt_input_i2i, checkpoint_dropdown, vae_dropdown], outputs=output_image_i2i)
 
     # Tab untuk ControlNet
     with gr.Tab("ControlNet"):
@@ -136,7 +143,7 @@ with gr.Blocks(css= "style.css") as app:
             
             with gr.Column():
                 controlnet_output_image = gr.Image(label="Hasil ControlNet")
-        controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown, model_dropdown, vae_dropdown], outputs=controlnet_output_image)
+        controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown, checkpoint_dropdown, vae_dropdown], outputs=controlnet_output_image)
 
     # Tab untuk Intpainting
     with gr.Tab ("Inpainting"):