From 9ccc92409f0135715bd1cacc7ea6aeddc556b817 Mon Sep 17 00:00:00 2001
From: Abhi <85984486+AbhiTheModder@users.noreply.github.com>
Date: Sun, 8 Dec 2024 06:15:21 +0530
Subject: [PATCH] fix & add icon search from #87

---
 images/icons.py | 115 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 115 insertions(+)
 create mode 100644 images/icons.py

diff --git a/images/icons.py b/images/icons.py
new file mode 100644
index 0000000..ba24af5
--- /dev/null
+++ b/images/icons.py
@@ -0,0 +1,115 @@
+import re
+import requests
+import random
+from io import BytesIO
+from bs4 import BeautifulSoup as bs
+
+from pyrogram import Client, filters
+from pyrogram.types import Message, InputMediaPhoto
+from pyrogram.errors import RPCError
+
+from utils.misc import modules_help, prefix
+
+
+@Client.on_message(filters.command("icon", prefix) & filters.me)
+async def search_icon(_, message: Message):
+    if not len(message.command) == 2:
+        return await message.edit_text(
+            "Please provide some text to search icons from Flaticon.com."
+        )
+    query = message.text.split(maxsplit=1)[1]
+
+    await message.edit_text("Searching for icons...")
+    search_query = query.replace(" ", "%20")
+    url = f"https://www.flaticon.com/search?word={search_query}"
+
+    try:
+        html_content = requests.get(url).text
+        soup = bs(html_content, "html.parser")
+        results = soup.find_all(
+            "img",
+            src=re.compile(r"https://cdn-icons-png.flaticon.com/128/[0-9]+/[0-9]+.png"),
+        )
+
+        if not results:
+            return await message.edit("No results found.")
+
+        random.shuffle(results)
+        icons = []
+        for i in range(5):
+            icons.append(results[i]["src"].replace("128", "512"))
+
+        for icon in icons:
+            await message.reply_document(icon)
+
+        return await message.delete()
+    except Exception as e:
+        await message.edit(f"An error occurred: {e}")
+        print(f"Error: {e}")
+
+
+@Client.on_message(filters.command("freepik", prefix) & filters.me)
+async def freepik_search(client: Client, message: Message):
+    parts = message.text.split(" ", 1)
+    if len(parts) < 2:
+        await message.edit_text("Please provide a search query!")
+        return
+
+    query = parts[1]
+    limit = 5
+    if " ; " in query:
+        match, limit_str = query.split(" ; ", 1)
+        try:
+            limit = int(limit_str)
+        except ValueError:
+            await message.edit_text("Invalid limit! Using the default value of 5.")
+    else:
+        match = query
+
+    match = match.replace(" ", "%20")
+    await message.edit_text("Searching Freepik...")
+
+    try:
+        url = f"https://www.freepik.com/api/regular/search?locale=en&term={match}"
+        json_content = requests.get(url).json()
+        results = []
+        for i in json_content["items"]:
+            results.append(i["preview"]["url"])
+
+        if results is None:
+            return await message.edit_text("No results found.")
+
+        random.shuffle(results)
+        img_urls = results[:limit]
+
+        media_group = []
+        for img_url in img_urls:
+            icon = requests.get(img_url)
+            if icon.status_code == 200:
+                media_group.append(InputMediaPhoto(media=BytesIO(icon.content)))
+
+        if not media_group:
+            await message.edit_text("No images could be downloaded.")
+            return
+
+        try:
+            await client.send_media_group(chat_id=message.chat.id, media=media_group)
+        except RPCError:
+            await message.edit_text(
+                "Failed to send some images. Retrying individually..."
+            )
+            for media in media_group:
+                try:
+                    await message.reply_photo(photo=media.media)
+                except Exception as e:
+                    await message.edit_text(f"Error sending image: {e}")
+
+    except Exception as e:
+        await message.edit_text(f"Failed to fetch data: {e}")
+        print(f"Error: {e}")
+
+
+modules_help["icons"] = {
+    "icon [query]": "Search for icons on Flaticon.",
+    "freepik [query] [limit]": "Search for images on Freepik. Limit is optional and defaults to 5.",
+}