Update app.py

app.py

@@ -138,6 +138,9 @@ def prompt_generator_app(data_manager):
         with gr.Group():
             gr.Markdown("### Character Selection")
 
+            # Initialize selected_characters_state before the if statement
+            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.")
@@ -156,14 +159,11 @@ def prompt_generator_app(data_manager):
                     character_labels.append(f"{char['name']} ({char['gender']})")
                 character_select = gr.Gallery(label="Select Characters").style(grid=[3], height='auto')
 
-                selected_characters_state = gr.State([])
-
                 # 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]
-                    selected_char_names = [char['name'] for char in selected_chars]
-                    return selected_characters_state.update(value=selected_chars)
+                    return selected_chars
 
                 # Update the gallery
                 character_select.render(character_images)
@@ -191,7 +191,6 @@ def prompt_generator_app(data_manager):
                     prompt_tags.extend(selected_tags)
 
             # Handle Characters
-            # Since Gradio's State cannot be passed directly, we retrieve it via args
             selected_chars = args[arg_idx]
             random_chars = args[arg_idx + 1]
             num_random_chars = args[arg_idx + 2]
@@ -205,7 +204,7 @@ def prompt_generator_app(data_manager):
                 selected_chars = random.sample(characters, num)
             else:
                 # selected_chars is already set from the selected characters in the gallery
-                pass
+                selected_chars = selected_chars or []
 
             # Determine the number of boys and girls
             num_girls = sum(1 for char in selected_chars if char.get('gender') == 'Girl')
@@ -229,7 +228,7 @@ def prompt_generator_app(data_manager):
                 # Get traits for the character
                 traits = ', '.join(char['traits'])
                 # Create a description for each character
-                # For SDXL models, use the format "[char1] AND [char2]"
+                # For SDXL models, use the format "[char1 description] AND [char2 description]"
                 # Each character's description is enclosed in parentheses
                 character_description = f"({traits})"
                 character_descriptions.append(character_description)