dont work translate from long message
add button send file to VK album
This commit is contained in:
parent
33685e3cde
commit
09cffdbac9
@ -10,6 +10,7 @@ pip install aiogram
|
|||||||
pip install webuiapi
|
pip install webuiapi
|
||||||
pip install translate
|
pip install translate
|
||||||
pip install transformers
|
pip install transformers
|
||||||
|
pip install vk_api
|
||||||
cd C:\stable-diffusion-webui\extensions\sd_telegram\
|
cd C:\stable-diffusion-webui\extensions\sd_telegram\
|
||||||
python bot.py
|
python bot.py
|
||||||
``````
|
``````
|
||||||
@ -76,6 +77,9 @@ Please pay attention to the file size limit in Telegram API:
|
|||||||
_**Chat History**
|
_**Chat History**
|
||||||
We go into any Telegram chat with prompts (individual messages), click three dots in the upper right corner, upload only text messages in json format. We get the result.json file, which we throw into the bot and select "Chat History". We get a random prompt, which we can save in data
|
We go into any Telegram chat with prompts (individual messages), click three dots in the upper right corner, upload only text messages in json format. We get the result.json file, which we throw into the bot and select "Chat History". We get a random prompt, which we can save in data
|
||||||
|
|
||||||
|
If you chose _/img_real = true_, the document will be unloaded for you, and after it the social network **VK** upload button.
|
||||||
|
Before that, you need to set up a token and enter the ID of the album where the photo will be sent.
|
||||||
|
|
||||||
<img src="https://raw.githubusercontent.com/partyfind/sd_bot/master/trash/photo_2023-06-22_15-29-24.jpg" alt="drawing" width="400"/>
|
<img src="https://raw.githubusercontent.com/partyfind/sd_bot/master/trash/photo_2023-06-22_15-29-24.jpg" alt="drawing" width="400"/>
|
||||||
<img src="https://raw.githubusercontent.com/partyfind/sd_bot/master/trash/photo_2023-06-22_15-29-27.jpg" width="350"/>
|
<img src="https://raw.githubusercontent.com/partyfind/sd_bot/master/trash/photo_2023-06-22_15-29-27.jpg" width="350"/>
|
||||||
|
|
||||||
@ -86,6 +90,7 @@ We go into any Telegram chat with prompts (individual messages), click three dot
|
|||||||
4. Ability to send everything with one command with settings
|
4. Ability to send everything with one command with settings
|
||||||
5. Preloading photos when waiting for a long time so that you can skip
|
5. Preloading photos when waiting for a long time so that you can skip
|
||||||
6. Progress in script
|
6. Progress in script
|
||||||
|
7. Translator capable of translating up to 4000 characters and detecting the language
|
||||||
|
|
||||||
**TNX**
|
**TNX**
|
||||||
[AUTOMATIC1111](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
[AUTOMATIC1111](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
||||||
|
70
bot.py
70
bot.py
@ -31,6 +31,8 @@ from translate import Translator
|
|||||||
import base64
|
import base64
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
import logging
|
import logging
|
||||||
|
import vk_api
|
||||||
|
from vk_api import VkUpload
|
||||||
|
|
||||||
# Настройка логгера
|
# Настройка логгера
|
||||||
logging.basicConfig(format="[%(asctime)s] %(levelname)s : %(name)s : %(message)s",
|
logging.basicConfig(format="[%(asctime)s] %(levelname)s : %(name)s : %(message)s",
|
||||||
@ -41,9 +43,16 @@ logging.getLogger('aiogram').setLevel(logging.DEBUG)
|
|||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
# from https://t.me/BotFather
|
# from https://t.me/BotFather
|
||||||
API_TOKEN = "TOKEN_HERE"
|
API_BOT_TOKEN = "TOKEN_HERE"
|
||||||
|
#заходим в https://oauth.vk.com/authorize?client_id=123&scope=photos&redirect_uri=http%3A%2F%2Foauth.vk.com%2Fblank.html&display=page&response_type=token,
|
||||||
|
# где 123 - номер вашего включенного приложения, созданного в https://vk.com/apps?act=manage,
|
||||||
|
# photos - зона доступа.
|
||||||
|
# После перехода и подтверждения выцепляем access_token из адресной строки
|
||||||
|
# TODO auto requests
|
||||||
|
VK_TOKEN = 'TOKEN_HERE'
|
||||||
|
vk_album_id = '789' # брать с адресной строки, когда открываешь ВК. Пример https://vk.com/album123_789
|
||||||
|
|
||||||
bot = Bot(token=API_TOKEN)
|
bot = Bot(token=API_BOT_TOKEN)
|
||||||
storage = MemoryStorage()
|
storage = MemoryStorage()
|
||||||
dp = Dispatcher(bot, storage=storage)
|
dp = Dispatcher(bot, storage=storage)
|
||||||
|
|
||||||
@ -198,7 +207,7 @@ def rnd_prmt_lxc():
|
|||||||
return txt
|
return txt
|
||||||
|
|
||||||
# get settings. TODO - cut 4000 symbols
|
# get settings. TODO - cut 4000 symbols
|
||||||
def get_prompt_settings():
|
def get_prompt_settings(typeCode = 'HTML'):
|
||||||
global sd
|
global sd
|
||||||
prompt = data['prompt'].replace('<', '<').replace('>', '>')
|
prompt = data['prompt'].replace('<', '<').replace('>', '>')
|
||||||
cfg_scale = data['cfg_scale']
|
cfg_scale = data['cfg_scale']
|
||||||
@ -211,7 +220,10 @@ def get_prompt_settings():
|
|||||||
sd_model_checkpoint = dataParams['sd_model_checkpoint']
|
sd_model_checkpoint = dataParams['sd_model_checkpoint']
|
||||||
else:
|
else:
|
||||||
sd_model_checkpoint = api.get_options()['sd_model_checkpoint']
|
sd_model_checkpoint = api.get_options()['sd_model_checkpoint']
|
||||||
txt = f"prompt = <code>{prompt}</code>\nsteps = {steps} \ncfg_scale = {cfg_scale} \nwidth = {width} \nheight = {height} \nsampler_name = {sampler_name} \nsd_model_checkpoint = {sd_model_checkpoint} \nnegative_prompt = <code>{negative_prompt}</code> "
|
if typeCode == 'HTML':
|
||||||
|
txt = f"prompt = <code>{prompt}</code>\nsteps = {steps} \ncfg_scale = {cfg_scale} \nwidth = {width} \nheight = {height} \nsampler_name = {sampler_name} \nsd_model_checkpoint = {sd_model_checkpoint} \nnegative_prompt = <code>{negative_prompt}</code> "
|
||||||
|
else:
|
||||||
|
txt = f"prompt = {prompt}\n\nsteps = {steps} cfg_scale = {cfg_scale} width = {width} height = {height} sampler_name = {sampler_name} sd_model_checkpoint = {sd_model_checkpoint} \n\nnegative_prompt = {negative_prompt} "
|
||||||
return txt
|
return txt
|
||||||
|
|
||||||
# Translate
|
# Translate
|
||||||
@ -606,12 +618,12 @@ async def inl_uplchat(callback: types.CallbackQuery) -> None:
|
|||||||
t = t2
|
t = t2
|
||||||
break
|
break
|
||||||
|
|
||||||
chatHistoryPrompt = translateRuToEng(t)
|
chatHistoryPrompt = t#translateRuToEng(t)
|
||||||
keyboard = InlineKeyboardMarkup(inline_keyboard=[getPromptFromJson(0), getStart(0)])
|
keyboard = InlineKeyboardMarkup(inline_keyboard=[getPromptFromJson(0), getStart(0)])
|
||||||
await bot.edit_message_text(
|
await bot.edit_message_text(
|
||||||
chat_id=callback.message.chat.id,
|
chat_id=callback.message.chat.id,
|
||||||
message_id=callback.message.message_id,
|
message_id=callback.message.message_id,
|
||||||
text=translateRuToEng(t).replace('<', '<').replace('>', '>'),
|
text=t.replace('<', '<').replace('>', '>'),#translateRuToEng(t).replace('<', '<').replace('>', '>'),
|
||||||
reply_markup=keyboard,
|
reply_markup=keyboard,
|
||||||
parse_mode = types.ParseMode.HTML
|
parse_mode = types.ParseMode.HTML
|
||||||
)
|
)
|
||||||
@ -736,8 +748,17 @@ async def inl_gen(message: Union[types.Message, types.CallbackQuery]) -> None:
|
|||||||
chat_id=chatId, media=pilToImages(res, "tg")
|
chat_id=chatId, media=pilToImages(res, "tg")
|
||||||
)
|
)
|
||||||
if dataParams["img_real"] == "true" or dataParams["img_real"] == "True":
|
if dataParams["img_real"] == "true" or dataParams["img_real"] == "True":
|
||||||
await bot.send_media_group(
|
mes_file = await bot.send_media_group(
|
||||||
chat_id=chatId, media=pilToImages(res, "real")
|
chat_id=chatId,
|
||||||
|
media=pilToImages(res, "real")
|
||||||
|
)
|
||||||
|
# send button load in VK
|
||||||
|
# TODO long message
|
||||||
|
await bot.send_message(
|
||||||
|
chat_id=chatId,
|
||||||
|
text="↓ send to VK",
|
||||||
|
reply_markup=InlineKeyboardMarkup(
|
||||||
|
inline_keyboard=[[InlineKeyboardButton(mes_file[0].document.file_id, callback_data='send_vk')]])
|
||||||
)
|
)
|
||||||
await bot.send_message(
|
await bot.send_message(
|
||||||
chat_id=chatId,
|
chat_id=chatId,
|
||||||
@ -763,6 +784,37 @@ async def inl_gen(message: Union[types.Message, types.CallbackQuery]) -> None:
|
|||||||
keyboard = InlineKeyboardMarkup(inline_keyboard=[getSet(0), getOpt(0), getStart(0)])
|
keyboard = InlineKeyboardMarkup(inline_keyboard=[getSet(0), getOpt(0), getStart(0)])
|
||||||
await getKeyboardUnion("Turn on SD"+sd, message, keyboard)
|
await getKeyboardUnion("Turn on SD"+sd, message, keyboard)
|
||||||
|
|
||||||
|
# upload in VK
|
||||||
|
@dp.callback_query_handler(text="send_vk")
|
||||||
|
async def inl_samplers(callback: types.CallbackQuery) -> None:
|
||||||
|
try:
|
||||||
|
global VK_TOKEN, vk_album_id
|
||||||
|
file_id = callback.message.reply_markup.inline_keyboard[0][0].text #TODO
|
||||||
|
file_obj = await bot.get_file(file_id)
|
||||||
|
vk_session = vk_api.VkApi(token=VK_TOKEN)
|
||||||
|
vk_upload = VkUpload(vk_session)
|
||||||
|
file_url = f'https://api.telegram.org/file/bot{API_BOT_TOKEN}/{file_obj.file_path}'
|
||||||
|
|
||||||
|
#TODO optimize
|
||||||
|
with open('temp.png', 'wb') as file:
|
||||||
|
file.write(requests.get(file_url).content)
|
||||||
|
vk_upload.photo(
|
||||||
|
photos='temp.png',
|
||||||
|
album_id=vk_album_id,
|
||||||
|
caption=get_prompt_settings(0)
|
||||||
|
)
|
||||||
|
# clear garbage
|
||||||
|
os.remove('temp.png')
|
||||||
|
await callback.message.edit_text(
|
||||||
|
'Фотка в VK загружена'
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
await bot.send_message(
|
||||||
|
chat_id=callback.message.chat.id,
|
||||||
|
text=e,
|
||||||
|
parse_mode=types.ParseMode.HTML
|
||||||
|
)
|
||||||
|
|
||||||
# Получить меню действий с промптами
|
# Получить меню действий с промптами
|
||||||
@dp.callback_query_handler(text="prompt")
|
@dp.callback_query_handler(text="prompt")
|
||||||
async def cmd_prompt(message: Union[types.Message, types.CallbackQuery]) -> None:
|
async def cmd_prompt(message: Union[types.Message, types.CallbackQuery]) -> None:
|
||||||
@ -1104,7 +1156,7 @@ async def change_json(message: types.Message):
|
|||||||
f"JSON параметры:\n{getJson()}\n{getJson(1)}", reply_markup=keyboard
|
f"JSON параметры:\n{getJson()}\n{getJson(1)}", reply_markup=keyboard
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
data["prompt"] = translateRuToEng(message.text)
|
data["prompt"] = message.text#translateRuToEng(message.text)
|
||||||
await bot.delete_message(chat_id=message.chat.id, message_id=message.message_id)
|
await bot.delete_message(chat_id=message.chat.id, message_id=message.message_id)
|
||||||
await message.answer(
|
await message.answer(
|
||||||
f"Записали промпт. JSON параметры:\n{getJson()}\n{getJson(1)}",
|
f"Записали промпт. JSON параметры:\n{getJson()}\n{getJson(1)}",
|
||||||
|
Loading…
Reference in New Issue
Block a user