Update app.py

app.py

@@ -107,7 +107,7 @@ class DataManager:
         image_path = os.path.join(self.images_folder, image_filename)
 
         # Save the image if provided
-        if image_data:
+        if image_path and os.path.exists(image_data):
             try:
                 # Open and save the uploaded image to the images folder
                 image = Image.open(image_data)
@@ -135,13 +135,13 @@ class DataManager:
             if char['name'] == original_name:
                 # Handle image update
                 if updated_character['image']:
-                    image_path = updated_character['image']  # File path
+                    image_data = updated_character['image']  # File path
                     safe_name = "".join(c for c in updated_character['name'] if c.isalnum() or c in (' ', '_', '-')).rstrip()
                     image_filename = f"{safe_name}.png"
                     new_image_path = os.path.join(self.images_folder, image_filename)
 
                     try:
-                        image = Image.open(image_path)
+                        image = Image.open(image_data)
                         image.save(new_image_path)
                         updated_character['image_path'] = new_image_path
                         # Remove old image if name has changed
@@ -354,37 +354,49 @@ def prompt_generator_app(data_manager):
 
 def character_creation_app(data_manager):
     with gr.Tab("Character Creation"):
-        gr.Markdown("## Create a New Character")
+        gr.Markdown("## Create and Manage Characters")
+        # **Section 1: Create a New Character**
         with gr.Row():
             name_input = gr.Textbox(label="Character Name", placeholder="Enter unique character name")
             traits_input = gr.Textbox(label="Traits/Appearance Tags (comma separated)", placeholder="e.g., blue hair, green eyes, tall")
         image_input = gr.Image(label="Upload Character Image", type="filepath")
         gender_input = gr.Radio(choices=["Boy", "Girl"], label="Gender")
         save_button = gr.Button("Save Character")
-        output = gr.Textbox(label="Status", interactive=False)
-
-        characters_renderer = gr.HTML(label="Existing Characters")
-
-        def list_characters():
-            characters = data_manager.get_characters()
-            if not characters:
-                return "<p>No characters created yet.</p>"
-
-            html_content = ""
-            for char in characters:
-                html_content += f"""
-                <div style="border:1px solid #ccc; padding:10px; margin-bottom:10px;">
-                    <h3>{char['name']}</h3>
-                    <p><strong>Gender:</strong> {char['gender']}</p>
-                    <p><strong>Traits:</strong> {', '.join(char['traits'])}</p>
-                </div>
-                """
-            return html_content
-
+        save_output = gr.Textbox(label="Status", interactive=False)
+
+        # **Section 2: Existing Characters**
+        gr.Markdown("---")
+        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]
+        selected_character = gr.Dropdown(choices=character_names, label="Select a Character to Edit/Delete", interactive=True)
+
+        # Buttons for Edit and Delete
+        edit_button = gr.Button("Edit Selected Character")
+        delete_button = gr.Button("Delete Selected Character")
+        action_output = gr.Textbox(label="Action Status", interactive=False)
+
+        # **Section 3: Edit Character**
+        edit_section = gr.Row(visible=False)
+        with edit_section:
+            edit_name = gr.Textbox(label="New Character Name", placeholder="Enter new character name")
+            edit_traits = gr.Textbox(label="New Traits/Appearance Tags (comma separated)", placeholder="e.g., red hair, blue eyes")
+            edit_image = gr.Image(label="Upload New Character Image", type="filepath")
+            edit_gender = gr.Radio(choices=["Boy", "Girl"], label="Gender")
+            update_button = gr.Button("Update Character")
+            update_output = gr.Textbox(label="Update Status", interactive=False)
+
+        # **Section 4: Delete Confirmation**
+        delete_confirmation = gr.Textbox(label="Are you sure you want to delete the selected character? Type 'YES' to confirm.", interactive=True, visible=False)
+        confirm_delete_button = gr.Button("Confirm Delete", visible=False)
+        cancel_delete_button = gr.Button("Cancel Delete", visible=False)
+
+        # **Function to Save New Character**
         def save_character(name, traits, image_path, gender):
             if not name.strip() or not traits.strip() or not gender:
                 return "Please enter all fields."
-
             # Check for duplicate names
             existing_names = [char['name'] for char in data_manager.get_characters()]
             if name in existing_names:
@@ -392,25 +404,90 @@ def character_creation_app(data_manager):
 
             character = {'name': name, 'traits': traits, 'gender': gender, 'image': image_path}
             data_manager.add_character(character)
-
             return f"Character '{name}' saved successfully."
 
         save_button.click(
             save_character,
             inputs=[name_input, traits_input, image_input, gender_input],
-            outputs=output
+            outputs=save_output
         )
 
-        # Display Existing Characters
-        refresh_characters_button = gr.Button("Refresh Characters")
-        refresh_characters_button.click(
-            lambda: list_characters(),
-            inputs=None,
-            outputs=characters_renderer
+        # **Function to Show Edit Section**
+        def show_edit(selected_name):
+            if not selected_name:
+                return "Please select a character to edit.", gr.update(visible=False)
+            return "", gr.update(visible=True)
+
+        edit_button.click(
+            show_edit,
+            inputs=selected_character,
+            outputs=[action_output, edit_section]
+        )
+
+        # **Function to Update Character**
+        def update_character(original_name, new_name, new_traits, new_image_path, new_gender):
+            if not new_name.strip() or not new_traits.strip() or not new_gender:
+                return "Please enter all fields.", gr.update(visible=False)
+            # If the name has changed, check for duplicates
+            if new_name != original_name:
+                existing_names = [char['name'] for char in data_manager.get_characters()]
+                if new_name in existing_names:
+                    return f"Character with name '{new_name}' already exists. Please choose a different name.", gr.update(visible=True)
+            updated_char = {
+                'name': new_name,
+                'traits': new_traits,
+                'gender': new_gender,
+                'image': new_image_path if new_image_path else [char for char in data_manager.get_characters() if char['name'] == original_name][0]['image']
+            }
+            success = data_manager.update_character(original_name, updated_char)
+            if success:
+                return f"Character '{new_name}' updated successfully.", gr.update(visible=False)
+            else:
+                return "Failed to update character.", gr.update(visible=True)
+
+        update_button.click(
+            update_character,
+            inputs=[selected_character, edit_name, edit_traits, edit_image, edit_gender],
+            outputs=[update_output, edit_section]
         )
 
-        # Initial load of characters
-        characters_renderer.render(list_characters())
+        # **Function to Show Delete Confirmation**
+        def show_delete_confirmation(selected_name):
+            if not selected_name:
+                return "Please select a character to delete.", gr.update(visible=False), gr.update(visible=False)
+            return f"Are you sure you want to delete '{selected_name}'? Type 'YES' to confirm.", gr.update(visible=True), gr.update(visible=True)
+
+        delete_button.click(
+            show_delete_confirmation,
+            inputs=selected_character,
+            outputs=[action_output, delete_confirmation, confirm_delete_button]
+        )
+
+        # **Function to Perform Deletion**
+        def perform_deletion(selected_name, confirmation):
+            if confirmation != "YES":
+                return "Deletion cancelled.", gr.update(visible=False), gr.update(visible=False)
+            success = data_manager.delete_character(selected_name)
+            if success:
+                return f"Character '{selected_name}' deleted successfully.", gr.update(visible=False), gr.update(visible=False)
+            else:
+                return f"Failed to delete character '{selected_name}'.", gr.update(visible=False), gr.update(visible=False)
+
+        confirm_delete_button.click(
+            perform_deletion,
+            inputs=[selected_character, delete_confirmation],
+            outputs=[action_output, delete_confirmation, confirm_delete_button]
+        )
+
+        # **Function to Cancel Deletion**
+        def cancel_deletion():
+            return "Deletion cancelled.", gr.update(visible=False), gr.update(visible=False)
+
+        cancel_delete_button.click(
+            cancel_deletion,
+            inputs=None,
+            outputs=[action_output, delete_confirmation, confirm_delete_button]
+        )
 
 def tags_app(data_manager):
     with gr.Tab("Tags"):
@@ -441,12 +518,12 @@ def tags_app(data_manager):
         save_persistent_tags_button = gr.Button("Save Persistent Tags")
         persistent_status_output = gr.Textbox(label="", interactive=False)
 
-        def save_persistent_tags(tags_string):
+        def save_persistent_tags_fn(tags_string):
             tags_list = [t.strip() for t in tags_string.split(',') if t.strip()]
             data_manager.set_persistent_tags(tags_list)
             return "Persistent tags saved successfully."
 
-        save_persistent_tags_button.click(save_persistent_tags, inputs=persistent_tags_input, outputs=persistent_status_output)
+        save_persistent_tags_button.click(save_persistent_tags_fn, inputs=persistent_tags_input, outputs=persistent_status_output)
 
 def main():
     data_manager = DataManager(base_dir='/data')
@@ -455,7 +532,6 @@ def main():
             prompt_generator_app(data_manager)
             character_creation_app(data_manager)
             tags_app(data_manager)
-
     demo.launch()
 
 if __name__ == "__main__":