const Discord = require('discord.js');
const fetch = require('node-fetch');
require('dotenv').config()
// hi
const { Client, GatewayIntentBits, ActivityType, Partials } = require('discord.js');

const client = new Client({
  intents: [
    GatewayIntentBits.DirectMessages,
    GatewayIntentBits.Guilds,
    GatewayIntentBits.GuildBans,
    GatewayIntentBits.GuildMessages,
    GatewayIntentBits.MessageContent,
  ],
  partials: [Partials.Channel],
});

const channelID = '1094494101631680653'; // Replace with your channel ID
const channelID2 = '1094628334727614605'; // Replace with your channel ID

const conversations = new Map();

client.once('ready', () => {
  console.log('Bot is ready.');
  client.user.setPresence({
    activities: [{ name: `AI`, type: ActivityType.Playing }],
    status: 'dnd',
  });
});

client.on('messageCreate', async (message) => {
  // Only respond in the specified channels
  if (message.channel.id !== channelID && message.channel.id !== channelID2) {
    return;
  }
    if (message.author.bot) return; // Ignore messages from bots

  if (conversations.get(message.author.id)?.busy) {
    message.delete()
    await message.author.send("I'm currently busy with another process. Please try again later.");
    return;
  }

  const userID = message.author.id;
  let conversation = conversations.get(userID) || { messages: [], busy: false };


  if (conversation.messages.length === 0) {
    conversation.messages.push({ role: 'user', content: `My name is ${message.author.username}.` });
    conversation.messages.push({ role: 'assistant', content: `Hello, ${message.author.username}.` });
    conversation.messages.push({ role: 'user', content: message.cleanContent });

  } else {
  // Append user message to conversation history
  conversation.messages.push({ role: 'user', content: message.cleanContent });
  }
  if (message.content === '!reset') {
    conversations.delete(userID); // Delete user's conversation map if they request reset
    await message.channel.send('Conversation reset.');
    return;
  }

  // Append user message to conversation history
  conversation.messages.push({ role: 'user', content: message.cleanContent });

  try {
    conversation.busy = true;

    const response = await generateResponse(conversation);

    // Append bot message to conversation history
    conversation.messages.push({ role: 'assistant', content: response });

    await message.channel.send(response);

    conversations.set(userID, conversation); // Update user's conversation map in memory
  } catch (err) {
    console.error(err);
    await message.channel.send('Oops, something went wrong!');
  } finally {
    conversation.busy = false;
  }
});

async function generateResponse(conversation) {
  const controller = new AbortController();
  const timeout = setTimeout(() => {
    controller.abort();
  }, 900000);

  try {
    const response = await fetch('http://192.168.0.15/v1/chat/completions', {
      method: 'POST',
      headers: {
        'accept': 'application/json',
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({ messages: conversation.messages}),
      signal: controller.signal
    });

    const responseData = await response.json();
    const choice = responseData.choices[0];

    // Remove "user None:" and any text after it from the response
    const responseText = choice.message.content.trim();
    const startIndex = responseText.indexOf('user None:');
    const sanitizedResponse = startIndex === -1 ? responseText : responseText.substring(0, startIndex);

    return sanitizedResponse;
  } catch (err) {
    console.error(err);
    return 'Oops, something went wrong!';
  } finally {
    clearTimeout(timeout);
  }
}


client.login(process.env.THE_TOKEN); // Replace with your bot token