register gallery close writer thread on shutdown

Author: mythz

llms/extensions/gallery/__init__.py

@@ -19,6 +19,7 @@ def get_gallery_db():
             try:
                 db_path = os.path.join(ctx.get_user_path(), "gallery", "gallery.sqlite")
                 g_db = GalleryDB(ctx, db_path)
+                ctx.register_shutdown_handler(g_db.db.close)
             except Exception as e:
                 ctx.err("Failed to init GalleryDB", e)
         return g_db