Update app.py

app.py

@@ -138,9 +138,9 @@ def create_ui():
                         output = gr.Textbox(label="Generated Prompt", lines=5)
                         
                         with gr.Row():
-                            for char_name, char_data in data["characters"].items():
-                                if char_data["image"]:
-                                    gr.Image(value=char_data["image"], label=char_name, height=100, width=100)
+                            char_images = [char_data["image"] for char_data in data["characters"].values() if char_data["image"]]
+                            char_names = [char_name for char_name, char_data in data["characters"].items() if char_data["image"]]
+                            gr.Gallery(value=char_images, label="Character Images", show_label=True, elem_id="char_gallery", columns=2, rows=2, height="auto")
             
             with gr.TabItem("Character Creation"):
                 with gr.Row():
@@ -151,7 +151,7 @@ def create_ui():
                         create_char_button = gr.Button("Create/Update Character")
                     
                     with gr.Column():
-                        char_gallery = gr.Gallery(label="Existing Characters", show_label=False, elem_id="char_gallery").style(grid=2, height="auto")
+                        char_gallery = gr.Gallery(label="Existing Characters", show_label=True, elem_id="char_gallery", columns=2, rows=2, height="auto")
 
         def update_and_generate(*args):
             nonlocal data
@@ -178,7 +178,8 @@ def create_ui():
         )
 
         def update_char_gallery():
-            return gr.Gallery(value=[char_data["image"] for char_data in data["characters"].values() if char_data["image"]])
+            char_images = [char_data["image"] for char_data in data["characters"].values() if char_data["image"]]
+            return gr.Gallery(value=char_images)
 
         create_char_button.click(
             create_character,