tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Creating new prompt on prompt command

Browse Source
This commit is contained in:
BarsTiger
2023-03-02 19:34:29 +02:00
parent 3c3d644137
commit 17f00689a5
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bot/handlers/txt2img/set_settings.py
View File

@@ -20,6 +20,8 @@ async def _set_property(message: types.Message, prop: str, value=None):
f"medium hair, cat ears, closed eyes, looking at viewer, :3, cute, scarf, " f"medium hair, cat ears, closed eyes, looking at viewer, :3, cute, scarf, "
f"jacket, outdoors, streets</code>", parse_mode='HTML') f"jacket, outdoors, streets</code>", parse_mode='HTML')
return return
elif prompt is None:
prompt = Prompt(message.get_args(), creator=message.from_id)
prompt.__setattr__(prop, message.get_args() if value is None else value) prompt.__setattr__(prop, message.get_args() if value is None else value)
prompt.creator = message.from_id prompt.creator = message.from_id
@@ -139,13 +141,14 @@ async def set_sampler_command(message: types.Message):
except ClientConnectorError: except ClientConnectorError:
await message.reply('❌ Error! Maybe, StableDiffusion API endpoint is incorrect ' await message.reply('❌ Error! Maybe, StableDiffusion API endpoint is incorrect '
'or turned off') 'or turned off')
await temp_message.delete()
return
except Exception as e: except Exception as e:
exception_id = f'{message.message_thread_id}-{message.message_id}' exception_id = f'{message.message_thread_id}-{message.message_id}'
db[DBTables.exceptions][exception_id] = PrettyException(e) db[DBTables.exceptions][exception_id] = PrettyException(e)
await message.reply('❌ Error happened while processing your request', parse_mode='html', await message.reply('❌ Error happened while processing your request', parse_mode='html',
reply_markup=get_exception_keyboard(exception_id)) reply_markup=get_exception_keyboard(exception_id))
await temp_message.delete() await temp_message.delete()
db[DBTables.queue]['n'] = db[DBTables.queue].get('n', 1) - 1
return return
await _set_property(message, 'sampler') await _set_property(message, 'sampler')