Update app.py

app.py

@@ -4,58 +4,83 @@ import gradio as gr
 models = ["Model A", "Model B", "Model C"]
 controlnet_types = ["Canny", "Depth", "Normal", "Pose"]
 
+# Fungsi placeholder
 def load_model(selected_model):
-    # Logika untuk memuat model berdasarkan pilihan
     return f"Model {selected_model} telah dimuat."
 
-def generate_image(prompt, model):
+def generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model):
     # Logika untuk menghasilkan gambar dari teks menggunakan model
-    return f"Gambar untuk prompt '{prompt}' dengan model '{model}'"
+    return [f"Gambar {i+1} untuk prompt '{prompt}' dengan model '{model}'" for i in range(num_images)]
 
-def process_image(image, controlnet_type):
+def process_image(image, prompt, neg_prompt, model):
+    # Logika untuk memproses gambar menggunakan model
+    return f"Proses gambar dengan prompt '{prompt}' dan model '{model}'"
+
+def controlnet_process(image, controlnet_type, model):
     # Logika untuk memproses gambar menggunakan ControlNet
-    return f"Proses gambar dengan tipe ControlNet '{controlnet_type}'"
+    return f"Proses gambar dengan ControlNet '{controlnet_type}' dan model '{model}'"
 
 with gr.Blocks() as app:
+    # Dropdown untuk memilih model di luar tab
+    with gr.Row():
+        model_dropdown = gr.Dropdown(choices=models, label="Pilih Model")
+
+    # Tab untuk Text-to-Image
     with gr.Tab("Text-to-Image"):
         with gr.Row():
-            model_dropdown = gr.Dropdown(choices=models, label="Pilih Model")
-            prompt_input = gr.Textbox(label="Prompt")
-            generate_button = gr.Button("Generate")
+            with gr.Column():
+                prompt_input = gr.Textbox(label="Prompt", placeholder="Masukkan prompt teks")
+                neg_prompt_input = gr.Textbox(label="Neg Prompt", placeholder="Masukkan negasi prompt")
+                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)
+                generate_button = gr.Button("Generate")
+            
+            with gr.Column():
+                output_gallery = gr.Gallery(label="Hasil Gambar").style(height=512)
         
-        output_image = gr.Image(label="Hasil Gambar")
-        
-        def update_image(prompt, model):
+        def update_images(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model):
             # Update fungsi sesuai kebutuhan
-            return generate_image(prompt, model)
+            return generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model)
         
-        generate_button.click(fn=update_image, inputs=[prompt_input, model_dropdown], outputs=output_image)
+        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)
 
+    # Tab untuk Image-to-Image
     with gr.Tab("Image-to-Image"):
         with gr.Row():
-            image_input = gr.Image(label="Unggah Gambar")
-            image_output = gr.Image(label="Hasil Gambar")
-            image_button = gr.Button("Proses Gambar")
+            with gr.Column():
+                image_input = gr.Image(label="Unggah Gambar")
+                prompt_input_i2i = gr.Textbox(label="Prompt", placeholder="Masukkan prompt teks")
+                neg_prompt_input_i2i = gr.Textbox(label="Neg Prompt", placeholder="Masukkan negasi prompt")
+                generate_button_i2i = gr.Button("Proses Gambar")
+            
+            with gr.Column():
+                output_image_i2i = gr.Image(label="Hasil Gambar")
         
-        def process_image_func(image):
+        def process_image_func(image, prompt, neg_prompt, model):
             # Update fungsi sesuai kebutuhan
-            return f"Proses gambar yang diunggah"
+            return process_image(image, prompt, neg_prompt, model)
         
-        image_button.click(fn=process_image_func, inputs=image_input, outputs=image_output)
+        generate_button_i2i.click(fn=process_image_func, inputs=[image_input, prompt_input_i2i, neg_prompt_input_i2i, model_dropdown], outputs=output_image_i2i)
 
+    # Tab untuk ControlNet
     with gr.Tab("ControlNet"):
         with gr.Row():
-            controlnet_dropdown = gr.Dropdown(choices=controlnet_types, label="Pilih Tipe ControlNet")
-            controlnet_image_input = gr.Image(label="Unggah Gambar untuk ControlNet")
-            controlnet_button = gr.Button("Proses dengan ControlNet")
-        
-        controlnet_output_image = gr.Image(label="Hasil ControlNet")
+            with gr.Column():
+                controlnet_dropdown = gr.Dropdown(choices=controlnet_types, label="Pilih Tipe ControlNet")
+                controlnet_image_input = gr.Image(label="Unggah Gambar untuk ControlNet")
+                controlnet_button = gr.Button("Proses dengan ControlNet")
+            
+            with gr.Column():
+                controlnet_output_image = gr.Image(label="Hasil ControlNet")
         
-        def controlnet_process_func(image, controlnet_type):
+        def controlnet_process_func(image, controlnet_type, model):
             # Update fungsi sesuai kebutuhan
-            return process_image(image, controlnet_type)
+            return controlnet_process(image, controlnet_type, model)
         
-        controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown], outputs=controlnet_output_image)
+        controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown, model_dropdown], outputs=controlnet_output_image)
 
 # Jalankan antarmuka
 app.launch()