Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -44,14 +44,6 @@ import akenoai.logger as akeno_log
 
 from . import force_sub
 
-@Client.on_message(
-    ~filters.scheduled
-    & filters.command(["onchat"])
-    & ~filters.forwarded
-)
-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")
 
 BASE_PROMPT = f"""
 You are my name Akeno AI and python language powered by @xtdevs on telegram support and language models GPT-5-ULTRA
@@ -111,12 +103,80 @@ async def startbot(client: Client, message: Message):
 
 @Client.on_message(
     ~filters.scheduled
-    & filters.command(["offchat"])
+    & filters.command(["onchat"])
     & ~filters.forwarded
 )
-async def rmchatbot_user(client: Client, message: Message):
-    await db.remove_chatbot(message.chat.id)
-    await message.reply_text("ok stopped GPT")
+async def okon_(client: Client, message: Message):
+    buttons = [
+        [
+            InlineKeyboardButton(
+                text="Offline",
+                callback_data=f"off_{message.from_user.id}_{message.chat.id}"
+            ),
+            InlineKeyboardButton(
+                text="Online",
+                callback_data=f"on_{message.from_user.id}_{message.chat.id}"
+            ),
+        ],
+    ]
+    await message.reply_text(
+        "I want to go offline",
+        reply_markup=InlineKeyboardMarkup(buttons)
+    )
+
+@Client.on_callback_query(filters.regex("^off_"))
+async def off_mode(client, cb):
+    data = cb.data.split("_")
+    if len(data) != 3:
+        await cb.answer("❌ Format data tidak valid.", show_alert=True)
+        return
+    _, user_id_str, chat_id_str = data
+    try:
+        user_id = int(user_id_str)
+    except ValueError:
+        await cb.answer("❌ Invalid user ID.", show_alert=True)
+        return
+    try:
+        chat_id = int(chat_id_str)
+    except ValueError:
+        await cb.answer("❌ Invalid chat ID.", show_alert=True)
+        return
+    if cb.from_user.id != user_id:
+        await cb.answer("❌ You have no right to do this.", show_alert=True)
+        return
+    try:
+        await db.remove_chatbot(chat_id)
+        await cb.edit_message_text("ok stopped GPT")
+        await cb.answer()
+    except Exception as e:
+        await cb.answer(f"❌ Error: {e}", show_alert=True)
+
+@Client.on_callback_query(filters.regex("^on_"))
+async def on_mode(client: Client, cb: CallbackQuery):
+    data = cb.data.split("_")
+    if len(data) != 3:
+        await cb.answer("❌ Format data tidak valid.", show_alert=True)
+        return
+    _, user_id_str, chat_id_str = data
+    try:
+        user_id = int(user_id_str)
+    except ValueError:
+        await cb.answer("❌ Invalid user ID.", show_alert=True)
+        return
+    try:
+        chat_id = int(chat_id_str)
+    except ValueError:
+        await cb.answer("❌ Invalid chat ID.", show_alert=True)
+        return
+    if cb.from_user.id != user_id:
+        await cb.answer("❌ You have no right to do this.", show_alert=True)
+        return
+    try:
+        await db.add_chatbot(chat_id, client.me.id)
+        await cb.edit_message_text("Added chatbot user")
+        await cb.answer()
+    except Exception as e:
+        await cb.answer(f"❌ Error: {e}", show_alert=True)
 
 @Client.on_message(
     ~filters.scheduled