delete msg with question md

1 files changed, 8 insertions, 2 deletions

diff --git a/callbacks.py b/callbacks.py
index 432c0af..f6649a7 100644
--- a/callbacks.py
+++ b/callbacks.py
@@ -361,7 +361,7 @@ async def set_markdown_callback(callback: CallbackQuery):
 
 
 @router.callback_query(F.data.startswith("new_md%"))
-async def set_markdown_callback(callback: CallbackQuery):
+async def set_markdown_callback(callback: CallbackQuery, message=None):
     current_user_id = str(callback.from_user.id)
     data = callback.data.split("%", 2)
     user_id = data[1]
@@ -378,11 +378,14 @@ async def set_markdown_callback(callback: CallbackQuery):
         config.get("notify"), 
         "new",
     )
+    if message:
+        await message.delete()
     await callback.answer("выбран новый стиль!!")
+    await callback.message.delete()
 
 
 @router.callback_query(F.data.startswith("old_md%"))
-async def set_markdown_callback(callback: CallbackQuery):
+async def set_markdown_callback(callback: CallbackQuery, message=None):
     current_user_id = str(callback.from_user.id)
     data = callback.data.split("%", 2)
     user_id = data[1]
@@ -399,7 +402,10 @@ async def set_markdown_callback(callback: CallbackQuery):
         config.get("notify"), 
         "old",
     )
+    if message:
+        await message.delete()
     await callback.answer("выбран старый стиль!!")
+    await callback.message.delete()
 
 
 @router.callback_query(F.data == "start")