Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -282,8 +282,28 @@ async def deletemydata(client, callback):
     else:
         await callback.answer(delm, True)
 
+@Client.on_callback_query(filters.regex("^menutools_(\d+)$"))
+async def menu_tools(client, callback):
+    user_id = int(callback.matches[0].group(1))
+    keyboard = []
+    keyboard.append([
+        InlineKeyboardButton("🆕 Post to Channel", callback_data=f"autopost_{user_id}"),
+        InlineKeyboardButton("🔁 Refresh", callback_data="refreshch")
+    ])
+    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("^autopost_(\d+)$"))
-async def auto_post(client, callback):
+async def auto_post_channel(client, callback):
     user_id = int(callback.matches[0].group(1))
     try:
         backup_chat = await db._get_chatbot_chat_from_db(user_id)
@@ -294,13 +314,20 @@ async def auto_post(client, callback):
         if data.get("is_channel_photo", False):
             if not file_id:
                 return await callback.answer("Can't found file_id", True)
+            who_post = data.get("channel_id", None)
+            if not who_post:
+                return await callback.answer("Can't found channel id", True)
+            await callback.answer("Ok process", False)
             await client.send_photo(
-                data.get("channel_id"),
+                who_post,
                 photo=file_id,
-                caption=caption
+                caption=data.get("translate_text", "")
             )
+            await callback.answer("Posted to Your Channel Check Now", True)
+        return await callback.answer("Need Channel as admin", True)
     except Exception as e:
         LOGS.error(f"auto_post failed: {str(e)}")
+        await callback.answer("Server busy try again later", True)
 
 @Client.on_callback_query(filters.regex("^genprompt_(\d+)$"))
 async def geminigen_prompt(client, callback):
@@ -421,10 +448,14 @@ async def flux_prompt(client, callback):
             ),
             reply_markup=keyboard
         )
-        # photo = await client.send_photo("@xpushz", file_path)
+        photo = await client.send_photo("LicenseAknBotDB", file_path)
         await db.backup_chatbot.update_one(
             {"user_id": user_id},
-            {"$set": {"translate_text": response.text}},
+            {"$set": {
+                "translate_text": response.text,
+                "file_id": photo.file_id,
+                "is_channel_photo": True
+            }},
             upsert=True
         )
         backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
@@ -596,7 +627,7 @@ def create_keyboard(likes: int = 0, unlikes: int = 0, user_id: int = None):
                 InlineKeyboardButton(f"👍 {likes}", callback_data="like"),
                 InlineKeyboardButton(f"👎 {unlikes}", callback_data="unlike"),
                 InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{user_id}"),
-                InlineKeyboardButton(f"🔁", callback_data="refreshch"),
+                InlineKeyboardButton(f"🗃️", callback_data=f"autopost_{user_id}"),
                 InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}"),
             ]
         ]