Update app.py

app.py

@@ -8,7 +8,7 @@ controlnet_types = ["Canny", "Depth", "Normal", "Pose"]
 def load_model(selected_model):
     return f"Model {selected_model} telah dimuat."
 
-def generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model):
+def generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, cfg_scale, seed, model):
     # Logika untuk menghasilkan gambar dari teks menggunakan model
     return [f"Gambar {i+1} untuk prompt '{prompt}' dengan model '{model}'" for i in range(num_images)], {"prompt": prompt, "neg_prompt": neg_prompt}
 
@@ -27,36 +27,36 @@ with gr.Blocks() as app:
     
     # Tab untuk Text-to-Image
     with gr.Tab("Text-to-Image"):
+        # Prompt dan Neg Prompt
         with gr.Row():
-            with gr.Column(scale=80):
-                # Prompt dan Neg Prompt
-                prompt_input = gr.Textbox(label="Prompt", placeholder="Masukkan prompt teks", lines=2, elem_id="prompt-input")
-                neg_prompt_input = gr.Textbox(label="Neg Prompt", placeholder="Masukkan negasi prompt", lines=2, elem_id="neg-prompt-input")
-            
-            with gr.Column(scale=20):
-                # Tombol Generate
-                generate_button = gr.Button("Generate", elem_id="generate-button")
-        
+            prompt_input = gr.Textbox(label="Prompt", placeholder="Masukkan prompt teks", lines=2, elem_id="prompt-input")
+            generate_button = gr.Button("Generate", elem_id="generate-button")
+
         with gr.Row():
-            with gr.Column():
+            neg_prompt_input = gr.Textbox(label="Neg Prompt", placeholder="Masukkan negasi prompt", lines=2, elem_id="neg-prompt-input")
+
+        with gr.Row():
+            with gr.Column(scale=2):
                 # Konfigurasi
-                width_input = gr.Slider(minimum=64, maximum=2048, step=64, label="Lebar", value=512)
-                height_input = gr.Slider(minimum=64, maximum=2048, step=64, label="Tinggi", value=512)
-                scheduler_input = gr.Dropdown(choices=["Euler", "LMS", "DDIM"], label="Scheduler")
-                num_steps_input = gr.Slider(minimum=1, maximum=100, step=1, label="Num Steps", value=20)
-                num_images_input = gr.Slider(minimum=1, maximum=10, step=1, label="Num Images", value=1)
+                scheduler_input = gr.Dropdown(choices=["Euler", "LMS", "DDIM"], label="Sampling method")
+                num_steps_input = gr.Slider(minimum=1, maximum=100, step=1, label="Sampling steps", value=20)
+                width_input = gr.Slider(minimum=64, maximum=2048, step=64, label="Width", value=512)
+                height_input = gr.Slider(minimum=64, maximum=2048, step=64, label="Height", value=512)
+                cfg_scale_input = gr.Slider(minimum=1, maximum=20, step=1, label="CFG Scale", value=7)
+                seed_input = gr.Number(label="Seed", value=-1)
+                num_images_input = gr.Slider(minimum=1, maximum=10, step=1, label="Batch size", value=1)
             
-            with gr.Column():
+            with gr.Column(scale=1):
                 # Gallery untuk output gambar
                 output_gallery = gr.Gallery(label="Hasil Gambar")
                 # Output teks JSON di bawah gallery
                 output_text = gr.Textbox(label="Output JSON", placeholder="Hasil dalam format JSON", lines=2)
 
-        def update_images(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model):
+        def update_images(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, cfg_scale, seed, model):
             # Update fungsi sesuai kebutuhan
-            return generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model)
+            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, num_images_input, model_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, num_images_input, cfg_scale_input, seed_input, model_dropdown], outputs=[output_gallery, output_text])
 
     # Tab untuk Image-to-Image
     with gr.Tab("Image-to-Image"):