Update app.py

app.py

@@ -223,14 +223,23 @@ for ex_human in human_list_path:
     ex_dict['composite'] = None
     human_ex_list.append(ex_dict)
 
+def get_examples_lst(folder_name, file_list):
+    example_list = []
+    for i in file_list:
+        example_list.append([os.path.join(example_path,folder_name,i),i.split(".")[0]])
+    return example_list
+    
 topwear_list = os.listdir(os.path.join(example_path,"topwear"))
-topwear_list_path = [os.path.join(example_path,"topwear",topwear) for topwear in topwear_list]
+#topwear_list_path = [os.path.join(example_path,"topwear",topwear) for topwear in topwear_list]
+topwear_ex_list = get_examples_lst("topwear", topwear_list)
 
 bottomwear_list = os.listdir(os.path.join(example_path,"bottomwear"))
-bottomwear_list_path = [os.path.join(example_path,"bottomwear",bottomwear) for bottomwear in bottomwear_list]
+# bottomwear_list_path = [os.path.join(example_path,"bottomwear",bottomwear) for bottomwear in bottomwear_list]
+bottomwear_ex_list = get_examples_lst("bottomwear", bottomwear_list)
 
 dress_list = os.listdir(os.path.join(example_path,"dresses"))
-dress_list_path = [os.path.join(example_path,"dresses",dress) for dress in dress_list]
+# dress_list_path = [os.path.join(example_path,"dresses",dress) for dress in dress_list]
+dress_ex_list = get_examples_lst("dresses", dress_list)
 
 
 image_blocks = gr.Blocks(theme="Nymbo/Alyx_Theme").queue()
@@ -250,22 +259,19 @@ with image_blocks as demo:
         with gr.Column(scale=1):
             with gr.Row():
                 topwear_image = gr.Image(label="Topwear", sources='upload', type="pil")
-                with gr.Row(elem_id="topwear-prompt-container"):
-                    with gr.Row():
-                        topwear_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
-                example = gr.Examples(inputs=topwear_image, examples_per_page=4,examples=topwear_list_path)
+                with gr.Row():
+                    topwear_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
+                example = gr.Examples(inputs=[topwear_image,topwear_desc], examples_per_page=4,examples=topwear_ex_list)
             with gr.Row():
                 bottomwear_image = gr.Image(label="Bottomwear", sources='upload', type="pil")
-                with gr.Row(elem_id="bottomwear-prompt-container"):
-                    with gr.Row():
-                        bottomwear_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
-                example = gr.Examples(inputs=bottomwear_image, examples_per_page=4, examples=bottomwear_list_path)
+                with gr.Row():
+                    bottomwear_desc = gr.Textbox(placeholder="Description of garment ex) Cargo pants", show_label=False, elem_id="prompt")
+                example = gr.Examples(inputs=[bottomwear_image,bottomwear_desc], examples_per_page=4, examples=bottomwear_ex_list)
             with gr.Row():
                 dress_image = gr.Image(label="Dress", sources='upload', type="pil")
-                with gr.Row(elem_id="dress-prompt-container"):
-                    with gr.Row():
-                        dress_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
-                example = gr.Examples(inputs=dress_image, examples_per_page=4, examples=dress_list_path)
+                with gr.Row():
+                    dress_desc = gr.Textbox(placeholder="Description of garment ex) Jumper suit", show_label=False, elem_id="prompt")
+                example = gr.Examples(inputs=[dress_image,dress_desc], examples_per_page=4, examples=dress_ex_list)
 
     with gr.Row():
         with gr.Column():