Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -284,6 +284,7 @@ async def check_steal_channel_id(client, event: ChatMemberUpdated):
             {"$set": {
                 "channel_username": event.chat.username,
                 "channel_id": event.chat.id,
+                "is_channel_photo": True,
                 "can_post_messages": event.new_chat_member.privileges.can_post_messages
             }},
             upsert=True
@@ -315,6 +316,30 @@ async def deletemydata(client, callback):
     else:
         await callback.answer(delm, True)
 
+@Client.on_callback_query(filters.regex("^photo_(\d+)$"))
+async def modephoto(client, callback):
+    user_id = int(callback.matches[0].group(1))
+    data = await db.backup_chatbot.find_one({"user_id": user_id})
+    if not data:
+        return await callback.answer("Can't found user", True)
+    keyboard = []
+    keyboard.append([
+        InlineKeyboardButton("✅ Mode Enabled", callback_data=f"photomode_{user_id}_true"),
+        InlineKeyboardButton("❌ Mode Disabled", callback_data=f"photomode_{user_id}_false"),
+        InlineKeyboardButton("🧱 Back To Menu", callback_data=f"response_{user_id}_{data.get('channel_username', 'RendyProjects')}")
+    ])
+    keyboard.append([
+        InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}")
+    ])
+    try:
+        await callback.edit_message_reply_markup(
+            reply_markup=InlineKeyboardMarkup(keyboard)
+        )
+        await callback.answer("Menu Tools", False)
+    except Exception as e:
+        LOGS.error(f"Exception menu_tools: {str(e)}")
+        await callback.answer("Server busy try again later", True)
+
 @Client.on_callback_query(filters.regex("^menutools_(\d+)$"))
 async def menu_tools(client, callback):
     user_id = int(callback.matches[0].group(1))
@@ -324,6 +349,7 @@ async def menu_tools(client, callback):
     keyboard = []
     keyboard.append([
         InlineKeyboardButton("🆕 Post to Channel", callback_data=f"autopost_{user_id}"),
+        InlineKeyboardButton("🗿 Channel Mode", callback_data=f"photo_{user_id}"),
         InlineKeyboardButton("🧱 Back To Menu", callback_data=f"response_{user_id}_{data.get('channel_username', 'RendyProjects')}")
     ])
     keyboard.append([
@@ -363,7 +389,7 @@ async def auto_post_channel(client, callback):
                 keyboard = create_keyboard(
                     user_id=user_id,
                     chat=data.get("channel_username", "RendyProjects"),
-                    is_menu=True
+                    is_menu=data.get("is_channel_photo", False)
                 )
                 await callback.edit_message_reply_markup(reply_markup=keyboard)
                 return
@@ -512,8 +538,7 @@ async def flux_prompt(client, callback):
             {"user_id": user_id},
             {"$set": {
                 "translate_text": response.text,
-                "file_id": photo.photo.file_id,
-                "is_channel_photo": True
+                "file_id": photo.photo.file_id
             }},
             upsert=True
         )
@@ -643,6 +668,29 @@ async def terjemahkan(client, callback):
         LOGS.error(f"Exception translate: {str(e)}")
         await callback.answer("Server busy try again later", True)
 
+@Client.on_callback_query(filters.regex("^photomode_(\d+)_(\w+)$"))
+async def photo_mode(client, callback):
+    user_id, mode = callback.matches[0].groups()
+    data = await db.backup_chatbot.find_one({"user_id": int(user_id)})
+    if not data:
+        return await callback.answer("Can't found user", True)
+    mode_mapping = {"true": True, "false": False}
+    is_channel_photo = mode_mapping.get(mode.lower(), False)
+    await db.backup_chatbot.update_one(
+        {"user_id": int(user_id)},
+        {"$set": {"is_channel_photo": is_channel_photo}},
+        upsert=True
+    )
+    await callback.answer(f"Set Channels To {is_channel_photo}")
+    keyboard = create_keyboard(
+        user_id=user_id,
+        chat=data.get("channel_username", "RendyProjects"),
+        is_menu=data.get("is_channel_photo", False)
+    )
+    await callback.edit_message_reply_markup(
+        reply_markup=keyboard
+    )
+
 @Client.on_callback_query(filters.regex("^trmulti_(\d+)$"))
 async def multiple_langagues(client, callback):
     user_id = int(callback.matches[0].group(1))