From c639fa79a7965ceb07c6623ded2fac734b9415ea Mon Sep 17 00:00:00 2001 From: Vito Sansevero Date: Tue, 27 May 2025 12:35:27 -0700 Subject: [PATCH] fix(database): delete related images before prompts --- database/operations.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/database/operations.py b/database/operations.py index e467bfe..017e63c 100644 --- a/database/operations.py +++ b/database/operations.py @@ -440,6 +440,9 @@ class PromptDatabase: ids_to_delete = ids[1:] # Skip the first ID for id_to_delete in ids_to_delete: + # First delete related images to avoid foreign key constraint + conn.execute("DELETE FROM generated_images WHERE prompt_id = ?", (id_to_delete,)) + # Then delete the prompt conn.execute("DELETE FROM prompts WHERE id = ?", (int(id_to_delete),)) total_removed += 1