Aleksey Khlopkov commited on
Commit
0efd01c
·
1 Parent(s): 8730736
Files changed (1) hide show
  1. main.py +10 -10
main.py CHANGED
@@ -24,8 +24,8 @@ bezumtsi_gif_id = "CgACAgIAAxkBAAE4zMtojLmiH_CGW5cT7G0QVXHR7D4g6wAC53UAApkBmEmM-
24
 
25
  last_maxim_insult = 1.0
26
  last_gif_sent = 1.0
27
- maxim_insult_cooldown = 5.0
28
- gif_sent_cooldown = 5.0
29
 
30
  torch.manual_seed(0)
31
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -42,20 +42,21 @@ def handle_response(text: str) -> Optional[str]:
42
  return None
43
 
44
 
45
- def edit_response(text: str) -> str:
 
 
46
  text = re.sub(r'\s+([,.!?;])\s+', r'\1 ', text)
47
-
48
  return text
49
 
50
 
51
  async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
52
  if update.message.chat_id == CHAT_ID:
53
- response: Optional[str] = ""
54
  global last_maxim_insult, last_gif_sent
55
  if update.message.from_user.username == "WhoReadThisWillDie" and \
56
  time.time() - last_maxim_insult >= maxim_insult_cooldown:
57
- response = "Максим, иди нахуй"
58
  last_maxim_insult = time.time()
 
59
  elif "роман" in update.message.text.lower() and \
60
  time.time() - last_gif_sent >= gif_sent_cooldown:
61
  await context.bot.send_animation( chat_id=update.message.chat_id, animation=romantiki_gif_id)
@@ -67,9 +68,8 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
67
  else:
68
  text = update.message.text.replace(BOT_USERNAME, '').strip().lower()
69
  response = edit_response(handle_response(text))
70
-
71
- if response != "":
72
- await context.bot.sendMessage(update.message.chat_id, response, reply_to_message_id=update.message.id)
73
 
74
 
75
  async def error(update: Update, context: ContextTypes.DEFAULT_TYPE):
@@ -85,7 +85,7 @@ def main() -> None:
85
  application = Application.builder().token(TOKEN).build()
86
 
87
  application.add_handler(MessageHandler(filters.TEXT, handle_message))
88
- application.add_error_handler(error)
89
 
90
  application.run_polling(allowed_updates=Update.ALL_TYPES)
91
 
 
24
 
25
  last_maxim_insult = 1.0
26
  last_gif_sent = 1.0
27
+ maxim_insult_cooldown = 180.0
28
+ gif_sent_cooldown = 180.0
29
 
30
  torch.manual_seed(0)
31
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
42
  return None
43
 
44
 
45
+ def edit_response(text: Optional[str]) -> Optional[str]:
46
+ if text is None:
47
+ return None
48
  text = re.sub(r'\s+([,.!?;])\s+', r'\1 ', text)
 
49
  return text
50
 
51
 
52
  async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
53
  if update.message.chat_id == CHAT_ID:
54
+ # response: Optional[str] = ""
55
  global last_maxim_insult, last_gif_sent
56
  if update.message.from_user.username == "WhoReadThisWillDie" and \
57
  time.time() - last_maxim_insult >= maxim_insult_cooldown:
 
58
  last_maxim_insult = time.time()
59
+ await context.bot.sendMessage(update.message.chat_id, "Максим, иди нахуй", reply_to_message_id=update.message.id)
60
  elif "роман" in update.message.text.lower() and \
61
  time.time() - last_gif_sent >= gif_sent_cooldown:
62
  await context.bot.send_animation( chat_id=update.message.chat_id, animation=romantiki_gif_id)
 
68
  else:
69
  text = update.message.text.replace(BOT_USERNAME, '').strip().lower()
70
  response = edit_response(handle_response(text))
71
+ if response:
72
+ await context.bot.sendMessage(update.message.chat_id, response, reply_to_message_id=update.message.id)
 
73
 
74
 
75
  async def error(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
85
  application = Application.builder().token(TOKEN).build()
86
 
87
  application.add_handler(MessageHandler(filters.TEXT, handle_message))
88
+ # application.add_error_handler(error)
89
 
90
  application.run_polling(allowed_updates=Update.ALL_TYPES)
91