update

chatbot/plugins/chat.py

@@ -276,6 +276,10 @@ def inline_error(text: str):
 
 @Client.on_inline_query(filters.regex(".*"))
 async def inline_search(client: Client, inline_query: InlineQuery):
+    user_id = inline_query.from_user.id
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
+        return
     GOOGLE_CX = "8386c816e6fff4b2a"
     search_query = inline_query.query.strip()
     if not search_query:
@@ -783,11 +787,19 @@ def anonymous_user(user_id: int = None):
 async def menu_back(client, callback):
     uuidstr = callback.matches[0].group(1)
     user_id = callback.from_user.id
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
+        await callback.answer("Sorry your account is frozen", True)
+        return
+
     keyboard = create_keyboard(
         user_id=user_id,
         uuid_str=uuidstr
     )
-    await callback.edit_message_reply_markup(reply_markup=keyboard)
+    try:
+        await callback.edit_message_reply_markup(reply_markup=keyboard)
+    except MessageNotModified:
+        pass
 
 def create_keyboard(
     likes: int = 0,
@@ -1221,6 +1233,7 @@ async def chatbot_talk(client: Client, message: Message):
             )
             backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+            LOGS.info(f"Response Upload Photo: {message.from_user.first_name} | {message.from_user.id}\n\nText: {caption}")
             await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             return
 
@@ -1284,6 +1297,7 @@ async def chatbot_talk(client: Client, message: Message):
             )
             backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+            LOGS.info(f"Response Upload Video: {message.from_user.first_name} | {message.from_user.id}\n\nText: {caption}")
             await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             return
 
@@ -1347,6 +1361,7 @@ async def chatbot_talk(client: Client, message: Message):
                 )
                 backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
                 await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+                LOGS.info(f"Response Upload Sticker: {message.from_user.first_name} | {message.from_user.id}")
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return
 
@@ -1449,7 +1464,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}")
+                    LOGS.info(f"Response Image Generate: {message.from_user.first_name} | {message.from_user.id}\n\nText: {query_base}")
                     await message.reply_photo(
                         photo="loading.jpg",
                         caption="Are you sure you want to prompt this Image generate?",
@@ -1542,7 +1557,7 @@ 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}")
+                LOGS.info(f"Response Deep Search: {message.from_user.first_name} | {message.from_user.id}\n\nText: {query_base}")
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return
 
@@ -1568,6 +1583,7 @@ async def chatbot_talk(client: Client, message: Message):
                             )
                         ]
                     ]
+                    LOGS.info(f"Response info data system: {message.from_user.first_name} | {message.from_user.id}\n\nText: {query_base}")
                     await message.reply_text(
                         "Are you sure you want to delete this data history?",
                         reply_markup=InlineKeyboardMarkup(buttons)
@@ -1626,7 +1642,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}")
+                LOGS.info(f"Response Gemini System: {message.from_user.first_name} | {message.from_user.id}\n\nText: {query_base}")
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return
 
@@ -1652,6 +1668,7 @@ async def chatbot_talk(client: Client, message: Message):
                         )
                     ]
                 ]
+                LOGS.info(f"Response info data: {message.from_user.first_name} | {message.from_user.id}\n\nText: {query_base}")
                 await message.reply_text(
                     "Are you sure you want to delete this data history?",
                     reply_markup=InlineKeyboardMarkup(buttons)
@@ -1705,7 +1722,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}")
+            LOGS.info(f"Response Gemini: {message.from_user.first_name} | {message.from_user.id}\n\nText: {query_base}")
             await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             return