Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -92,7 +92,6 @@ blacklist_patterns = [
     r"\bdeath\s*threat\b",
 ]
 
-
 def is_command_disabled(text):
   text_lower = text.lower()
   if text_lower.startswith("/") and not (text_lower.startswith("/setmodel") or text_lower.startswith("/start")):
@@ -1158,20 +1157,21 @@ async def chatbot_talk(client: Client, message: Message):
                         user_data.get("is_premium", False) and 
                         user_data.get("premium_expiry", dt.min) > dt.now()
                     )
+                    new_credits_used = user_data.get("credits_used", 0) + 1
+                    should_reset = (dt.now() - user_data.get("last_reset", dt.now())) > timedelta(days=1)
+                    update = {
+                        "$set": {
+                            "credits_used": 1 if should_reset else new_credits_used,
+                            "last_reset": dt.now() if should_reset else user_data["last_reset"],
+                            "is_premium": is_active_premium
+                        }
+                    }
+                    if "premium_expiry" not in user_data:
+                        update["$set"]["premium_expiry"] = None
+                        
                     await db.user_premium.update_one(
                         {"user_id": message.from_user.id},
-                        {
-                            "$inc": {"credits_used": 1},
-                            "$set": {
-                                "last_reset": user_data["last_reset"],
-                                "is_premium": is_active_premium
-                            },
-                            "$setOnInsert": {
-                                "credits_used": 0,
-                                "last_reset": dt.now(),
-                                "premium_expiry": None
-                            }
-                        },
+                        update,
                         upsert=True
                     )
                     await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)