"""
Discord AI Bot - AI Features (Enhanced with Memory)
Author: RSK World
Website: https://rskworld.in
Email: help@rskworld.in
Phone: +91 93305 39277
Year: 2026
"""

import discord
from discord.ext import commands
import openai
from collections import deque
from config import OPENAI_API_KEY, AI_MODEL

class AI(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.use_new_api = False
        if not OPENAI_API_KEY:
            print("Warning: OPENAI_API_KEY not set. AI features will not work.")
        else:
            # Support both old and new OpenAI API versions
            try:
                # Try new API (v1.0+)
                self.client = openai.OpenAI(api_key=OPENAI_API_KEY)
                self.use_new_api = True
            except (AttributeError, TypeError):
                # Fall back to old API
                openai.api_key = OPENAI_API_KEY
                self.use_new_api = False
        # Memory structure: {user_id: deque([{"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}, ...], maxlen=10)}
        self.memory = {}

    def get_user_memory(self, user_id):
        if user_id not in self.memory:
            self.memory[user_id] = deque(maxlen=10)
        return self.memory[user_id]

    @commands.hybrid_command(name="ask", description="Ask the AI a question with memory")
    async def ask(self, ctx, *, question: str):
        """Interact with the AI using GPT-3.5 Turbo and conversation memory."""
        await ctx.defer()
        user_id = ctx.author.id
        user_memory = self.get_user_memory(user_id)
        
        # Build messages payload
        messages = [{"role": "system", "content": "You are a helpful Discord assistant powered by RSK World. You remember the last few exchanges with this user."}]
        messages.extend(list(user_memory))
        messages.append({"role": "user", "content": question})

        if not OPENAI_API_KEY:
            await ctx.send("❌ OpenAI API key is not configured. Please set OPENAI_API_KEY in your .env file.")
            return

        try:
            if hasattr(self, 'use_new_api') and self.use_new_api and hasattr(self, 'client'):
                # New API (v1.0+)
                response = self.client.chat.completions.create(
                    model=AI_MODEL,
                    messages=messages
                )
                answer = response.choices[0].message.content
            else:
                # Old API
                response = openai.ChatCompletion.create(
                    model=AI_MODEL,
                    messages=messages
                )
                answer = response.choices[0].message.content
            
            # Update memory
            user_memory.append({"role": "user", "content": question})
            user_memory.append({"role": "assistant", "content": answer})

            embed = discord.Embed(
                title="AI Response",
                description=answer,
                color=discord.Color.blue()
            )
            embed.set_footer(text=f"Memory Active | Requested by {ctx.author}", icon_url=ctx.author.display_avatar.url)
            await ctx.send(embed=embed)
        except Exception as e:
            await ctx.send(f"An error occurred: {e}")

    @commands.hybrid_command(name="clear_memory", description="Clear your conversation memory")
    async def clear_memory(self, ctx):
        """Wipe the user's AI conversation history."""
        if ctx.author.id in self.memory:
            self.memory[ctx.author.id].clear()
            await ctx.send("Your conversation memory has been cleared!", delete_after=5)
        else:
            await ctx.send("You have no active conversation memory.", delete_after=5)

async def setup(bot):
    await bot.add_cog(AI(bot))