GENERATING AGAIN

.gitignore

@@ -1,3 +1,4 @@
 /.idea/
 /.Chatter/
 .aider*
+.env

ChatAI/chat_ai.py

@@ -1,4 +1,4 @@
 from transformers import pipeline
 
 
-pipe = pipeline("text-generation", model="microsoft/phi-2")
+pipe = pipeline( model="facebook/blenderbot-400M-distill")

app.py

@@ -1,9 +1,21 @@
-<<<<<<< SEARCH
-@bot.command
+import discord
+from discord.ext import commands
+from ChatAI.chat_ai import pipe as ai
+
+
+# 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="ai_chatter")
-    
+
     # Verification
     if message.channel != channel: return;
     if message.author.bot: return;
@@ -16,25 +28,33 @@ async def on_message(message):
     print(message_counts[message.channel.id])
 
     messages = []
-    logging.info("Initializing")
-    channel = discord.utils.get(guild.text_channels, name="ai_chatter")
-    if channel:
-        print(f"Channel found: {channel.name}")
-=======
-@bot.command
-async def on_message(message):
-    # Get the guild (server) the message is from
-    guild = message.guild
-    channel = discord.utils.get(guild.text_channels, name="ai_chatter")
-    
-    # Verification
-    if message.channel != channel: return;
-    if message.author.bot: return;
+    if message_counts[message.channel.id] >= 4: # Check if the count reaches 10
+        async for message in channel.history(limit=4):
+            messages.append(message.content)
 
-    # Track message count
-    global message_count
-    message_count += 1
-    print(f"Message {message_count} in {channel.name}")
+        response = ai("\n".join(messages), max_length=1900, truncation=True, pad_token_id=50256)
 
-    messages = []
->>>>>>> REPLACE
+        await channel.send(response[0]['generated_text'])
+
+
+        message_counts[message.channel.id] = 0 # Reset the counter
+
+
+    await bot.process_commands(message) # Ensure commands still work
+
+@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="ai_chatter")
+        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")

discord.py

@@ -1,9 +0,0 @@
-<<<<<<< SEARCH
-from discord.ext import commands
-from transformers import PreTrainedModel, PreTrained
-from ChatAI.chat_ai import pipe as ai
-=======
-from discord.ext import commands
-from transformers import PreTrainedModel, PreTrained
-from discord import Webhook
->>>>>>> REPLACE