add booru image search

bot.py

@@ -1,34 +1,34 @@
-
-from telegram import Update, InlineKeyboardMarkup, InlineKeyboardButton
+from telegram import Update
 from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler, MessageHandler, filters, InlineQueryHandler
 import logging
-from render import render_text_on_image, image_to_string
+
+from tgbot.config import TOKEN
+from tgbot.shit.handlers import handle_xitter, handle_red_ebalo, handle_hentai
+from tgbot.shit.render import render_text_on_image, image_to_string
 from io import BytesIO
 
-import os
-
-
 logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                     level=logging.INFO)
 
 logger = logging.getLogger(__name__)
 
-TOKEN = os.getenv("TOKEN")
-
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await context.bot.send_message(chat_id=update.effective_chat.id, text='Hi! I respond with a picture to any message.')
+    await context.bot.send_message(chat_id=update.effective_chat.id,
+                                   text='WE ARE A PARTY OF UTTER MADNESS, '
+                                        'AND I AM A PERSON OF COMPLETE AND TOTAL DELUSION!')
+
 
 async def respond_with_picture(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    file = render_text_on_image("red_ebalo.png", update.message.text)
+    file = render_text_on_image("tgbot/assets/red_ebalo.png", update.message.text)
     await context.bot.send_sticker(chat_id=update.effective_chat.id, sticker=file)
 
-async def image2string(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    photo = update.message.photo[-1]  
-    
 
-    file = await photo.get_file()  
-    bio = BytesIO()          
-    await file.download_to_memory(out=bio)   
+async def image2string(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    photo = update.message.photo[-1]
+
+    file = await photo.get_file()
+    bio = BytesIO()
+    await file.download_to_memory(out=bio)
 
     bio.seek(0)
 
@@ -36,45 +36,22 @@ async def image2string(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
     await update.message.reply_text(s)
 
-async def inline_respond_with_sticker(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
+async def handle_inline(update: Update, context: ContextTypes.DEFAULT_TYPE):
     query = update.inline_query.query
     if not query:
         return
-    
-    xcom = "https://x.com"
-
-    if xcom in query:
-
-        results = [
-            {
-                "type": "article",
-                "id": "1",
-                "title": "fix xitter links",
-                "description": query.replace(xcom, "https://fxtwitter.com"),
-                "input_message_content": {
-                    "message_text": query.replace(xcom, "https://fxtwitter.com")
-                }
-            }
-        ]
-
-        await context.bot.answer_inline_query(update.inline_query.id, results)
 
+    if "https://x.com" in query:
+        await handle_xitter(update, context)
         return
 
-    bot_info = await context.bot.get_me()
+    if query.startswith("hentai"):
+        await handle_hentai(update, context)
+        return
 
-    file = render_text_on_image("red_ebalo.png", query)
+    await handle_red_ebalo(update, context)
 
-    file_id = (await context.bot.send_sticker(chat_id=-1002471390283, sticker=file)).sticker.file_id
-    results = [
-        {
-            "type": "sticker",
-            "id": "1",
-            "sticker_file_id": file_id 
-        } 
-    ]
-    await context.bot.answer_inline_query(update.inline_query.id, results)
 
 def main():
     application = ApplicationBuilder().token(TOKEN).build()
@@ -86,9 +63,10 @@ def main():
 
     application.add_handler(MessageHandler(filters.PHOTO, image2string))
     application.add_handler(picture_handler)
-    application.add_handler(InlineQueryHandler(inline_respond_with_sticker))
+    application.add_handler(InlineQueryHandler(handle_inline))
 
     application.run_polling()
 
+
 if __name__ == '__main__':
     main()