Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -48,6 +48,28 @@ async def geni_files_delete(name: str):
         return None
     return response.text
 
+
+@Client.on_message(
+    ~filters.scheduled
+    & command(["onchat"])
+    & filters.me
+    & ~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")
+
+
+@Client.on_message(
+    ~filters.scheduled
+    & command(["offchat"])
+    & filters.me
+    & ~filters.forwarded
+)
+async def rmchatbot_user(client: Client, message: Message):
+    await db.remove_chatbot(message.chat.id)
+    await message.reply_text("ok stopped gemini")
+
 @Client.on_message(
     filters.incoming
     & (
@@ -58,7 +80,8 @@ async def geni_files_delete(name: str):
         | filters.voice
         | filters.regex(r"\b(Randy|Rendi)\b(.*)", flags=re.IGNORECASE)
     )
-    & filters.private
+    & (filters.private | filters.group)
+    & filters.reply
     & ~filters.bot
     & ~filters.via_bot
     & ~filters.forwarded,
@@ -66,6 +89,9 @@ async def geni_files_delete(name: str):
 )
 async def chatbot_talk(client: Client, message: Message):
     genai.configure(api_key=GOOGLE_API_KEY)
+    chat_user = await db.get_chatbot(message.chat.id)
+        if not chat_user:
+            return
     if message.photo:
         await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
         await asyncio.sleep(1.5)