Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -38,6 +38,7 @@ from database import db
 from logger import LOGS
 
 import google.generativeai as genai
+import akeno.logger as akeno
 from google.api_core.exceptions import InvalidArgument
 
 async def geni_files_delete(name: str):
@@ -48,6 +49,12 @@ async def geni_files_delete(name: str):
         return None
     return response.text
 
+DISABLE_COMMAND = [
+    "start",
+    "status",
+    "offchat",
+    "onchat"
+]
 
 FREE_GEMINI_TEXT = """
 • User Free : {name}
@@ -65,6 +72,7 @@ FREE_GEMINI_TEXT = """
     & filters.command(["status"])
     & ~filters.forwarded
 )
+@akeno.LogChannel(channel_id=-1002290885889, is_track=True)
 async def userstatus(client: Client, message: Message):
     is_check_plan = await db.is_gemini_plan(user_id=message.from_user.id)
     chat_user = await db.get_chatbot(message.chat.id)
@@ -101,6 +109,7 @@ async def test_payment(client: Client, message: Message):
     & filters.command(["onchat"])
     & ~filters.forwarded
 )
+@akeno.LogChannel(channel_id=-1002290885889, is_track=True)
 async def addchatbot_user(client: Client, message: Message):
     await db.add_chatbot(message.chat.id, client.me.id)
     await message.reply_text("Added chatbot user")
@@ -111,6 +120,7 @@ async def addchatbot_user(client: Client, message: Message):
     & filters.command(["offchat"])
     & ~filters.forwarded
 )
+@akeno.LogChannel(channel_id=-1002290885889, is_track=True)
 async def rmchatbot_user(client: Client, message: Message):
     await db.remove_chatbot(message.chat.id)
     await message.reply_text("ok stopped gemini")
@@ -127,11 +137,13 @@ async def rmchatbot_user(client: Client, message: Message):
         | filters.regex(r"\b(Randy|Rendi)\b(.*)", flags=re.IGNORECASE)
     )
     & (filters.private | filters.group)
+    & ~filters.command(DISABLE_COMMAND)
     & ~filters.bot
     & ~filters.via_bot
     & ~filters.forwarded,
     group=2,
 )
+@akeno.LogChannel(channel_id=-1002290885889, is_track=True)
 async def chatbot_talk(client: Client, message: Message):
     genai.configure(api_key=GOOGLE_API_KEY)
     chat_user = await db.get_chatbot(message.chat.id)