Update app.py

app.py

@@ -162,7 +162,16 @@ def submit_function(
     # Xử lý mặt nạ (mask)
     person_image, 
     mask = person_image["background"],      # Lấy ảnh người từ lớp nền.
-    person_image["layers"][0]               # Lấy mặt nạ do người dùng vẽ (nếu có).
+    #person_image["layers"][0]               # Lấy mặt nạ do người dùng vẽ (nếu có).
+    if len(person_image["layers"]) > 0:
+        # Nếu danh sách không rỗng, lấy phần tử đầu tiên
+        layer = person_image["layers"][0]
+    else:
+        # Nếu danh sách rỗng, thực hiện hành động thay thế hoặc thông báo lỗi
+        layer = None
+    print("Không có layers trong person_image.")
+
+    
     mask = Image.open(mask).convert("L")    # Chuyển mặt nạ thành ảnh thang độ xám
     if len(np.unique(np.array(mask))) == 1: # Nếu mặt nạ chỉ chứa một giá trị (ví dụ: toàn đen hoặc toàn trắng), thì không sử dụng mặt nạ (mask = None).
         mask = None
@@ -251,9 +260,10 @@ HEADER = ""
 def app_gradio():
     with gr.Blocks(title="CatVTON") as demo:
         gr.Markdown(HEADER)
-        with gr.Tab("Mask-based & SD1.5"):
+        with gr.Tab("Mask-based"):
             with gr.Row():
                 with gr.Column(scale=1, min_width=350):
+                    # Ảnh model (người)
                     with gr.Row():
                         image_path = gr.Image(
                             type="filepath",
@@ -264,6 +274,7 @@ def app_gradio():
                             interactive=True, label="Person Image", type="filepath"
                         )
 
+                    # Ảnh quần áo
                     with gr.Row():
                         with gr.Column(scale=1, min_width=230):
                             cloth_image = gr.Image(
@@ -279,12 +290,13 @@ def app_gradio():
                                 value="upper",
                             )
 
-
+                    # Submit button - Run
                     submit = gr.Button("Submit")
                     gr.Markdown(
                         '<center><span style="color: #FF0000">!!! Click only Once, Wait for Delay !!!</span></center>'
                     )
                     
+                    # Advance setting
                     gr.Markdown(
                         '<span style="color: #808080; font-size: small;">Advanced options can adjust details:<br>1. `Inference Step` may enhance details;<br>2. `CFG` is highly correlated with saturation;<br>3. `Random seed` may improve pseudo-shadow.</span>'
                     )
@@ -306,7 +318,9 @@ def app_gradio():
                             value="input & mask & result",
                         )
 
+               
                 with gr.Column(scale=2, min_width=500):
+                    # Result image
                     result_image = gr.Image(interactive=False, label="Result")
                     with gr.Row():
                         # Photo Examples
@@ -369,6 +383,7 @@ def app_gradio():
                     person_example_fn, inputs=image_path, outputs=person_image
                 )
 
+                # Function khi ấn nút submit
                 submit.click(
                     submit_function,
                     [
@@ -382,8 +397,9 @@ def app_gradio():
                     ],
                     result_image,
                 )
-
+        
     demo.queue().launch(share=True, show_error=True)
+    #demo.queue().launch()
 
 
 if __name__ == "__main__":