Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -92,6 +92,7 @@ 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")):
@@ -1153,17 +1154,19 @@ async def chatbot_talk(client: Client, message: Message):
                     )
                     backup_chat.append({"role": "model", "parts": [{"text": captions}]})
                     await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+                    is_active_premium = (
+                        user_data.get("is_premium", False) and 
+                        user_data.get("premium_expiry", dt.min) > dt.now()
+                    )
                     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": user_data.get("is_premium", False) 
-                                and (user_data.get("premium_expiry", dt.min) > dt.now())
+                                "is_premium": is_active_premium
                             },
                             "$setOnInsert": {
-                                "is_premium": False,
                                 "credits_used": 0,
                                 "last_reset": dt.now(),
                                 "premium_expiry": None