Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -64,6 +64,22 @@ async def geni_files_delete(name: str):
         return None
     return response.text
 
+async def remove_bg_myapi(return_image):
+    with open(return_image, "rb") as image_file:
+        encoded_string = base64.b64encode(image_file.read()).decode("utf-8")
+    payload = {
+        "image_base64": encoded_string
+    }
+    output_bg = "output.png"
+    response = requests.post("https://randydev-meta-ai.hf.space/remove-bg-base64", json=payload)
+    if response.status_code == 200:
+        result_base64 = response.json().get("output_image_base64")
+        with open(output_bg, "wb") as out_file:
+            out_file.write(base64.b64decode(result_base64))
+        return output_bg
+    else:
+        return None
+
 CREDITS_DEVELOPER = """
 🗓️ 2 May 2025
 
@@ -239,6 +255,30 @@ async def deletemydata(client, callback):
     else:
         await callback.answer(delm, True)
 
+@Client.on_callback_query(filters.regex("^removerbg_(\d+)$"))
+async def remover_bg(client, callback):
+    user_id = int(callback.matches[0].group(1))
+    try:
+        data = await db.backup_chatbot.find_one({"user_id": int(user_id)})
+        if not data:
+            return await callback.answer("Can't found user", True)
+        get_response = data.get("background_file_id", None)
+        if not get_response:
+            return await callback.answer("Server busy try again later", True)
+        media = await client.download_media(get_response)
+        output_bg = await remove_bg_myapi(media)
+        if not output_bg:
+            return await callback.answer("Server busy try again later", True)
+        await callback.edit_message_text("Please wait Downloading Background Remover.....")
+        await callback.message.edit_media(
+            media=InputMediaDocument(media=output_bg)
+        )
+        os.remove(output_bg)
+        await callback.answer()
+    except Exception as e:
+        LOGS.error(f"remover_bg failed: {str(e)}")
+        await callback.answer("Server error", show_alert=True)
+
 @Client.on_callback_query(filters.regex("^refreshch$"))
 async def reshch(client, callback):
     await callback.answer("Coming Soon", True)
@@ -443,6 +483,33 @@ async def chatbot_talk(client: Client, message: Message):
         try:
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
             await asyncio.sleep(1.5)
+
+            if re.findall(r"\b(.editimage)\b", query_base, re.IGNORECASE):
+                await db.backup_chatbot.update_one(
+                    {"user_id": message.from_user.id},
+                    {"$set": {"background_file_id": message.photo.file_id}},
+                    upsert=True
+                )
+                buttons = [
+                    [
+                        InlineKeyboardButton(
+                            text="✂️ Background Remover",
+                            callback_data=f"removerbg_{message.from_user.id}"
+                        )
+                    ],
+                    [
+                        InlineKeyboardButton(
+                            text="❌ Cancel",
+                            callback_data="closedd"
+                        )
+                    ]
+                ]
+                await message.reply_text(
+                    "Are you sure you want to image this edit image?",
+                    reply_markup=InlineKeyboardMarkup(buttons)
+                )
+                return
+    
             caption = message.caption or "What this?"
             backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
             backup_chat.append({"role": "user", "parts": [{"text": caption}]})