randydev commited on
Commit
1e5f0d2
·
verified ·
1 Parent(s): 06d7db6
Files changed (1) hide show
  1. chatbot/plugins/chat.py +4 -0
chatbot/plugins/chat.py CHANGED
@@ -379,6 +379,10 @@ async def modelgeminimenu(client, callback):
379
  async def gemini_setmodel(client, callback):
380
  model, uuidstr = callback.matches[0].groups()
381
  user_id = callback.from_user.id
 
 
 
 
382
  await users_collection.update_one(
383
  {"user_id": user_id},
384
  {"$set": {"model": model}},
 
379
  async def gemini_setmodel(client, callback):
380
  model, uuidstr = callback.matches[0].groups()
381
  user_id = callback.from_user.id
382
+ user_bl = await db.user_blacklists.find_one({"user_id": user_id})
383
+ if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
384
+ await callback.answer("Sorry your account is frozen", True)
385
+ return
386
  await users_collection.update_one(
387
  {"user_id": user_id},
388
  {"$set": {"model": model}},