refactor: migrate modules to aiohttp for asynchronous requests and remove risearch module.

Author: AbhiTheModder

admin/banall.py

@@ -1,11 +1,11 @@
-from pyrogram import Client, filters, enums
+import asyncio
+
+from pyrogram import Client, enums, filters
 from pyrogram.errors.exceptions.flood_420 import FloodWait
 from pyrogram.types import *
-import asyncio
 from pyrogram.types import Message
+
 from utils import modules_help, prefix
-import re
-import os
 
 
 @Client.on_message(filters.command("banall", prefix) & filters.me)

admin/blacklist.py

@@ -1,8 +1,8 @@
-from pyrogram import Client, filters, enums
+from pyrogram import Client, enums, filters
 from pyrogram.types import Message
+from utils.scripts import format_exc
 
 from utils import modules_help, prefix
-from utils.scripts import format_exc
 
 
 @Client.on_message(filters.command(["block"], prefix) & filters.me)

admin/gmute.py

@@ -1,8 +1,9 @@
-from pyrogram import Client, filters, errors, enums
+from pyrogram import Client, enums, errors, filters
 from pyrogram.types import Message
-from utils import modules_help, prefix
 from utils.db import db
 
+from utils import modules_help, prefix
+
 
 def get_arg(message):
     msg = message.text

ai/aimage.py

@@ -3,11 +3,11 @@
 # YOu can check it out for uses example
 import os
 
-from pyrogram import Client, filters, enums
+from pyrogram import Client, enums, filters
 from pyrogram.types import Message
+from utils.scripts import format_exc, import_library
 
 from utils import modules_help, prefix
-from utils.scripts import format_exc, import_library
 
 genai = import_library("google.genai", "google-genai")
 
@@ -24,28 +24,32 @@
 
 MODEL_NAME = "gemini-2.5-flash"
 
+
 @Client.on_message(filters.command("getai", prefix) & filters.me)
 async def getai(_, message: Message):
     local_file_path = None
     uploaded_file = None
     try:
         await message.edit_text("<code>Please Wait...</code>")
-        
+
         try:
             if not message.reply_to_message or not message.reply_to_message.media:
-                return await message.edit_text("<code>Please reply to an image...</code>")
-            
+                return await message.edit_text(
+                    "<code>Please reply to an image...</code>"
+                )
+
             local_file_path = await message.reply_to_message.download()
         except Exception:
-            return await message.edit_text("<code>Please reply to an image or media that can be downloaded.</code>")
+            return await message.edit_text(
+                "<code>Please reply to an image or media that can be downloaded.</code>"
+            )
 
         uploaded_file = client.files.upload(file=local_file_path)
-        
+
         prompt = "Get details of given image, be as accurate as possible."
 
         response = client.models.generate_content(
-            model=MODEL_NAME,
-            contents=[prompt, uploaded_file]
+            model=MODEL_NAME, contents=[prompt, uploaded_file]
         )
 
         await message.edit_text(
@@ -68,34 +72,32 @@ async def aicook(_, message: Message):
     uploaded_file = None
     if not message.reply_to_message or not message.reply_to_message.media:
         return await message.edit_text("<code>Please reply to an image...</code>")
-    
+
     try:
         await message.edit_text("<code>Cooking...</code>")
 
         try:
             local_file_path = await message.reply_to_message.download()
         except Exception:
-            return await message.edit_text("<code>Please reply to an image or media that can be downloaded.</code>")
+            return await message.edit_text(
+                "<code>Please reply to an image or media that can be downloaded.</code>"
+            )
 
         uploaded_file = client.files.upload(file=local_file_path)
-        
+
         cook_contents = [
             "Accurately identify the baked good in the image and provide an appropriate and recipe consistent with your analysis. ",
             uploaded_file,
         ]
 
         response = client.models.generate_content(
-            model=MODEL_NAME,
-            contents=cook_contents,
-            config=generation_config_cook
+            model=MODEL_NAME, contents=cook_contents, config=generation_config_cook
         )
 
-        await message.edit_text(
-            f"{response.text}", parse_mode=enums.ParseMode.MARKDOWN
-        )
+        await message.edit_text(f"{response.text}", parse_mode=enums.ParseMode.MARKDOWN)
     except Exception as e:
         await message.edit_text(f"An error occurred: {format_exc(e)}")
-    
+
     finally:
         if uploaded_file:
             client.files.delete(name=uploaded_file.name)
@@ -108,11 +110,13 @@ async def aiseller(_, message: Message):
     local_file_path = None
     uploaded_file = None
     if not message.reply_to_message or not message.reply_to_message.media:
-        return await message.edit_text("<code>Please reply to a product image...</code>")
+        return await message.edit_text(
+            "<code>Please reply to a product image...</code>"
+        )
 
     try:
         await message.edit_text("<code>Generating...</code>")
-        
+
         if len(message.command) > 1:
             taud = message.text.split(maxsplit=1)[1]
         else:
@@ -123,10 +127,12 @@ async def aiseller(_, message: Message):
         try:
             local_file_path = await message.reply_to_message.download()
         except Exception:
-            return await message.edit_text("<code>Please reply to an image or media that can be downloaded.</code>")
+            return await message.edit_text(
+                "<code>Please reply to an image or media that can be downloaded.</code>"
+            )
 
         uploaded_file = client.files.upload(file=local_file_path)
-        
+
         sell_contents = [
             "Given an image of a product and its target audience, write an engaging marketing description",
             "Product Image: ",
@@ -140,18 +146,17 @@ async def aiseller(_, message: Message):
             contents=sell_contents,
         )
 
-        await message.edit_text(
-            f"{response.text}", parse_mode=enums.ParseMode.MARKDOWN
-        )
+        await message.edit_text(f"{response.text}", parse_mode=enums.ParseMode.MARKDOWN)
     except Exception as e:
         await message.edit_text(f"An error occurred: {format_exc(e)}")
-    
+
     finally:
         if uploaded_file:
             client.files.delete(name=uploaded_file.name)
         if local_file_path and os.path.exists(local_file_path):
             os.remove(local_file_path)
 
+
 modules_help["aimage"] = {
     "getai [reply to image]*": "Get details of image with Ai",
     "aicook [reply to image]*": "Generate Cooking instrunctions of the given food image",

ai/blackbox.py

@@ -1,8 +1,7 @@
-import uuid
 import re
+import uuid
 
 from aiohttp import ClientSession, FormData
-
 from pyrogram import Client, filters
 
 from utils import modules_help, prefix

ai/cdxl.py

@@ -1,10 +1,11 @@
-import base64, os
+import base64
+import os
 
 from pyrogram import Client, filters
 from pyrogram.types import Message
+from utils.scripts import format_exc, import_library
 
 from utils import modules_help, prefix
-from utils.scripts import format_exc, import_library
 
 clarifai = import_library("clarifai")
 

ai/chatbot.py

@@ -1,11 +1,11 @@
 from pyrogram import Client, enums, filters
 from pyrogram.types import Message
-
-from utils import modules_help, prefix
 from utils.config import cohere_key
 from utils.db import db
 from utils.scripts import format_exc, import_library, restart
 
+from utils import modules_help, prefix
+
 cohere = import_library("cohere")
 
 co = cohere.Client(cohere_key)

ai/cohere.py

@@ -1,23 +1,23 @@
 import asyncio
 from json import tool
-from utils.scripts import import_library
+
 from utils.config import cohere_key
+from utils.scripts import import_library
 
 cohere = import_library("cohere")
 
 import cohere
 
 co = cohere.Client(cohere_key)
 
-from utils import modules_help, prefix
-from utils.scripts import format_exc
+from pyrogram import Client, enums, filters
+from pyrogram.errors import MessageTooLong
+from pyrogram.types import Message
 from utils.db import db
 from utils.rentry import paste as rentry_paste
+from utils.scripts import format_exc
 
-
-from pyrogram import Client, filters, enums
-from pyrogram.types import Message
-from pyrogram.errors import MessageTooLong
+from utils import modules_help, prefix
 
 
 @Client.on_message(filters.command("cohere", prefix) & filters.me)

ai/copilot.py

@@ -1,12 +1,18 @@
 import aiohttp
 from pyrogram import Client, enums, filters
 from pyrogram.types import Message
+
 from utils import modules_help, prefix
 
 COPILOT_API_URL = "https://api.deline.web.id/ai/copilot?text="
 
+
 async def fetch_copilot_response(query: str, message: Message, is_self: bool):
-    response_msg = await (message.edit("<code>Thinking...</code>") if is_self else message.reply("<code>Thinking...</code>"))
+    response_msg = await (
+        message.edit("<code>Thinking...</code>")
+        if is_self
+        else message.reply("<code>Thinking...</code>")
+    )
     try:
         async with aiohttp.ClientSession() as session:
             async with session.get(f"{COPILOT_API_URL}{query.strip()}") as resp:
@@ -18,9 +24,14 @@ async def fetch_copilot_response(query: str, message: Message, is_self: bool):
             formatted_response = "Failed to fetch a response. Please try again."
         if len(formatted_response) > 4000:
             formatted_response = formatted_response[:4000] + "…"
-        await response_msg.edit_text(formatted_response, parse_mode=enums.ParseMode.MARKDOWN)
+        await response_msg.edit_text(
+            formatted_response, parse_mode=enums.ParseMode.MARKDOWN
+        )
     except Exception:
-        await response_msg.edit_text("An error occurred while connecting to the API. Please try again later.")
+        await response_msg.edit_text(
+            "An error occurred while connecting to the API. Please try again later."
+        )
+
 
 @Client.on_message(filters.command("copilot", prefix))
 async def copilot_command(client: Client, message: Message):
@@ -29,10 +40,13 @@ async def copilot_command(client: Client, message: Message):
     elif len(message.command) > 1:
         query = " ".join(message.command[1:])
     else:
-        await message.reply(f"<b>Usage:</b> <code>{prefix}copilot [prompt]</code> or reply to a message with <code>{prefix}copilot</code>")
+        await message.reply(
+            f"<b>Usage:</b> <code>{prefix}copilot [prompt]</code> or reply to a message with <code>{prefix}copilot</code>"
+        )
         return
     await fetch_copilot_response(query, message, message.from_user.is_self)
 
+
 modules_help["copilot"] = {
     "copilot [query]*": "Ask anything to Copilot AI",
 }

ai/flux.py

@@ -1,22 +1,24 @@
-import os
 import io
+import os
 import time
-import requests
-from pyrogram import filters, Client
+
+import aiohttp
+from pyrogram import Client, filters
 from pyrogram.types import Message
+from utils.scripts import format_exc, progress
 
 from utils import modules_help, prefix
-from utils.scripts import format_exc, progress
 
 
-def schellwithflux(args):
+async def schellwithflux(args):
     API_URL = "https://randydev-ryuzaki-api.hf.space/api/v1/akeno/fluxai"
     payload = {"user_id": 1191668125, "args": args}  # Please don't edit here
-    response = requests.post(API_URL, json=payload)
-    if response.status_code != 200:
-        print(f"Error status {response.status_code}")
-        return None
-    return response.content
+    async with aiohttp.ClientSession() as session:
+        async with session.post(API_URL, json=payload) as resp:
+            if resp.status != 200:
+                print(f"Error status {resp.status}")
+                return None
+            return await resp.read()
 
 
 @Client.on_message(filters.command("fluxai", prefix) & filters.me)
@@ -25,7 +27,7 @@ async def imgfluxai_(client: Client, message: Message):
     if not question:
         return await message.reply_text("Please provide a question for Flux.")
     try:
-        image_bytes = schellwithflux(question)
+        image_bytes = await schellwithflux(question)
         if image_bytes is None:
             return await message.reply_text("Failed to generate an image.")
         pro = await message.reply_text("Generating image, please wait...")