Try remover

akn/Gemini/gemini.py

@@ -24,6 +24,7 @@ import asyncio
 import io
 import os
 import re
+import base64
 from PIL import Image
 from pyrogram import *
 from pyrogram import enums
@@ -46,6 +47,21 @@ from akenoai.types import DifferentAPIDefault
 
 js = AkenoXJs(DifferentAPIDefault()).connect()
 
+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
+    }
+    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.png", "wb") as out_file:
+            out_file.write(base64.b64decode(result_base64))
+        return return_image
+    else:
+        return None
+    
 async def geni_files_delete(name: str):
     url = f"https://generativelanguage.googleapis.com/v1beta/{name}"
     params = {"key": GOOGLE_API_KEY}
@@ -148,11 +164,29 @@ async def chatbot_talk(client: Client, message: Message):
         file_path = await message.download()
         caption = message.caption or "What's this?"
         x = GeminiLatest(api_keys=GOOGLE_API_KEY)
-        if client.me.is_premium:
-            ai_reply = await message.reply_text(f"{custom_loading}Processing...")
-        else:
-            ai_reply = await message.reply_text(f"Processing...")
         try:
+            new_check_bg = re.findall(r"\bremover\b", caption)
+            if caption == "remover":
+                return await message.reply_text("i don't have, what do you mean by create remover?")
+            if new_check_bg:
+                new_js = None
+                new_js = await message.reply_text("Please wait, it's still being processed")
+                response_path = await remove_bg_myapi(file_path)
+                if not response_path:
+                    return await message.reply_text("Error try remover bg")
+                await new_js.edit_text("Uploading image completed...")
+                await message.reply_photo(
+                    response_path,
+                    progress=progress,
+                    progress_args=(
+                        new_js,
+                        time.time(),
+                        "Uploading image completed..."
+                    )
+                )
+                await new_js.delete()
+                return
+            ai_reply = await message.reply_text(f"Processing...")
             await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
             await asyncio.sleep(1.5)
             backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)