Update app.py

app.py

@@ -271,11 +271,12 @@ def generate_grid(prompt, selected_models, custom_lora, negative_prompt, steps,
     
     # 초기 이미지 배열 생성
     images = [None] * 4
+    total_models = len(selected_models[:4])
     
     # 각 모델별로 이미지 생성
     for idx, model_name in enumerate(selected_models[:4]):
         try:
-            progress(idx / len(selected_models), f"Generating image for {model_name}...")
+            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)
             images[idx] = img
         except Exception as e:
@@ -292,30 +293,6 @@ def generate_grid(prompt, selected_models, custom_lora, negative_prompt, steps,
     progress(1.0, "Generation complete!")
     return images
 
-    
-    def update_gallery():
-        return images
-    
-    # 각 모델별로 이미지 생성
-    for idx, model_name in enumerate(selected_models[:4]):
-        try:
-            img = query(prompt, model_name, custom_lora, negative_prompt, steps, cfg_scale, seed, strength, width, height)
-            images[idx] = img
-            # 갤러리 업데이트를 위해 중간 결과 반환
-            yield images
-        except Exception as e:
-            print(f"Error generating image for {model_name}: {str(e)}")
-            continue
-    
-    # 남은 슬롯을 마지막 생성된 이미지로 채움
-    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
-    
-    yield images
-
 css = """
 footer {
     visibility: hidden;