Fix model image list to grid type

app.py

@@ -197,8 +197,8 @@ def load_image_from_url(url):
         return None
     
 
-#URL 입력 처리 함수
 def process_url_image(url):
+    """Process image from URL and return PIL Image"""
     if not url or not url.strip():
         return None
     
@@ -211,14 +211,25 @@ def process_url_image(url):
     except Exception as e:
         print(f"Error parsing URL: {e}")
         return None
-    
+
     img = load_image_from_url(url)
     if img is None:
         print("Failed to load image from URL")
         return None
-        
+    
     return preprocess_image(img)
 
+def load_example_for_editor(image_path):
+    """Load example image for ImageEditor component"""
+    if image_path is None:
+        return None
+    
+    # ImageEditor는 특정 형식을 기대하므로 딕셔너리 형태로 반환
+    return {
+        "background": image_path,
+        "layers": None,
+        "composite": None
+    }
 
 def download_model_file(model_path, urls):
     """Download model file from multiple URLs if it doesn't exist"""
@@ -453,13 +464,8 @@ garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
 human_list = os.listdir(os.path.join(example_path,"human"))
 human_list_path = [os.path.join(example_path,"human",human) for human in human_list]
 
-human_ex_list = []
-for ex_human in human_list_path:
-    ex_dict= {}
-    ex_dict['background'] = ex_human
-    ex_dict['layers'] = None
-    ex_dict['composite'] = None
-    human_ex_list.append(ex_dict)
+# human_ex_list를 단순한 이미지 경로 리스트로 변경 (그리드 표시를 위해)
+human_ex_list = human_list_path
 
 ##default human
 
@@ -479,7 +485,8 @@ with image_blocks as demo:
             example = gr.Examples(
                 inputs=imgs,
                 examples_per_page=8,
-                examples=human_ex_list
+                examples=human_ex_list,
+                fn=load_example_for_editor
             )
 
         with gr.Column():