Update app.py

app.py

@@ -145,26 +145,26 @@ def prompt_generator_app(data_manager):
             refresh_characters_button = gr.Button("Refresh Character List")
 
             # Create a placeholder for the character_select component
-            character_select = gr.Gallery(label="Select Characters").style(grid=[3], height='auto')
+            character_select = gr.Gallery(label="Select Characters", columns=3, height='auto')
 
             # Load and display characters
             def load_characters():
                 characters = data_manager.get_characters()
                 if not characters:
-                    return [], [], []
+                    return [], []
                 else:
                     # Build character images and labels
                     character_images = []
-                    character_indices = []
                     for idx, char in enumerate(characters):
                         image_path = char.get('image_path', '')
                         if image_path and os.path.exists(image_path):
                             image_url = image_path
                         else:
                             image_url = None  # Or a placeholder image
-                        character_images.append(image_url)
-                        character_indices.append(idx)
-                    return character_images, character_indices, characters
+                        # Adding the character's name and gender as a caption
+                        caption = f"{char['name']} ({char['gender']})"
+                        character_images.append((image_url, caption))
+                    return character_images, characters
 
             # Function to update selected characters
             def update_selected_characters(selected_indices):
@@ -175,14 +175,14 @@ def prompt_generator_app(data_manager):
 
             # Function to refresh characters
             def refresh_characters():
-                character_images, character_indices, characters = load_characters()
+                character_images, characters = load_characters()
                 # Update character_select with new images
                 return gr.update(value=character_images)
 
             # Load initial characters
-            character_images, character_indices, characters = load_characters()
+            character_images, characters = load_characters()
             if character_images:
-                character_select.update(value=character_images)
+                character_select.value = character_images
             else:
                 gr.Markdown("No characters available. Please add characters in the Character Creation tab.")