Update app.py

app.py

@@ -10,7 +10,16 @@ def load_model(selected_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 {i+1} untuk prompt '{prompt}' dengan model '{model}'" for i in range(num_images)]
+    return [f"Gambar {i+1} untuk prompt '{prompt}' dengan model '{model}'" for i in range(num_images)], {
+        "prompt": prompt,
+        "neg_prompt": neg_prompt,
+        "width": width,
+        "height": height,
+        "scheduler": scheduler,
+        "num_steps": num_steps,
+        "num_images": num_images,
+        "model": model
+    }
 
 def process_image(image, prompt, neg_prompt, model):
     # Logika untuk memproses gambar menggunakan model
@@ -23,37 +32,43 @@ def controlnet_process(image, controlnet_type, 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")
+        model_dropdown = gr.Dropdown(choices=models, label="Pilih Model", elem_id="model-dropdown").style(width=200)
 
     # Tab untuk Text-to-Image
     with gr.Tab("Text-to-Image"):
         with gr.Row():
             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")
+                prompt_input = gr.Textbox(label="Prompt", placeholder="Masukkan prompt teks", lines=2)
+                neg_prompt_input = gr.Textbox(label="Neg Prompt", placeholder="Masukkan negasi prompt", lines=2)
+                generate_button = gr.Button("Generate")
+            
+            with gr.Column():
+                # Output gallery and text
+                output_gallery = gr.Gallery(label="Hasil Gambar", elem_id="output-gallery")
+                output_text = gr.Textbox(label="Output JSON", elem_id="output-text", lines=10)
+        
+        with gr.Row():
+            with gr.Column():
                 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")
         
         def update_images(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model):
             # Update fungsi sesuai kebutuhan
-            return generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model)
+            images, json_output = generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, model)
+            return images, json_output
         
-        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)
+        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])
 
     # Tab untuk Image-to-Image
     with gr.Tab("Image-to-Image"):
         with gr.Row():
             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")
+                prompt_input_i2i = gr.Textbox(label="Prompt", placeholder="Masukkan prompt teks", lines=2)
+                neg_prompt_input_i2i = gr.Textbox(label="Neg Prompt", placeholder="Masukkan negasi prompt", lines=2)
                 generate_button_i2i = gr.Button("Proses Gambar")
             
             with gr.Column():