Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -1093,7 +1093,7 @@ async def chatbot_talk(client: Client, message: Message):
                         "credits_used": 0,
                         "last_reset": dt.now(),
                         "is_premium": False,
-                        "premium_expiry" = None
+                        "premium_expiry": None
                     }
                     if dt.now() - user_data["last_reset"] > timedelta(days=1):
                         await db.user_premium.update_one(
@@ -1157,8 +1157,17 @@ async def chatbot_talk(client: Client, message: Message):
                         {"user_id": message.from_user.id},
                         {
                             "$inc": {"credits_used": 1},
-                            "$set": {"last_reset": user_data["last_reset"]},
-                            "$setOnInsert": {"is_premium": False}
+                            "$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())
+                            },
+                            "$setOnInsert": {
+                                "is_premium": False,
+                                "credits_used": 0,
+                                "last_reset": dt.now(),
+                                "premium_expiry": None
+                            }
                         },
                         upsert=True
                     )