src/components/model-selector.tsx

"use client"

import { useState, useEffect, useRef } from 'react'
import { MODEL_CONFIG_CODE_GENERATION } from '@/lib/constants'
import { useModel } from '@/lib/contexts/model-context'
import { ChevronDown } from 'lucide-react'

export function ModelSelector() {
  const { selectedModelIndex, setSelectedModelIndex } = useModel()
  const [isOpen, setIsOpen] = useState(false)
  const selectorRef = useRef<HTMLDivElement>(null)

  const handleSelect = (index: number) => {
    setSelectedModelIndex(index)
    setIsOpen(false)
  }

  useEffect(() => {
    const handleClickOutside = (event: MouseEvent) => {
      if (selectorRef.current && !selectorRef.current.contains(event.target as Node)) {
        setIsOpen(false)
      }
    }

    if (isOpen) {
      document.addEventListener('mousedown', handleClickOutside)
    }

    return () => {
      document.removeEventListener('mousedown', handleClickOutside)
    }
  }, [isOpen])

  return (
    <div className="relative inline-block text-left" ref={selectorRef}>
      <div
        onClick={() => setIsOpen(!isOpen)}
        className={`
          cursor-pointer flex items-center justify-between gap-2 
          text-white/90 hover:text-white 
          px-3 py-1.5 rounded-md
          border border-novita-gray/30 hover:border-novita-gray/60
          bg-novita-gray/5 hover:bg-novita-gray/20
          transition-all duration-200 ease-in-out
          ${isOpen ? 'border-novita-gray/60 bg-novita-gray/20' : ''}
        `}
      >
        <span className="text-sm">{MODEL_CONFIG_CODE_GENERATION[selectedModelIndex]?.id || 'Select model'}</span>
        <ChevronDown className={`w-4 h-4 transition-transform duration-200 ${isOpen ? 'transform rotate-180' : ''}`} />
      </div>
      
      {isOpen && (
        <div className="absolute z-40 mt-1 w-full origin-top-right rounded-md bg-novita-dark border border-novita-gray/20 shadow-lg">
          <div className="py-1">
            {MODEL_CONFIG_CODE_GENERATION.map((model, index) => (
              <div
                key={model.id}
                onClick={() => handleSelect(index)}
                className={`
                  px-4 py-2 text-xs cursor-pointer hover:bg-novita-gray/20
                  ${selectedModelIndex === index ? 'text-white bg-novita-gray/40' : 'text-white/70'}
                `}
              >
                {model.id}
              </div>
            ))}
          </div>
        </div>
      )}
    </div>
  )
}