Ask for category during search

Author: hemantapkh

app/plugins/functions/keyboards.py

@@ -54,6 +54,33 @@ def language(self, welcome=False):
             self.client.misc.split_list(results, 2),
         )
 
+    def categories(self, query):
+        category_map = {
+            "movies": "🎬 Movies",
+            "tv": "📺 TV",
+            "games": "🎮 Games",
+            "music": "🎵 Music",
+            "apps": "🖥️ Apps",
+            "anime": "🎎 Anime",
+            "documentaries": "🎥 Documentaries",
+            "xxx": "📦 Other",
+        }
+
+        # Truncate query to the max safe length to avoid hitting Telegram's limit
+        truncated_query = query[:46]
+
+        buttons = [
+            types.InlineKeyboardButton(
+                text=display_text,
+                callback_data=f"cat_{category_name}_{truncated_query}",
+            )
+            for category_name, display_text in category_map.items()
+        ]
+
+        return types.InlineKeyboardMarkup(
+            self.client.misc.split_list(buttons, 2),
+        )
+
     def torrent_info(self, user_lang, hash, bookmarked=False):
         if bookmarked:
             results = [

app/plugins/groups/search.py

@@ -9,4 +9,4 @@ async def search(Client, message):
         text=Client.language.STR("queryToSearch", user_lang),
         reply_to_message_id=message.id,
         reply_markup=types.ForceReply(selective=True),
-    )
+    )

app/plugins/py1337x/search.py

@@ -11,19 +11,32 @@
 async def search(Client: Client, message):
     user_lang = await Client.misc.user_lang(message)
 
-    msg = await Client.send_message(
+    await Client.send_message(
         chat_id=message.chat.id,
-        text=Client.language.STR("searchingQuery", user_lang).format(message.text),
+        text=Client.language.STR("selectCategory", user_lang),
         reply_to_message_id=message.id,
+        reply_markup=Client.keyboard.categories(message.text),
     )
 
-    response = await Client.py1337x.search(message.text)
+
+@Client.on_callback_query(filters.regex(r"cat_"))
+async def search_category(Client, callback_query):
+    _, category, query = callback_query.data.split("_", 2)
+    user_lang = await Client.misc.user_lang(callback_query)
+
+    await Client.edit_message_text(
+        chat_id=callback_query.message.chat.id,
+        message_id=callback_query.message.id,
+        text=Client.language.STR("searchingQuery", user_lang).format(query),
+    )
+
+    response = await Client.py1337x.search(query, category=category)
 
     text = Client.struct.search_message(response.to_dict(), user_lang)
 
     await Client.edit_message_text(
-        chat_id=message.chat.id,
-        message_id=msg.id,
+        chat_id=callback_query.message.chat.id,
+        message_id=callback_query.message.id,
         text=text,
-        reply_markup=Client.keyboard.sites(message.text),
+        reply_markup=Client.keyboard.sites(query),
     )

app/torrenthunt.py

@@ -3,6 +3,10 @@
 import asyncio
 from os import environ
 from sys import argv
+from dotenv import load_dotenv
+
+# It's crucial to load the environment variables before other imports
+load_dotenv()  # noqa
 
 import sentry_sdk
 import uvloop