update

chatbot/plugins/chat.py

@@ -7,46 +7,36 @@
 # This project is proprietary software. Unauthorized redistribution is prohibited.
 
 import asyncio
-import random
-import os
 import base64
+import json
+import os
+import random
 import re
 import uuid
-import requests
-import json
-import cohere # type: ignore
 from datetime import datetime as dt, timedelta
-
-from gpytranslate import SyncTranslator
 from io import BytesIO
 from typing import List
 
+import cohere
+import requests
+from akenoai import *
+from akenoai.types import DifferentAPIDefault
+from google import genai
+from google.genai import types as ty
+from gpytranslate import SyncTranslator
 from PIL import Image
 from pyrogram import *
-from pyrogram import enums
-from pyrogram import Client, filters
+from pyrogram import Client, enums, filters
 from pyrogram.errors import *
+from pyrogram.types import (InlineKeyboardButton, InlineKeyboardMarkup,
+                            InlineQuery, InlineQueryResultArticle,
+                            InlineQueryResultPhoto, InputMediaPhoto,
+                            InputTextMessageContent, Message)
+
 from config import *
-from database import users_collection, db
+from database import db, users_collection
 from logger import LOGS
 
-from pyrogram.types import (
-    InlineKeyboardButton,
-    InlineKeyboardMarkup,
-    InlineQueryResultArticle,
-    InputTextMessageContent,
-    InputMediaPhoto,
-    InlineQuery,
-    Message,
-    InlineQueryResultPhoto
-)
-
-from akenoai import *
-from akenoai.types import DifferentAPIDefault
-
-from google import genai
-from google.genai import types as ty
-
 trans = SyncTranslator()
 js = AkenoXJs(DifferentAPIDefault()).connect()
 
@@ -269,72 +259,63 @@ async def coheresearch(client, callback):
     ])
     await callback.edit_message_reply_markup(reply_markup=InlineKeyboardMarkup(keyboard))
 
-
-def inline_error(inline_text, **args):
-    answers = [
+def inline_error(text: str):
+    return [
         InlineQueryResultArticle(
-            title="Error!",
-            thumb_url="https://telegra.ph//file/586a3867c3e16ca6bb4fa.jpg",
+            title="❌ Error",
             input_message_content=InputTextMessageContent(
-                message_text=inline_text,
-                disable_web_page_preview=True
-            ),
-            **args
+                message_text=f"⚠️ {text}"
+            )
         )
     ]
-    return answers
 
-@Client.on_inline_query()
-async def inline_search(client, inline_query: InlineQuery):
+@Client.on_inline_query(filters.regex(".*"))
+async def inline_search(client: Client, inline_query: InlineQuery):
+    GOOGLE_CX = "8386c816e6fff4b2a"
     search_query = inline_query.query.strip()
     if not search_query:
         return
-    if search_query:
-        try:
-            response = requests.get(
-                'https://www.googleapis.com/customsearch/v1',
-                params={
-                    "key": str(GOOGLE_API_SEARCH),
-                    "cx": "8386c816e6fff4b2a",
-                    "q": search_query,
-                    "searchType": "image"
-                }
-            )
-            response_json = response.json()
-            if 'items' in response_json:
-                results = []
-                for i, item in enumerate(response_json['items']):
-                    title = item['title']
-                    description = item['image']['contextLink']
-                    url = item['link']
-                    result = InlineQueryResultPhoto(
-                        photo_url=url,
-                        id=str(i),
-                        title=title,
-                        description=description,
-                        thumb_url=url,
-                        input_message_content=InputTextMessageContent(
-                            message_text=f"**Title:** {title}\n\n**Description:** {description}"
-                        ),
-                        reply_markup=InlineKeyboardMarkup(
-                            [[InlineKeyboardButton(text="Open Link", url=url)]]
-                        ),
-                    )
-                    results.append(result)
-                await client.answer_inline_query(inline_query.id, results=results, cache_time=10)
-            else:
-                no_found = inline_error("No found result.")
-                await client.answer_inline_query(
-                    inline_query.id,
-                    results=no_found
+    try:
+        response = requests.get(
+            "https://www.googleapis.com/customsearch/v1",
+            params={
+                "key": GOOGLE_API_SEARCH,
+                "cx": GOOGLE_CX,
+                "q": search_query,
+                "searchType": "image"
+            }
+        )
+        response_json = response.json()
+
+        if "items" in response_json:
+            results = []
+            for i, item in enumerate(response_json["items"]):
+                title = item.get("title", "No Title")
+                description = item["image"].get("contextLink", "No Link")
+                url = item.get("link")
+
+                result = InlineQueryResultPhoto(
+                    photo_url=url,
+                    id=str(i),
+                    title=title,
+                    description=description,
+                    thumb_url=url,
+                    input_message_content=InputTextMessageContent(
+                        message_text=f"**Title:** {title}\n\n**Source:** {description}"
+                    ),
+                    reply_markup=InlineKeyboardMarkup(
+                        [[InlineKeyboardButton("🔗 Open Link", url=url)]]
+                    ),
                 )
-        except Exception as e:
-            LOGS.error(f"Error Inline: {type(e).__name__}")
-            error_ = inline_error(f"Error try again.")
-            await client.answer_inline_query(
-                inline_query.id,
-                results=error_
-            )
+                results.append(result)
+
+            await inline_query.answer(results, cache_time=10)
+
+        else:
+            await inline_query.answer(inline_error("No result found."), cache_time=1)
+    except Exception as e:
+        await inline_query.answer(inline_error("Internal error. Try again later."), cache_time=1)
+        print(f"[INLINE ERROR] {type(e).__name__}: {e}")
 
 @Client.on_callback_query(filters.regex("^deeps_(on|off)_([a-f0-9]{8})$"))
 async def deepsearchmode(client, callback):
@@ -1409,6 +1390,7 @@ async def chatbot_talk(client: Client, message: Message):
                         {"$set": {"prompt_image": user_result["prompt"]}},
                         upsert=True
                     )
+                    LOGS.info(f"Response Image Generate: {message.from_user.first_name} | {message.from_user.id}")
                     await message.reply_photo(
                         photo="loading.jpg",
                         caption="Are you sure you want to prompt this Image generate?",
@@ -1501,6 +1483,8 @@ async def chatbot_talk(client: Client, message: Message):
                 )
                 backup_chat.append({"role": "model", "parts": [{"text": results}]})
                 await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+                LOGS.info(f"Response Deep Search: {message.from_user.first_name} | {message.from_user.id}")
+                await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return
 
             if check_is_system.get("is_system", False):
@@ -1583,6 +1567,7 @@ async def chatbot_talk(client: Client, message: Message):
                 )
                 backup_chat.append({"role": "model", "parts": [{"text": test_result}]})
                 await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+                LOGS.info(f"Response Gemini System: {message.from_user.first_name} | {message.from_user.id}")
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return
 
@@ -1661,6 +1646,7 @@ async def chatbot_talk(client: Client, message: Message):
             )
             backup_chat.append({"role": "model", "parts": [{"text": output}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+            LOGS.info(f"Response Gemini: {message.from_user.first_name} | {message.from_user.id}")
             await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             return