Update database.py

database.py

@@ -30,6 +30,7 @@ class Database:
         self.gachabots = self.db["gachabots"]
         self.cohere = self.db["cohere"]
         self.chatbot = self.db["chatbot"]
+        self.gemini_plan = self.db["gemini_plan"]
         self.backup_chatbot = self.db["backupchatbot"]
         self.antiarabic = self.db["antiarabic"]
         self.antinsfw = self.db["antinsfw"]
@@ -639,6 +640,10 @@ 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 gemini_plan(self, chat_id):
+        user_data = await self.gemini_plan.find_one({"user_id": user_id})
+        return user_data.get("is_plan") if user_data else None
+
     async def remove_chatbot(self, chat_id):
         unset_data = {"user_id": None}
         return await self.chatbot.update_one({"chat_id": chat_id}, {"$unset": unset_data})