Update app.py

app.py

@@ -150,21 +150,24 @@ def character_creation_app(data_manager):
 
         save_button.click(save_character, inputs=[name_input, traits_input, image_input, gender_input], outputs=output)
 
-def prompt_generator_app(data_manager, category_tags_state):
+def prompt_generator_app(data_manager):
     with gr.Tab("Prompt Generator"):
         gr.Markdown("## Prompt Generator")
 
+        # Add a refresh tags button
+        refresh_tags_button = gr.Button("Refresh Tags")
+
         inputs = {}
         tag_displays = {}
         for category_name, var_name in categories:
-            tags_list = category_tags_state.get(var_name, [])
+            tags_list = data_manager.get_category_tags(var_name)
             tags_string = ', '.join(tags_list)
             max_tags = len(tags_list)
             if max_tags == 0:
                 default_value = 0
             else:
                 default_value = min(1, max_tags)
-            with gr.Group():
+            with gr.Box():
                 gr.Markdown(f"### {category_name}")
                 tag_display = gr.Markdown(f"**Tags:** {tags_string}")
                 tag_num = gr.Slider(minimum=0, maximum=max_tags, step=1, value=default_value, label=f"Number of {category_name} Tags to Select")
@@ -201,12 +204,12 @@ def prompt_generator_app(data_manager, category_tags_state):
         generate_button = gr.Button("Generate Prompt")
         prompt_output = gr.Textbox(label="Generated Prompt", lines=5)
 
-        def generate_prompt(*args, data_manager=data_manager, categories=categories):
+        def generate_prompt(*args):
             arg_idx = 0
 
             prompt_tags = []
             for category_name, var_name in categories:
-                tags_list = category_tags_state.get(var_name, [])
+                tags_list = data_manager.get_category_tags(var_name)
                 tags_num = args[arg_idx]
                 arg_idx += 1
 
@@ -296,22 +299,28 @@ def prompt_generator_app(data_manager, category_tags_state):
 
         generate_button.click(generate_prompt, inputs=inputs_list, outputs=prompt_output)
 
-        # Function to update the tags display and sliders when tags are changed
-        def update_prompt_generator_tags(category_var_name, tags_list):
-            tags_string = ', '.join(tags_list)
-            max_tags = len(tags_list)
-            if max_tags == 0:
-                slider_value = 0
-            else:
-                slider_value = min(1, max_tags)
-            # Update the tag display and slider
-            tag_display, tag_num = tag_displays[category_var_name]
-            return {
-                tag_display: gr.update(value=f"**Tags:** {tags_string}"),
-                tag_num: gr.update(maximum=max_tags, value=slider_value)
-            }
-
-def tags_app(data_manager, category_tags_state):
+        # 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)
+                tags_string = ', '.join(tags_list)
+                max_tags = len(tags_list)
+                if max_tags == 0:
+                    slider_value = 0
+                else:
+                    slider_value = min(1, max_tags)
+                # Update the tag display and slider
+                tag_display, tag_num = tag_displays[var_name]
+                updates[tag_display] = gr.update(value=f"**Tags:** {tags_string}")
+                updates[tag_num] = gr.update(maximum=max_tags, value=slider_value)
+            return updates
+
+        # Connect the refresh_tags function to the refresh_tags_button
+        refresh_tags_button.click(refresh_tags, outputs=list(tag_displays.values()))
+
+def tags_app(data_manager):
     with gr.Tab("Tags"):
         gr.Markdown("## Edit Tags for Each Category")
 
@@ -323,21 +332,16 @@ def tags_app(data_manager, category_tags_state):
             save_button = gr.Button(f"Save {category_name} Tags")
             status_output = gr.Textbox(label="", interactive=False)
 
-            # Function to save tags and update state
+            # Function to save tags
             def make_save_category_tags_fn(var_name, category_name):
                 def fn(tags_string):
                     tags_list = [t.strip() for t in tags_string.split(',') if t.strip()]
                     data_manager.set_category_tags(var_name, tags_list)
-                    category_tags_state[var_name] = tags_list  # Update the state
-
-                    # Update the prompt generator components
-                    updates = prompt_generator_app.update_prompt_generator_tags(var_name, tags_list)
-                    return updates, f"{category_name} tags saved successfully."
+                    return f"{category_name} tags saved successfully."
                 return fn
 
             save_fn = make_save_category_tags_fn(var_name, category_name)
-            # Update both the prompt generator components and the status output
-            save_button.click(save_fn, inputs=tag_input, outputs=[gr.Group.update(), status_output])
+            save_button.click(save_fn, inputs=tag_input, outputs=status_output)
 
         # Persistent Tags
         gr.Markdown(f"### Persistent Tags")
@@ -355,12 +359,10 @@ def tags_app(data_manager, category_tags_state):
 
 def main():
     data_manager = DataManager()
-    category_tags_state = {var_name: data_manager.get_category_tags(var_name) for _, var_name in categories}
-
     with gr.Blocks() as demo:
-        prompt_generator_app(data_manager, category_tags_state)
+        prompt_generator_app(data_manager)
         character_creation_app(data_manager)
-        tags_app(data_manager, category_tags_state)
+        tags_app(data_manager)
 
     demo.launch()