Update app.py

app.py

@@ -8,7 +8,7 @@ import io
 
 # Define categories at the top so they are accessible throughout the code
 categories = [
-    ('Scene', 'scene_tags'),
+    ('Setting', 'scene_tags'),
     ('Position', 'position_tags'),
     ('Outfit', 'outfit_tags'),
     ('Camera View/Angle', 'camera_tags'),
@@ -20,18 +20,20 @@ categories = [
 ]
 
 class DataManager:
-    def __init__(self, characters_file='characters.json',
-                 persistent_tags_file='persistent_tags.json',
-                 category_tags_file='category_tags.json',
-                 images_folder='character_images'):
-        self.characters_file = characters_file
-        self.persistent_tags_file = persistent_tags_file
-        self.category_tags_file = category_tags_file
-        self.images_folder = images_folder
-
-        # Make sure image folder exists
-        if not os.path.exists(images_folder):
-            os.makedirs(images_folder)
+    def __init__(self, base_dir='/persistent'):
+        self.base_dir = base_dir
+        # Ensure the base directory exists
+        if not os.path.exists(self.base_dir):
+            os.makedirs(self.base_dir)
+        
+        self.characters_file = os.path.join(self.base_dir, 'characters.json')
+        self.persistent_tags_file = os.path.join(self.base_dir, 'persistent_tags.json')
+        self.category_tags_file = os.path.join(self.base_dir, 'category_tags.json')
+        self.images_folder = os.path.join(self.base_dir, 'character_images')
+
+        # Ensure the images folder exists
+        if not os.path.exists(self.images_folder):
+            os.makedirs(self.images_folder)
 
         self.load_data()
 
@@ -100,7 +102,8 @@ class DataManager:
     def add_character(self, character):
         # Save image to disk and store the filename
         image_data = character['image']  # This is base64 encoded string
-        image_filename = f"{character['name']}.png"
+        safe_name = "".join(c for c in character['name'] if c.isalnum() or c in (' ', '_', '-')).rstrip()
+        image_filename = f"{safe_name}.png"
         image_path = os.path.join(self.images_folder, image_filename)
 
         # Decode the base64 image data and save it
@@ -361,7 +364,7 @@ def tags_app(data_manager):
         save_persistent_tags_button.click(save_persistent_tags, inputs=persistent_tags_input, outputs=persistent_status_output)
 
 def main():
-    data_manager = DataManager()
+    data_manager = DataManager(base_dir='/persistent')
     with gr.Blocks() as demo:
         prompt_generator_app(data_manager)
         character_creation_app(data_manager)