s

app.py

@@ -12,6 +12,7 @@ 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
@@ -29,19 +30,22 @@ async def on_message(message):
     print(message_counts[message.channel.id])
 
     messages = []
-    if message_counts[message.channel.id] >= 1: # Check if the count reaches 10
-        async for message in channel.history(limit=4):
+    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))
         response = ai.text_generation(previous_messages, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
-
         splited_response = split_string(response)
         for part in splited_response:
             await channel.send(part)
 
         message_counts[message.channel.id] = 0 # Reset the counter
 
+    response = ai.text_generation(message, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
+    splited_response = split_string(response)
+    for part in splited_response:
+        await channel.send(part)
+
     await bot.process_commands(message) # Ensure commands still work
 
 def split_string(text: str) -> list[str]: