From 929b05a37d2bdbe1b93b433026a014988a37aa14 Mon Sep 17 00:00:00 2001 From: Konstantine Tsafatinos Date: Fri, 21 Jul 2023 17:20:31 -0400 Subject: [PATCH] fix increment generated images bugs, add func to redo, fix str error --- skynet/frontend/telegram/handlers.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/skynet/frontend/telegram/handlers.py b/skynet/frontend/telegram/handlers.py index c6982e5..532198d 100644 --- a/skynet/frontend/telegram/handlers.py +++ b/skynet/frontend/telegram/handlers.py @@ -159,7 +159,7 @@ def create_handler_context(frontend: 'SkynetTelegramFrontend'): ec = await work_request(user, status_msg, 'txt2img', params) - if ec == 0: + if ec == None: await db_call('increment_generated', user.id) @@ -187,7 +187,7 @@ def create_handler_context(frontend: 'SkynetTelegramFrontend'): user_row = await db_call('get_or_create_user', user.id) # init new msg - init_msg = 'started processing txt2img request...' + init_msg = 'started processing img2img request...' status_msg = await bot.reply_to(message, init_msg) await db_call( 'new_user_request', user.id, message.id, status_msg.id, status=init_msg) @@ -249,7 +249,7 @@ def create_handler_context(frontend: 'SkynetTelegramFrontend'): binary_data=ipfs_hash ) - if ec == 0: + if ec == None: await db_call('increment_generated', user.id) @@ -307,12 +307,14 @@ def create_handler_context(frontend: 'SkynetTelegramFrontend'): **user_config } - await work_request( + ec = await work_request( user, status_msg, 'redo', params, file_id=file_id, binary_data=binary ) + if ec == None: + await db_call('increment_generated', user.id) # "proxy" handlers just request routers