Update app.py

app.py

@@ -312,14 +312,27 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
                 strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.1)
                 seed = gr.Slider(label="Seed (-1 for random)", value=-1, minimum=-1, maximum=1000000000, step=1)
 
-            with gr.Group():
-                gr.Markdown("### Model Selection")
+            with gr.Accordion("Model Selection", open=False):
                 model_search = gr.Textbox(
                     label="Search Models",
                     placeholder="Type to filter models...",
                     lines=1
                 )
                 
+                # 상위 9개 모델을 기본으로 설정
+                default_models = [
+                    "Stable Diffusion 3.5 Large",
+                    "Stable Diffusion 3.5 Large Turbo",
+                    "Stable Diffusion XL",
+                    "FLUX.1 [Schnell]",
+                    "FLUX.1 [Dev]",
+                    "Midjourney",
+                    "DreamPhotoGASM",
+                    "Disney",
+                    "Leonardo AI Style Illustration"
+                ]
+
+                # 전체 모델 리스트
                 models_list = [
                     "Stable Diffusion 3.5 Large",
                     "Stable Diffusion 3.5 Large Turbo",
@@ -363,17 +376,34 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
                 )
 
     with gr.Row():
-        generate_btn = gr.Button("Generate Image", variant="primary", size="lg")
+        generate_btn = gr.Button("Generate 2x2 Grid", variant="primary", size="lg")
         
     with gr.Row():
-        image_output = gr.Image(
-            type="pil",
-            label="Generated Image",
-            show_label=True
+        gallery = gr.Gallery(
+            label="Generated Images",
+            show_label=True,
+            elem_id="gallery",
+            columns=2,
+            rows=2,
+            height="auto"
         )
 
+    # 2x2 그리드 생성 함수
+    def generate_grid(prompt, model, custom_lora, negative_prompt, steps, cfg_scale, seed, strength, width, height):
+        images = []
+        for i in range(4):
+            current_seed = random.randint(1, 1000000000) if seed == -1 else seed + i
+            try:
+                img = query(prompt, model, custom_lora, negative_prompt, steps, cfg_scale, current_seed, strength, width, height)
+                images.append(img)
+            except Exception as e:
+                print(f"Error generating image {i+1}: {str(e)}")
+                continue
+        return images
+
+    # 이벤트 핸들러
     generate_btn.click(
-        fn=query,
+        fn=generate_grid,
         inputs=[
             text_prompt,
             model,
@@ -386,7 +416,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
             width,
             height
         ],
-        outputs=image_output
+        outputs=gallery
     )
 
     def filter_models(search_term):
@@ -396,4 +426,4 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
     model_search.change(filter_models, inputs=model_search, outputs=model)
 
 if __name__ == "__main__":
-    dalle.launch(show_api=False, share=False)            
+    dalle.launch(show_api=False, share=False)