Update app.py

app.py

@@ -138,37 +138,59 @@ def prompt_generator_app(data_manager):
         with gr.Group():
             gr.Markdown("### Character Selection")
 
-            # Initialize selected_characters_state before the if statement
+            # Initialize selected_characters_state
             selected_characters_state = gr.State([])
 
-            characters = data_manager.get_characters()
-            if not characters:
-                gr.Markdown("No characters available. Please add characters in the Character Creation tab.")
+            # Add the Refresh Character List button
+            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')
+
+            # Load and display characters
+            def load_characters():
+                characters = data_manager.get_characters()
+                if not characters:
+                    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
+
+            # Function to update selected characters
+            def update_selected_characters(selected_indices):
+                characters = data_manager.get_characters()
+                selected_indices = selected_indices or []
+                selected_chars = [characters[i] for i in selected_indices]
+                return selected_chars
+
+            # Function to refresh characters
+            def refresh_characters():
+                character_images, character_indices, 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()
+            if character_images:
+                character_select.update(value=character_images)
             else:
-                # Display characters in a gallery
-                character_images = []
-                character_labels = []
-                for idx, char in enumerate(characters):
-                    image_path = char.get('image_path', '')
-                    image_url = ''
-                    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_labels.append(f"{char['name']} ({char['gender']})")
-                character_select = gr.Gallery(label="Select Characters").style(grid=[3], height='auto')
-
-                # Function to update selected characters
-                def update_selected_characters(selected_indices):
-                    selected_indices = selected_indices or []
-                    selected_chars = [characters[i] for i in selected_indices]
-                    return selected_chars
-
-                # Update the gallery
-                character_select.render(character_images)
-
-                character_select.select(fn=update_selected_characters, inputs=[character_select], outputs=[selected_characters_state])
+                gr.Markdown("No characters available. Please add characters in the Character Creation tab.")
+
+            # Update selected characters when selection changes
+            character_select.select(fn=update_selected_characters, inputs=None, outputs=selected_characters_state)
+
+            # Update character gallery when refresh button is clicked
+            refresh_characters_button.click(fn=refresh_characters, inputs=None, outputs=character_select)
 
         random_characters = gr.Checkbox(label="Select Random Characters")
         num_characters = gr.Slider(minimum=0, maximum=10, step=1, value=1, label="Number of Characters (if random)")