Update database.py

database.py

@@ -640,6 +640,13 @@ class Database:
         user_data = await self.chatbot.find_one({"chat_id": chat_id})
         return user_data.get("user_id") if user_data else None
 
+    async def add_gemini_plan(self, user_id):
+        await self.gemini_plan.update_one(
+            {"user_id": user_id},
+            {"$set": {"is_plan": True}},
+            upsert=True
+        )
+
     async def is_gemini_plan(self, user_id):
         user_data = await self.gemini_plan.find_one({"user_id": user_id})
         return user_data.get("is_plan", False) if user_data else None