mirror of https://github.com/skygpu/skynet.git
Change img2img msg handler on telegram frontend
parent
67286bfa23
commit
d1fc9ab952
|
@ -125,12 +125,15 @@ async def run_skynet_telegram(
|
||||||
|
|
||||||
await bot.reply_to(message, resp_txt)
|
await bot.reply_to(message, resp_txt)
|
||||||
|
|
||||||
@bot.message_handler(commands=['img2img'], content_types=['photo'])
|
@bot.message_handler(func=lambda message: True, content_types=['photo'])
|
||||||
async def send_img2img(message):
|
async def send_img2img(message):
|
||||||
chat = message.chat
|
chat = message.chat
|
||||||
if chat.type != 'group' and chat.id != GROUP_ID:
|
if chat.type != 'group' and chat.id != GROUP_ID:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
if not message.caption.startswith('/img2img'):
|
||||||
|
return
|
||||||
|
|
||||||
prompt = ' '.join(message.caption.split(' ')[1:])
|
prompt = ' '.join(message.caption.split(' ')[1:])
|
||||||
|
|
||||||
if len(prompt) == 0:
|
if len(prompt) == 0:
|
||||||
|
|
Loading…
Reference in New Issue