Ok updated
Browse files- chatbot/plugins/chat.py +56 -10
chatbot/plugins/chat.py
CHANGED
@@ -1053,6 +1053,7 @@ async def chatbot_talk(client: Client, message: Message):
|
|
1053 |
stream_image_now = re.match(r"--stream-images\s*(.*)", caption, re.IGNORECASE)
|
1054 |
if stream_image_now:
|
1055 |
STREAM_IMAGE_PATH = await message.download()
|
|
|
1056 |
try:
|
1057 |
backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
|
1058 |
backup_chat.append({"role": "user", "parts": [{"text": caption}]})
|
@@ -1070,16 +1071,24 @@ async def chatbot_talk(client: Client, message: Message):
|
|
1070 |
if captions == "":
|
1071 |
return await message.reply_text("Server busy try again later")
|
1072 |
|
1073 |
-
keyboard = create_keyboard(
|
|
|
|
|
|
|
1074 |
await message.reply_text(
|
1075 |
captions,
|
1076 |
disable_web_page_preview=True,
|
1077 |
reply_markup=keyboard
|
1078 |
)
|
1079 |
|
|
|
|
|
|
|
|
|
|
|
1080 |
await db.backup_chatbot.update_one(
|
1081 |
{"user_id": message.from_user.id},
|
1082 |
-
{"$
|
1083 |
upsert=True
|
1084 |
)
|
1085 |
backup_chat.append({"role": "model", "parts": [{"text": captions}]})
|
@@ -1236,20 +1245,29 @@ async def chatbot_talk(client: Client, message: Message):
|
|
1236 |
model=model_,
|
1237 |
contents=[str(caption), file_photo]
|
1238 |
)
|
|
|
1239 |
if len(response.text) > 4096:
|
1240 |
text_parts = split_message(response.text)
|
1241 |
for part in text_parts:
|
1242 |
await message.reply_text(part)
|
1243 |
else:
|
1244 |
-
keyboard_like = create_keyboard(
|
|
|
|
|
|
|
1245 |
await message.reply_text(
|
1246 |
response.text,
|
1247 |
disable_web_page_preview=True,
|
1248 |
reply_markup=keyboard_like
|
1249 |
)
|
|
|
|
|
|
|
|
|
|
|
1250 |
await db.backup_chatbot.update_one(
|
1251 |
{"user_id": message.from_user.id},
|
1252 |
-
{"$
|
1253 |
upsert=True
|
1254 |
)
|
1255 |
backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
|
@@ -1290,20 +1308,29 @@ async def chatbot_talk(client: Client, message: Message):
|
|
1290 |
model=model_,
|
1291 |
contents=[str(caption), video_file]
|
1292 |
)
|
|
|
1293 |
if len(response.text) > 4096:
|
1294 |
text_parts = split_message(response.text)
|
1295 |
for part in text_parts:
|
1296 |
await message.reply_text(part)
|
1297 |
else:
|
1298 |
-
keyboard_like = create_keyboard(
|
|
|
|
|
|
|
1299 |
await message.reply_text(
|
1300 |
response.text,
|
1301 |
disable_web_page_preview=True,
|
1302 |
reply_markup=keyboard_like
|
1303 |
)
|
|
|
|
|
|
|
|
|
|
|
1304 |
await db.backup_chatbot.update_one(
|
1305 |
{"user_id": message.from_user.id},
|
1306 |
-
{"$
|
1307 |
upsert=True
|
1308 |
)
|
1309 |
backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
|
@@ -1343,20 +1370,30 @@ async def chatbot_talk(client: Client, message: Message):
|
|
1343 |
model=model_,
|
1344 |
contents=["funny random meme sticker words", sticker_file]
|
1345 |
)
|
|
|
1346 |
if len(response.text) > 4096:
|
1347 |
text_parts = split_message(response.text)
|
1348 |
for part in text_parts:
|
1349 |
await message.reply_text(part)
|
1350 |
else:
|
1351 |
-
keyboard_like = create_keyboard(
|
|
|
|
|
|
|
1352 |
await message.reply_text(
|
1353 |
response.text,
|
1354 |
disable_web_page_preview=True,
|
1355 |
reply_markup=keyboard_like
|
1356 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
1357 |
await db.backup_chatbot.update_one(
|
1358 |
{"user_id": message.from_user.id},
|
1359 |
-
{"$
|
1360 |
upsert=True
|
1361 |
)
|
1362 |
backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
|
@@ -1552,6 +1589,7 @@ async def chatbot_talk(client: Client, message: Message):
|
|
1552 |
return
|
1553 |
|
1554 |
test_result = ""
|
|
|
1555 |
backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
|
1556 |
backup_chat.append({"role": "user", "parts": [{"text": query_base}]})
|
1557 |
async for chunk in await gen.aio.models.generate_content_stream(
|
@@ -1581,15 +1619,23 @@ async def chatbot_talk(client: Client, message: Message):
|
|
1581 |
for part in text_parts:
|
1582 |
await message.reply_text(part)
|
1583 |
else:
|
1584 |
-
keyboard_like = create_keyboard(
|
|
|
|
|
|
|
1585 |
await message.reply_text(
|
1586 |
test_result,
|
1587 |
disable_web_page_preview=True,
|
1588 |
reply_markup=keyboard_like
|
1589 |
)
|
|
|
|
|
|
|
|
|
|
|
1590 |
await db.backup_chatbot.update_one(
|
1591 |
{"user_id": message.from_user.id},
|
1592 |
-
{"$
|
1593 |
upsert=True
|
1594 |
)
|
1595 |
backup_chat.append({"role": "model", "parts": [{"text": test_result}]})
|
|
|
1053 |
stream_image_now = re.match(r"--stream-images\s*(.*)", caption, re.IGNORECASE)
|
1054 |
if stream_image_now:
|
1055 |
STREAM_IMAGE_PATH = await message.download()
|
1056 |
+
uuid_code = str(uuid.uuid4())[:8]
|
1057 |
try:
|
1058 |
backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
|
1059 |
backup_chat.append({"role": "user", "parts": [{"text": caption}]})
|
|
|
1071 |
if captions == "":
|
1072 |
return await message.reply_text("Server busy try again later")
|
1073 |
|
1074 |
+
keyboard = create_keyboard(
|
1075 |
+
user_id=message.from_user.id,
|
1076 |
+
uuid_str=uuid_code
|
1077 |
+
)
|
1078 |
await message.reply_text(
|
1079 |
captions,
|
1080 |
disable_web_page_preview=True,
|
1081 |
reply_markup=keyboard
|
1082 |
)
|
1083 |
|
1084 |
+
translation_entry = {
|
1085 |
+
"uuid": uuid_code,
|
1086 |
+
"text": captions,
|
1087 |
+
"timestamp": dt.now().isoformat()
|
1088 |
+
}
|
1089 |
await db.backup_chatbot.update_one(
|
1090 |
{"user_id": message.from_user.id},
|
1091 |
+
{"$push": {"translate_history": translation_entry}},
|
1092 |
upsert=True
|
1093 |
)
|
1094 |
backup_chat.append({"role": "model", "parts": [{"text": captions}]})
|
|
|
1245 |
model=model_,
|
1246 |
contents=[str(caption), file_photo]
|
1247 |
)
|
1248 |
+
uuid_code = str(uuid.uuid4())[:8]
|
1249 |
if len(response.text) > 4096:
|
1250 |
text_parts = split_message(response.text)
|
1251 |
for part in text_parts:
|
1252 |
await message.reply_text(part)
|
1253 |
else:
|
1254 |
+
keyboard_like = create_keyboard(
|
1255 |
+
user_id=message.from_user.id,
|
1256 |
+
uuid_str=uuid_code
|
1257 |
+
)
|
1258 |
await message.reply_text(
|
1259 |
response.text,
|
1260 |
disable_web_page_preview=True,
|
1261 |
reply_markup=keyboard_like
|
1262 |
)
|
1263 |
+
translation_entry = {
|
1264 |
+
"uuid": uuid_code,
|
1265 |
+
"text": response.text,
|
1266 |
+
"timestamp": dt.now().isoformat()
|
1267 |
+
}
|
1268 |
await db.backup_chatbot.update_one(
|
1269 |
{"user_id": message.from_user.id},
|
1270 |
+
{"$push": {"translate_history": translation_entry}},
|
1271 |
upsert=True
|
1272 |
)
|
1273 |
backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
|
|
|
1308 |
model=model_,
|
1309 |
contents=[str(caption), video_file]
|
1310 |
)
|
1311 |
+
uuid_code = str(uuid.uuid4())[:8]
|
1312 |
if len(response.text) > 4096:
|
1313 |
text_parts = split_message(response.text)
|
1314 |
for part in text_parts:
|
1315 |
await message.reply_text(part)
|
1316 |
else:
|
1317 |
+
keyboard_like = create_keyboard(
|
1318 |
+
user_id=message.from_user.id,
|
1319 |
+
uuid_str=uuid_code
|
1320 |
+
)
|
1321 |
await message.reply_text(
|
1322 |
response.text,
|
1323 |
disable_web_page_preview=True,
|
1324 |
reply_markup=keyboard_like
|
1325 |
)
|
1326 |
+
translation_entry = {
|
1327 |
+
"uuid": uuid_code,
|
1328 |
+
"text": response.text,
|
1329 |
+
"timestamp": dt.now().isoformat()
|
1330 |
+
}
|
1331 |
await db.backup_chatbot.update_one(
|
1332 |
{"user_id": message.from_user.id},
|
1333 |
+
{"$push": {"translate_history": translation_entry}},
|
1334 |
upsert=True
|
1335 |
)
|
1336 |
backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
|
|
|
1370 |
model=model_,
|
1371 |
contents=["funny random meme sticker words", sticker_file]
|
1372 |
)
|
1373 |
+
uuid_code = str(uuid.uuid4())[:8]
|
1374 |
if len(response.text) > 4096:
|
1375 |
text_parts = split_message(response.text)
|
1376 |
for part in text_parts:
|
1377 |
await message.reply_text(part)
|
1378 |
else:
|
1379 |
+
keyboard_like = create_keyboard(
|
1380 |
+
user_id=message.from_user.id,
|
1381 |
+
uuid_str=uuid_code
|
1382 |
+
)
|
1383 |
await message.reply_text(
|
1384 |
response.text,
|
1385 |
disable_web_page_preview=True,
|
1386 |
reply_markup=keyboard_like
|
1387 |
)
|
1388 |
+
|
1389 |
+
translation_entry = {
|
1390 |
+
"uuid": uuid_code,
|
1391 |
+
"text": response.text,
|
1392 |
+
"timestamp": dt.now().isoformat()
|
1393 |
+
}
|
1394 |
await db.backup_chatbot.update_one(
|
1395 |
{"user_id": message.from_user.id},
|
1396 |
+
{"$push": {"translate_history": translation_entry}},
|
1397 |
upsert=True
|
1398 |
)
|
1399 |
backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
|
|
|
1589 |
return
|
1590 |
|
1591 |
test_result = ""
|
1592 |
+
uuid_code = str(uuid.uuid4())[:8]
|
1593 |
backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
|
1594 |
backup_chat.append({"role": "user", "parts": [{"text": query_base}]})
|
1595 |
async for chunk in await gen.aio.models.generate_content_stream(
|
|
|
1619 |
for part in text_parts:
|
1620 |
await message.reply_text(part)
|
1621 |
else:
|
1622 |
+
keyboard_like = create_keyboard(
|
1623 |
+
user_id=message.from_user.id,
|
1624 |
+
uuid_str=uuid_code
|
1625 |
+
)
|
1626 |
await message.reply_text(
|
1627 |
test_result,
|
1628 |
disable_web_page_preview=True,
|
1629 |
reply_markup=keyboard_like
|
1630 |
)
|
1631 |
+
translation_entry = {
|
1632 |
+
"uuid": uuid_code,
|
1633 |
+
"text": test_result,
|
1634 |
+
"timestamp": dt.now().isoformat()
|
1635 |
+
}
|
1636 |
await db.backup_chatbot.update_one(
|
1637 |
{"user_id": message.from_user.id},
|
1638 |
+
{"$push": {"translate_history": translation_entry}},
|
1639 |
upsert=True
|
1640 |
)
|
1641 |
backup_chat.append({"role": "model", "parts": [{"text": test_result}]})
|