Update app.py

app.py

@@ -324,12 +324,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
                     "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"
+                    "FLUX.1 [Schnell]"
                 ]
 
                 # 전체 모델 리스트
@@ -368,10 +363,10 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
                     "Collage Flux"
                 ]
 
-                model = gr.Radio(
-                    label="Select Model",
+                model = gr.Checkboxgroup(  # Radio를 Checkboxgroup으로 변경
+                    label="Select Models (Choose up to 4)",
                     choices=models_list,
-                    value="Stable Diffusion 3.5 Large",
+                    value=default_models,  # 기본값으로 상위 4개 선택
                     interactive=True
                 )
 
@@ -388,19 +383,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
             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=generate_grid,
@@ -421,7 +403,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
 
     def filter_models(search_term):
         filtered_models = [m for m in models_list if search_term.lower() in m.lower()]
-        return gr.update(choices=filtered_models)
+        return gr.update(choices=filtered_models, value=[])
 
     model_search.change(filter_models, inputs=model_search, outputs=model)