Update app.py

app.py

@@ -150,13 +150,14 @@ 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):
+def prompt_generator_app(data_manager, category_tags_state):
     with gr.Tab("Prompt Generator"):
         gr.Markdown("## Prompt Generator")
 
         inputs = {}
+        tag_displays = {}
         for category_name, var_name in categories:
-            tags_list = data_manager.get_category_tags(var_name)
+            tags_list = category_tags_state.get(var_name, [])
             tags_string = ', '.join(tags_list)
             max_tags = len(tags_list)
             if max_tags == 0:
@@ -165,9 +166,10 @@ def prompt_generator_app(data_manager):
                 default_value = min(1, max_tags)
             with gr.Group():
                 gr.Markdown(f"### {category_name}")
-                gr.Markdown(f"**Tags:** {tags_string}")
+                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")
             inputs[f"{var_name}_num"] = tag_num
+            tag_displays[var_name] = (tag_display, tag_num)
 
         # For Character Selection
         with gr.Group():
@@ -204,7 +206,7 @@ def prompt_generator_app(data_manager):
 
             prompt_tags = []
             for category_name, var_name in categories:
-                tags_list = data_manager.get_category_tags(var_name)
+                tags_list = category_tags_state.get(var_name, [])
                 tags_num = args[arg_idx]
                 arg_idx += 1
 
@@ -294,7 +296,22 @@ def prompt_generator_app(data_manager):
 
         generate_button.click(generate_prompt, inputs=inputs_list, outputs=prompt_output)
 
-def tags_app(data_manager):
+        # 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):
     with gr.Tab("Tags"):
         gr.Markdown("## Edit Tags for Each Category")
 
@@ -306,16 +323,21 @@ def tags_app(data_manager):
             save_button = gr.Button(f"Save {category_name} Tags")
             status_output = gr.Textbox(label="", interactive=False)
 
-            # Capture var_name and category_name in the closure
+            # Function to save tags and update state
             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)
-                    return f"{category_name} tags saved successfully."
+                    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 fn
 
             save_fn = make_save_category_tags_fn(var_name, category_name)
-            save_button.click(save_fn, inputs=tag_input, outputs=status_output)
+            # Update both the prompt generator components and the status output
+            save_button.click(save_fn, inputs=tag_input, outputs=[gr.Group.update(), status_output])
 
         # Persistent Tags
         gr.Markdown(f"### Persistent Tags")
@@ -333,10 +355,12 @@ def tags_app(data_manager):
 
 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)
+        prompt_generator_app(data_manager, category_tags_state)
         character_creation_app(data_manager)
-        tags_app(data_manager)
+        tags_app(data_manager, category_tags_state)
 
     demo.launch()