2025-06-28 11:25:55 +05:00
|
|
|
|
import os
|
|
|
|
|
|
import discord
|
|
|
|
|
|
from discord.ext import commands
|
|
|
|
|
|
from openai_interaction.image_recognition_test import image_recognition
|
|
|
|
|
|
|
|
|
|
|
|
# Загружаем токен из переменной среды
|
|
|
|
|
|
TOKEN = os.getenv('SCREAMING_OPOSSUM')
|
|
|
|
|
|
|
|
|
|
|
|
# Создаем клиент с префиксом (не обязателен для слэш-команд)
|
|
|
|
|
|
intents = discord.Intents.default()
|
|
|
|
|
|
intents.message_content = True
|
|
|
|
|
|
bot = commands.Bot(command_prefix='!', intents=intents)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# region Bot Stuff
|
|
|
|
|
|
@bot.event
|
|
|
|
|
|
async def on_ready():
|
|
|
|
|
|
print(f'Logged in as {bot.user} (ID: {bot.user.id})')
|
|
|
|
|
|
print('------')
|
|
|
|
|
|
try:
|
|
|
|
|
|
synced = await bot.tree.sync()
|
|
|
|
|
|
print(f'Synced {len(synced)} command(s).')
|
|
|
|
|
|
except Exception as e:
|
|
|
|
|
|
print(f'Failed to sync commands: {e}')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@bot.tree.command(name='hello', description='Replies with Hello World!')
|
|
|
|
|
|
async def hello_command(interaction: discord.Interaction):
|
|
|
|
|
|
await interaction.response.send_message('AAAAAAAAAAAAAAAAAAAAA!')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@bot.event
|
|
|
|
|
|
async def on_message(message: discord.Message):
|
|
|
|
|
|
if message.author == bot.user:
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
if message.attachments and bot.user in message.mentions:
|
|
|
|
|
|
message_attachments = [
|
|
|
|
|
|
{'filename': i_attachment.filename, 'url': i_attachment.url} for i_attachment in message.attachments
|
|
|
|
|
|
]
|
|
|
|
|
|
await message.channel.send(await image_recognition(message.content, message_attachments))
|
|
|
|
|
|
|
|
|
|
|
|
await bot.process_commands(message)
|
|
|
|
|
|
# endregion
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
|
if TOKEN is None:
|
|
|
|
|
|
raise RuntimeError('Environment variable SCREAMING_OPOSSUM is not set')
|
|
|
|
|
|
bot.run(TOKEN)
|
