Update app.py

app.py

@@ -293,45 +293,36 @@ def generate_grid(prompt, selected_models, custom_lora, negative_prompt, steps,
         raise gr.Error("Please select at least 1 model")
     
     # 초기 이미지 배열 생성
-    images = []
+    images = [None] * 4
     total_models = len(selected_models[:4])
     
+    def update_gallery():
+        # None이 아닌 이미지만 포함하여 갤러리 업데이트
+        return [img for img in images if img is not None]
+    
     # 각 모델별로 이미지 생성
     for idx, model_name in enumerate(selected_models[:4]):
         try:
             progress((idx + 1) / total_models, f"Generating image for {model_name}...")
             img = query(prompt, model_name, custom_lora, negative_prompt, steps, cfg_scale, seed, strength, width, height)
-            if img:
-                images.append(img)
+            images[idx] = img
+            # 이미지가 생성될 때마다 갤러리 업데이트
+            yield update_gallery()
         except Exception as e:
             print(f"Error generating image for {model_name}: {str(e)}")
             continue
     
-    # 최소한 하나의 이미지가 생성되었는지 확인
-    if not images:
-        raise gr.Error("Failed to generate any images. Please try again.")
-    
-    # 4개의 이미지 슬롯을 채움
-    while len(images) < 4:
-        images.append(images[-1])
-    
-    # 이미지가 올바르게 로드되었는지 확인
-    valid_images = []
-    for img in images:
-        try:
-            # 이미지 복사본 생성
-            img_copy = img.copy()
-            valid_images.append(img_copy)
-        except Exception as e:
-            print(f"Error processing image: {str(e)}")
-            # 오류가 발생한 경우 마지막 유효한 이미지로 대체
-            if valid_images:
-                valid_images.append(valid_images[-1].copy())
-            else:
-                raise gr.Error("Failed to process images. Please try again.")
+    # 남은 슬롯을 마지막 생성된 이미지로 채움
+    last_valid_image = next((img for img in reversed(images) if img is not None), None)
+    if last_valid_image:
+        for i in range(len(images)):
+            if images[i] is None:
+                images[i] = last_valid_image
     
     progress(1.0, "Generation complete!")
-    return valid_images
+    yield update_gallery()
+
+
 
 css = """
 footer {
@@ -437,6 +428,8 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
     with gr.Row():
         generate_btn = gr.Button("Generate 2x2 Grid", variant="primary", size="lg")
         
+
+
     with gr.Row():
         gallery = gr.Gallery(
             label="Generated Images",
@@ -444,9 +437,11 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
             elem_id="gallery",
             columns=2,
             rows=2,
-            height="auto"
+            height="auto",
+            preview=True,  # 실시간 프리뷰 활성화
         )
 
+# 이벤트 핸들러 수정
     generate_btn.click(
         fn=generate_grid,
         inputs=[
@@ -461,10 +456,9 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
             width,
             height
         ],
-        outputs=gallery
+        outputs=gallery,
+        show_progress=True
     )
-
-
     
     def filter_models(search_term):
         filtered_models = [m for m in models_list if search_term.lower() in m.lower()]