From 077806eb80ed7cc54ea02dfcdf5e7afeb4b425d6 Mon Sep 17 00:00:00 2001
From: Boris Kovalskii <36034823+JustYay@users.noreply.github.com>
Date: Thu, 24 Jul 2025 13:34:07 +1000
Subject: [PATCH] Custom amount and deletion fixes

---
 handlers/payments/kassai.py | 2 +-
 web/kassai_payment.py       | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/handlers/payments/kassai.py b/handlers/payments/kassai.py
index 833344fa..8231ad0a 100644
--- a/handlers/payments/kassai.py
+++ b/handlers/payments/kassai.py
@@ -204,7 +204,7 @@ async def handle_custom_amount_input(message: types.Message, state: FSMContext):
         return
     
     await state.update_data(amount=amount)
-    payment_url = await generate_kassai_payment_link(amount, callback_query.message.chat.id, method)
+    payment_url = await generate_kassai_payment_link(amount, message.chat.id, method)
     
     confirm_keyboard = InlineKeyboardMarkup(
         inline_keyboard=[
diff --git a/web/kassai_payment.py b/web/kassai_payment.py
index b9805163..89cde958 100644
--- a/web/kassai_payment.py
+++ b/web/kassai_payment.py
@@ -1,5 +1,6 @@
 import hashlib
 import json
+import bot
 from aiohttp import web
 from database import add_payment, async_session_maker, update_balance
 from handlers.payments.utils import send_payment_success_notification