Update app.py

app.py

@@ -21,7 +21,7 @@ categories = [
 ]
 
 class DataManager:
-    def __init__(self, base_dir='/data'):
+    def __init__(self, base_dir='data'):
         self.base_dir = base_dir
         # Ensure the base directory exists
         if not os.path.exists(self.base_dir):
@@ -235,11 +235,11 @@ def prompt_generator_app(data_manager):
 
             def refresh_character_selection():
                 new_options = get_character_options()
-                return gr.CheckboxGroup.update(choices=new_options)
+                return [gr.CheckboxGroup.update(choices=new_options)]
 
             refresh_characters_button.click(
                 refresh_character_selection,
-                outputs=character_select
+                outputs=[character_select]
             )
 
         random_characters = gr.Checkbox(label="Select Random Characters")
@@ -347,11 +347,14 @@ def prompt_generator_app(data_manager):
         # Add character_select directly to inputs
         inputs_list.extend([character_select, random_characters, num_characters])
 
-        generate_button.click(generate_prompt, inputs=inputs_list, outputs=prompt_output)
+        generate_button.click(
+            generate_prompt,
+            inputs=inputs_list,
+            outputs=prompt_output
+        )
 
         # Function to refresh tags display and sliders
         def refresh_tags():
-            updates = []
             for category_name, var_name in categories:
                 # Reload tags from data_manager
                 tags_list = data_manager.get_category_tags(var_name)
@@ -363,15 +366,16 @@ def prompt_generator_app(data_manager):
                     slider_value = min(1, max_tags)
                 # Update the tag display and slider
                 tag_display, tag_num = tag_displays[var_name]
-                updates.append(tag_display.update(value=f"**Tags:** {tags_string}"))
-                updates.append(tag_num.update(maximum=max_tags, value=slider_value))
-            return updates
-
-        # Prepare the outputs list
-        outputs = [component for pair in tag_displays.values() for component in pair]
+                tag_display.update(value=f"**Tags:** {tags_string}")
+                tag_num.update(maximum=max_tags, value=slider_value)
+            return None  # No outputs to return
 
         # Connect the refresh_tags function to the refresh_tags_button
-        refresh_tags_button.click(refresh_tags, outputs=outputs)
+        refresh_tags_button.click(
+            refresh_tags,
+            inputs=None,
+            outputs=None
+        )
 
 def character_creation_app(data_manager):
     with gr.Tab("Character Creation"):
@@ -397,10 +401,12 @@ def character_creation_app(data_manager):
         gr.Markdown("## Existing Characters")
 
         # Dropdown to select a character for editing or deleting
-        characters = data_manager.get_characters()
-        character_names = [char['name'] for char in characters]
+        def get_character_names():
+            characters = data_manager.get_characters()
+            return [char['name'] for char in characters]
+
         selected_character = gr.Dropdown(
-            choices=character_names,
+            choices=get_character_names(),
             label="Select a Character to Edit/Delete",
             interactive=True
         )
@@ -411,7 +417,7 @@ def character_creation_app(data_manager):
         action_output = gr.Textbox(label="Action Status", interactive=False)
 
         # **Section 3: Edit Character**
-        edit_section = gr.Column(visible=False)
+        edit_section = gr.Row(visible=False)
         with edit_section:
             with gr.Row():
                 edit_name = gr.Textbox(
@@ -465,9 +471,11 @@ def character_creation_app(data_manager):
             # Fetch character details
             character = next((char for char in data_manager.get_characters() if char['name'] == selected_name), None)
             if character:
-                edit_name_value = character['name']
-                edit_traits_value = ', '.join(character['traits'])
-                edit_gender_value = character['gender']
+                # Pre-fill the edit fields
+                edit_name.value = character['name']
+                edit_traits.value = ', '.join(character['traits'])
+                edit_gender.value = character['gender']
+                edit_image.value = None  # Reset image
                 return "", gr.update(visible=True)
             else:
                 return f"Character '{selected_name}' not found.", gr.update(visible=False)
@@ -492,10 +500,11 @@ def character_creation_app(data_manager):
                 'traits': new_traits,
                 'gender': new_gender,
                 'image': new_image_path if new_image_path else \
-                         next((char['image'] for char in data_manager.get_characters() if char['name'] == original_name), None)
+                         next((char for char in data_manager.get_characters() if char['name'] == original_name), {}).get('image')
             }
             success = data_manager.update_character(original_name, updated_char)
             if success:
+                # Update the dropdown choices
                 return f"Character '{new_name}' updated successfully.", gr.update(visible=False)
             else:
                 return "Failed to update character.", gr.update(visible=True)
@@ -591,7 +600,7 @@ def tags_app(data_manager):
         )
 
 def main():
-    data_manager = DataManager(base_dir='/data')
+    data_manager = DataManager(base_dir='data')
     with gr.Blocks() as demo:
         with gr.Tabs():
             prompt_generator_app(data_manager)