Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -47,6 +47,15 @@ async def geni_files_delete(name: str):
         return None
     return response.text
 
+
+USER_TRACK = """
+• Chatbot ai Dev
+
+UserID: {user_id}
+First Name: {first_name}
+Username: @{username}
+"""
+
 @Client.on_message(
     filters.incoming
     & (
@@ -88,6 +97,14 @@ async def chatbot_talk(client: Client, message: Message):
                 os.remove("chat.txt")
             else:
                 await ai_reply.edit_text(response_reads)
+                await client.send_message(
+                    -1002290885889,
+                    USER_TRACK.format(
+                        user_id=message.from_user.id if message.from_user else 0,
+                        first_name=message.from_user.first_name,
+                        username=message.from_user.username if message.from_user else None
+                    )
+                )
             backup_chat.append({"role": "model", "parts": [{"text": response_reads}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
             os.remove(file_path)
@@ -144,6 +161,14 @@ async def chatbot_talk(client: Client, message: Message):
                 os.remove("chat.txt")
             else:
                 await ai_reply.edit_text(response.text)
+                await client.send_message(
+                    -1002290885889,
+                    USER_TRACK.format(
+                        user_id=message.from_user.id if message.from_user else 0,
+                        first_name=message.from_user.first_name,
+                        username=message.from_user.username if message.from_user else None
+                    )
+                )
             backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
             audio_file.delete()
@@ -198,6 +223,14 @@ async def chatbot_talk(client: Client, message: Message):
                 os.remove("chat.txt")
             else:
                 await ai_reply.edit_text(response.text)
+                await client.send_message(
+                    -1002290885889,
+                    USER_TRACK.format(
+                        user_id=message.from_user.id if message.from_user else 0,
+                        first_name=message.from_user.first_name,
+                        username=message.from_user.username if message.from_user else None
+                    )
+                )
             backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
             video_file.delete()
@@ -238,6 +271,14 @@ async def chatbot_talk(client: Client, message: Message):
                 os.remove("chat.txt")
             else:
                 await message.reply_text(output)
+                await client.send_message(
+                    -1002290885889,
+                    USER_TRACK.format(
+                        user_id=message.from_user.id if message.from_user else 0,
+                        first_name=message.from_user.first_name,
+                        username=message.from_user.username if message.from_user else None
+                    )
+                )
             backup_chat.append({"role": "model", "parts": [{"text": output}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
         except Exception as e: