Update app.py

app.py

@@ -135,11 +135,11 @@ def prompt_generator_app(data_manager):
         with gr.Group():
             gr.Markdown("### Character Selection")
 
-            # Create placeholders for dynamic components
-            character_select = gr.State([])
-            selected_character_names = gr.State([])
+            # Placeholders for dynamic components
+            hidden_selected_characters = gr.Textbox(visible=False, elem_id='selected_character_names')
 
             characters_display = gr.HTML()
+
             refresh_characters_button = gr.Button("Refresh Character List")
 
         # Number of people in the scene
@@ -155,7 +155,7 @@ def prompt_generator_app(data_manager):
         def display_characters():
             characters = data_manager.get_characters()
             if not characters:
-                return "", []
+                return "<p>No characters available. Please add characters in the Character Creation tab.</p>"
 
             # Build HTML content for characters display
             html_content = "<div style='display: flex; flex-wrap: wrap;'>"
@@ -190,19 +190,18 @@ def prompt_generator_app(data_manager):
                         selected.push(checkboxes[i].value);
                     }
                 }
-                gradioApp().querySelector('#selected_character_names').value = JSON.stringify(selected);
+                document.getElementById('selected_character_names').value = JSON.stringify(selected);
             }
             </script>
             """
 
-            return html_content, []
+            return html_content
 
-        # Display characters on load and when refresh is clicked
-        characters_display.update(display_characters)
-        refresh_characters_button.click(display_characters, outputs=characters_display)
+        # Initialize characters_display with the HTML content
+        characters_display.value = display_characters()
 
-        # Hidden input to store selected character names
-        hidden_selected_characters = gr.Textbox(visible=False, elem_id='selected_character_names')
+        # When refresh_characters_button is clicked, update characters_display
+        refresh_characters_button.click(fn=display_characters, outputs=characters_display)
 
         def generate_prompt(*args):
             # args correspond to inputs in the order they are defined