Update UI

app.py

@@ -79,10 +79,11 @@ def infer(
 css = """
 #col-container {
     margin: 0 auto;
-    max-width: 640px;
+    max-width: 960px;
 }
 """
 
+
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # ID-Booth Demo")
@@ -91,66 +92,22 @@ with gr.Blocks(css=css) as demo:
             gr.Markdown("### Choose an Identity:")
 
         with gr.Row():
-            #gr.Markdown("### Choose an Identity:")
-            # identity_selectors = []
-
-            def make_identity_selector(identity):
-                image_path = os.path.join(folder_of_identity_images, identity + ".jpg")
-                img = gr.Image(value=image_path, show_label=False, interactive=False, width=128, height=128)
-                btn = gr.Button(value=f"Select {identity}")
-                btn.click(
-                    fn=lambda: (identity, identity),
-                    inputs=[],
-                    outputs=[selected_identity],
-                )
-
-            with gr.Column():
-                make_identity_selector(id_list[0])
-
-            with gr.Column():
-                make_identity_selector(id_list[1])
-
-            with gr.Column():
-                make_identity_selector( id_list[2])
-
-            with gr.Column():
-                make_identity_selector(id_list[3])
-
-            # with gr.Column():
-            #     identity = id_list[0]
-            #     image_path = os.path.join(folder_of_identity_images, identity + ".jpg")
-            #     make_identity_selector(identity, image_path)
-            # for identity in id_list:
-            #     image_path = os.path.join(folder_of_identity_images, identity + ".jpg")
-            #     make_identity_selector(identity, image_path)
-                
-            # for id in id_list:
-            #     btn = gr.Image(
-            #         value=os.path.join(folder_of_identity_images, id + ".jpg"),
-            #         label=id,
-            #         interactive=True,
-            #         height=128,
-            #         width=128,
-            #     )
-            #     identity_selectors.append(btn)
-
-        # Set up click handlers
-        # def select_identity(id):
-        #     return id
-
-        # for btn, identity in zip(identity_selectors, folder_of_identity_images):
-        #     btn.select(
-        #         select_identity,
-        #         inputs=[],
-        #         outputs=[selected_identity],
-        #     )
+            for id in id_list:
+                image_path = os.path.join(folder_of_identity_images, id + ".jpg")
+                img = gr.Image(value=image_path, label=id, 
+                               width=100, height=164,
+                               show_label=True, interactive=False,
+                               show_download_button=False, 
+                               show_fullscreen_button=False,
+                               )
+                   
             
         with gr.Row():
-            # which_id = gr.Dropdown(
-            #     label="Identity",
-            #     choices=id_list,
-            #     value=id_list[0],
-            # )
+            identity = gr.Dropdown(
+                label="Identity",
+                choices=id_list,
+                value=id_list[0],
+            )
 
             background = gr.Dropdown(
                 label="Background",
@@ -165,8 +122,9 @@ with gr.Blocks(css=css) as demo:
             )
 
         run_button = gr.Button("Run", scale=0, variant="primary")
+        # result = gr.Image(label="Result", show_label=False)
+        gallery_result = gr.Gallery(label="Generated Images", show_label=False)
 
-        result = gr.Image(label="Result", show_label=False)
 
         with gr.Accordion(open=False, label="Advanced Options"):
             negative_prompt = gr.Text(
@@ -206,18 +164,18 @@ with gr.Blocks(css=css) as demo:
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
 
-        gr.Examples(
-            examples=[
-                [id_list[0], backgrounds_list[0], poses_list[0], "A beautiful photo of a person", 0, False, 512, 512, 7.5, 50],
-            ],
-            inputs=[selected_identity, background, pose],
-        )
+        # gr.Examples(
+        #     examples=[
+        #         [id_list[0], backgrounds_list[0], poses_list[0], "A beautiful photo of a person", 0, False, 512, 512, 7.5, 50],
+        #     ],
+        #     inputs=[selected_identity, background, pose],
+        # )
 
     gr.on(
         triggers=[run_button.click],
         fn=infer,
         inputs=[
-            selected_identity,
+            identity,
             background,
             pose,
             negative_prompt,
@@ -227,7 +185,7 @@ with gr.Blocks(css=css) as demo:
             num_inference_steps,
             num_images
         ],
-        outputs=[result, seed],
+        outputs=[gallery_result],
     )
 
 if __name__ == "__main__":