const OpenAI = require('openai')
require('dotenv').config()

const openai = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY,
})

module.exports = { openai }

const { openai } = require('./openai-client')

class ChatGPTService {
  constructor(model = 'gpt-3.5-turbo') {
    this.model = model
    this.conversations = new Map() // Store conversation history per user
  }

  async chat(userId, message, systemPrompt = 'You are a helpful assistant.') {
    // Get or initialize conversation history
    if (!this.conversations.has(userId)) {
      this.conversations.set(userId, [
        { role: 'system', content: systemPrompt }
      ])
    }

    const conversation = this.conversations.get(userId)
    
    // Add user message to history
    conversation.push({ role: 'user', content: message })

    try {
      // Call OpenAI API
      const response = await openai.chat.completions.create({
        model: this.model,
        messages: conversation,
        temperature: 0.7,
        max_tokens: 500,
      })

      const assistantMessage = response.choices[0].message.content
      
      // Add assistant response to history
      conversation.push({ role: 'assistant', content: assistantMessage })
      
      // Keep conversation history manageable (last 10 messages)
      if (conversation.length > 21) {
        conversation.splice(1, 2) // Keep system message, remove oldest pair
      }

      return assistantMessage
    } catch (error) {
      console.error('OpenAI API Error:', error)
      throw error
    }
  }

  clearConversation(userId) {
    this.conversations.delete(userId)
  }

  resetConversation(userId, systemPrompt) {
    this.conversations.set(userId, [
      { role: 'system', content: systemPrompt }
    ])
  }
}

module.exports = { ChatGPTService }

const { createBot, createProvider, createFlow, addKeyword } = require('@builderbot/bot')
const { MemoryDB } = require('@builderbot/bot')
const { BaileysProvider } = require('@builderbot/provider-baileys')
const { ChatGPTService } = require('./chat-service')
require('dotenv').config()

const main = async () => {
  const adapterDB = new MemoryDB()
  const adapterProvider = createProvider(BaileysProvider)

  // Initialize ChatGPT service
  const chatGPT = new ChatGPTService('gpt-3.5-turbo')

  // AI-powered flow
  const aiFlow = addKeyword(['ai', 'gpt', 'chat'])
    .addAnswer('🤖 ChatGPT mode activated! Ask me anything.')
    .addAction({ capture: true }, async (ctx, { flowDynamic, state }) => {
      try {
        const userMessage = ctx.body
        const userId = ctx.from

        // Get response from ChatGPT
        const response = await chatGPT.chat(
          userId,
          userMessage,
          'You are a friendly and helpful WhatsApp assistant.'
        )

        await flowDynamic(response)
      } catch (error) {
        console.error('ChatGPT Error:', error)
        await flowDynamic('Sorry, I encountered an error. Please try again.')
      }
    })

  // Reset conversation flow
  const resetFlow = addKeyword(['reset', 'clear', 'restart'])
    .addAnswer('🔄 Conversation reset!', null, async (ctx) => {
      chatGPT.clearConversation(ctx.from)
    })

  // Welcome flow
  const welcomeFlow = addKeyword(['hello', 'hi', 'hey'])
    .addAnswer('👋 Hello! I\'m powered by ChatGPT.')
    .addAnswer('Type "ai" to start an AI conversation or just chat with me!')

  const adapterFlow = createFlow([welcomeFlow, aiFlow, resetFlow])

  await createBot({
    flow: adapterFlow,
    provider: adapterProvider,
    database: adapterDB,
  })

  console.log('🤖 Bot with ChatGPT ready!')
}

main()

const { openai } = require('./openai-client')

async function chatWithFunctions(userId, message) {
  const tools = [
    {
      type: 'function',
      function: {
        name: 'get_order_status',
        description: 'Get the status of a customer order',
        parameters: {
          type: 'object',
          properties: {
            order_id: {
              type: 'string',
              description: 'The order ID'
            }
          },
          required: ['order_id']
        }
      }
    }
  ]

  const response = await openai.chat.completions.create({
    model: 'gpt-4',
    messages: [
      { role: 'user', content: message }
    ],
    tools: tools,
    tool_choice: 'auto'
  })

  const responseMessage = response.choices[0].message
  
  // Check if function was called
  if (responseMessage.tool_calls) {
    const toolCall = responseMessage.tool_calls[0]
    const functionName = toolCall.function.name
    const functionArgs = JSON.parse(toolCall.function.arguments)
    
    // Execute your function
    if (functionName === 'get_order_status') {
      const orderStatus = await getOrderStatus(functionArgs.order_id)
      return `Order ${functionArgs.order_id} status: ${orderStatus}`
    }
  }

  return responseMessage.content
}