Update search.py

Mikobot/plugins/search.py

@@ -10,7 +10,9 @@ from pyrogram.types import InputMediaPhoto, Message
 
 from Mikobot import app
 from Mikobot.state import state
+import logging
 
+logging.basicConfig(level=logging.DEBUG)
 # <======================================== CONFIGURATION =========================================================>
 # Centralized configuration for API settings
 API_CONFIG = {
@@ -132,32 +134,47 @@ async def bingimg_search(client: Client, message: Message):
 
 
 @app.on_message(filters.command("googleimg"))
-async def googleimg_search(client: Client, message: Message):
+async def googleimg_search(_, message: Message):
+    logging.debug("Google Image Search command invoked")
     try:
-        text = message.text.split(None, 1)[1]
-    except IndexError:
-        return await message.reply_text("Provide me a query to search!")
+        query = message.text.split(None, 1)[1].strip() if len(message.text.split()) > 1 else None
+        if not query:
+            logging.debug("No query provided")
+            return await message.reply_text("Please provide a search query!")
 
-    search_message = await message.reply_text("💭")
+        logging.debug(f"Query: {query}")
+        search_message = await message.reply_text("💭 Searching for images...")
 
-    api_url = API_CONFIG["google_image_api"]["url"].format(
-        query=text, api_key=API_CONFIG["google_image_api"]["api_key"]
-    )
-    response = await state.get(api_url)
+        api_url = (
+            f"https://www.googleapis.com/customsearch/v1"
+            f"?q={query}&searchType=image&cx=YOUR_CSE_ID&key=YOUR_API_KEY&num=7"
+        )
+        logging.debug(f"API URL: {api_url}")
 
-    if response.status_code == 200:
-        images = response.json().get("items", [])
-        if not images:
-            await message.reply_text("No images found.")
-            return
+        async with aiohttp.ClientSession() as session:
+            async with session.get(api_url) as response:
+                logging.debug(f"API Response Status: {response.status}")
+                if response.status != 200:
+                    return await message.reply_text(f"Error: {response.status}")
 
-        media = [InputMediaPhoto(media=img["link"]) for img in images[:7]]
-        await message.reply_media_group(media=media)
-    else:
-        await message.reply_text("An error occurred while searching for images.")
+                data = await response.json()
+                logging.debug(f"API Response: {data}")
 
-    await search_message.delete()
-    await message.delete()
+                items = data.get("items", [])
+                if not items:
+                    return await message.reply_text("No images found.")
+
+                media = [InputMediaPhoto(item["link"]) for item in items[:7]]
+                await message.reply_media_group(media=media)
+
+        await search_message.delete()
+
+    except IndexError:
+        logging.exception("IndexError: No query provided")
+        await message.reply_text("Provide me a query to search!")
+    except Exception as e:
+        logging.exception(f"Exception occurred: {str(e)}")
+        await message.reply_text(f"An error occurred: {str(e)}")
 
 # <=======================================================================================================>