randydev commited on
Commit
68206fd
·
verified ·
1 Parent(s): c4e6eb7

Ok updated

Browse files
Files changed (1) hide show
  1. 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(user_id=message.from_user.id)
 
 
 
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
- {"$set": {"translate_text": captions}},
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(user_id=message.from_user.id)
 
 
 
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
- {"$set": {"translate_text": response.text}},
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(user_id=message.from_user.id)
 
 
 
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
- {"$set": {"translate_text": response.text}},
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(user_id=message.from_user.id)
 
 
 
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
- {"$set": {"translate_text": response.text}},
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(user_id=message.from_user.id)
 
 
 
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
- {"$set": {"translate_text": test_result}},
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}]})