app.py

import discord
from discord.ext import commands
from ChatAI.chat_ai import pipe as ai

DEFAULT_CHATTER_CHANNEL="ai_chatter"

# Set up Discord bot intents and command prefix
intents = discord.Intents.default()
intents.message_content = True
intents.messages = True
bot = commands.Bot(command_prefix="!", intents=intents)

# Dictionary to track message count per channel
message_counts = {}

@bot.event
async def on_message(message):
    guild = message.guild  # Get the guild (server) the message is from
    channel = discord.utils.get(guild.text_channels, name=DEFAULT_CHATTER_CHANNEL)

    # Verification
    if message.channel != channel or message.author.bot: return;
    # Ensure tracking exists for this channel
    if message.channel.id not in message_counts:
        message_counts[message.channel.id] = 0

    # Increment message count
    message_counts[message.channel.id] += 1
    print(message_counts[message.channel.id])

    respond_to_chat(message.content)

    messages = []
    if message_counts[message.channel.id] >= 10: # Check if the count reaches 10
        async for message in channel.history(limit=10):
            messages.append(message.content)

        previous_messages = ("\n".join(messages))
        respond_to_chat(message)

        message_counts[message.channel.id] = 0 # Reset the counter

    await bot.process_commands(message) # Ensure commands still work

def respond_to_chat(message: str) ->str:
    response = generate(message)
    parts = split_string(response)
    guild = discord.utils.get(bot.guilds, name="PrzebieralniaKoedukacyjna")
    channel = discord.utils.get(guild.channels, name=DEFAULT_CHATTER_CHANNEL)
    for part in parts:
        channel.send(part)
    return str("\n".join(parts))


def split_string(text: str) -> list[str]:
    """Helper function to split text into chunks"""
    return [text[i:i+3900] for i in range(0, len(text), 3900)]

def generate(
    prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0
):
    temperature = float(temperature)
    if temperature < 1e-2:
        temperature = 1e-2
    top_p = float(top_p)

    generate_kwargs = dict(
        temperature=temperature,
        max_new_tokens=max_new_tokens,
        top_p=top_p,
        repetition_penalty=repetition_penalty,
        do_sample=True,
        seed=42,
    )

    return ai.text_generation(prompt, **generate_kwargs, stream=False, details=False, return_full_text=True)

@bot.event
async def on_ready():
    print(f'Logged in as {bot.user}')  # Logs bot login in console

    guild = discord.utils.get(bot.guilds, name="PrzebieralniaKoedukacyjna")
    if guild:
        # Get the channel by name
        channel = discord.utils.get(guild.channels, name=DEFAULT_CHATTER_CHANNEL)
        if channel:
            print(f"Channel found: {channel.name} (ID: {channel.id})")
        else:
            print("Channel not found!")

        await channel.send(f"{bot.user} logged in, runnin on 'huggingface.co/spaces")

bot.run("MTMzODQ4NTY2MzY3MDA3OTQ4OA.GlmK1T.7ZeEiDz7ViY3zvuSqlacVocDMSZ-ln80c09AS4")