'use client';

import { useState, useRef, useMemo } from 'react';
import { motion, AnimatePresence } from 'framer-motion';
import { useRouter } from 'next/navigation';
import Link from 'next/link';
import Image from 'next/image';
import { useTranslation } from 'react-i18next';
import {
  User, Mail, Phone, Lock, Eye, EyeOff, MessageCircle,
  Camera, X, CheckCircle, XCircle, ArrowLeft, Building,
  Loader2, Shield, KeyRound, FileText
} from 'lucide-react';
import toast, { Toaster } from 'react-hot-toast';
import { addOwner, loginWithEmail, sendEmailOTP, verifyEmail } from '../../utils/api';
import AuthService from '../../services/AuthService';
import { OwnerType, OwnerTypeLabels } from '../../enums';

export default function OwnerRegisterPage() {
  const router = useRouter();
  const { t } = useTranslation();
  const [step, setStep] = useState(1);
  const [showOtpModal, setShowOtpModal] = useState(false);
  const [showPassword, setShowPassword] = useState(false);
  const [showConfirmPassword, setShowConfirmPassword] = useState(false);
  const [isLoading, setIsLoading] = useState(false);

  const [formData, setFormData] = useState({
    firstName: '',
    lastName: '',
    email: '',
    phone: '',
    whatsapp: '',
    phone2: '',
    nationalNumber: '',
    password: '',
    confirmPassword: '',
    ownerType: OwnerType.PERSON,
    agreeTerms: false
  });

  const [idImages, setIdImages] = useState({ front: null, back: null, license: null });
  const [idImagePreviews, setIdImagePreviews] = useState({ front: '', back: '', license: '' });
  const [otpCode, setOtpCode] = useState('');
  const [errors, setErrors] = useState({});

  const fileInputFrontRef = useRef(null);
  const fileInputBackRef = useRef(null);
  const fileInputLicenseRef = useRef(null);

  const isCompany = Number(formData.ownerType) === OwnerType.REAL_ESTATE_AGENCY;

  const handleImageUpload = (side, file) => {
    if (!file) return;
    if (!file.type.startsWith('image/')) {
      toast.error(t('register.selectValidImage'));
      return;
    }
    if (file.size > 5 * 1024 * 1024) {
      toast.error(t('register.imageSizeLimit'));
      return;
    }
    const reader = new FileReader();
    reader.onloadend = () => {
      setIdImagePreviews(prev => ({ ...prev, [side]: reader.result }));
    };
    reader.readAsDataURL(file);
    setIdImages(prev => ({ ...prev, [side]: file }));
    toast.success(t('register.imageUploadSuccess'), { style: { background: '#dcfce7', color: '#166534' } });
  };

  const validateEmail = (email) => /^[^\s@]+@[^\s@]+\.[^\s@]+$/.test(email);
  const validatePhone = (phone) => /^(09|05)[0-9]{8}$/.test(phone);

  const validateStep1 = () => {
    const newErrors = {};
    if (!formData.firstName) newErrors.firstName = t('register.firstNameRequired');
    if (!formData.lastName) newErrors.lastName = t('register.lastNameRequired');
    if (!formData.email) newErrors.email = t('register.emailRequired');
    else if (!validateEmail(formData.email)) newErrors.email = t('register.emailInvalid');
    if (!formData.phone) newErrors.phone = t('register.phoneRequired');
    else if (!validatePhone(formData.phone)) newErrors.phone = t('register.phoneInvalid') + ` (${t('register.phoneMustStartWith')} 09/05)`;
    if (!formData.whatsapp) newErrors.whatsapp = t('register.whatsappRequired');
    else if (!validatePhone(formData.whatsapp)) newErrors.whatsapp = t('register.phoneInvalid') + ` (${t('register.phoneMustStartWith')} 09/05)`;
    if (!formData.password) newErrors.password = t('register.passwordRequired');
    else if (formData.password.length < 6) newErrors.password = t('validation.passwordMinLength');
    if (formData.password !== formData.confirmPassword) newErrors.confirmPassword = t('validation.passwordsMatch');
    setErrors(newErrors);
    return Object.keys(newErrors).length === 0;
  };

  const validateStep2 = () => {
    const newErrors = {};
    if (!idImages.front) newErrors.front = t('register.frontIdRequired');
    if (!idImages.back) newErrors.back = t('register.backIdRequired');
    setErrors(newErrors);
    return Object.keys(newErrors).length === 0;
  };

  const handleNextStep = () => {
    if (validateStep1()) {
      setStep(2);
      window.scrollTo({ top: 0, behavior: 'smooth' });
    } else {
      toast.error(t('validation.correctErrors'));
    }
  };

  const handleSubmit = async (e) => {
    e.preventDefault();
    if (!validateStep2()) {
      toast.error(t('register.completeRequiredImages'));
      return;
    }
    if (!formData.agreeTerms) {
      toast.error(t('validation.agreeToTerms'));
      return;
    }

    setIsLoading(true);
    const payload = {
      firstName: formData.firstName,
      lastName: formData.lastName,
      email: formData.email,
      phoneNumber: formData.phone || '',
      whatsAppNumber: formData.whatsapp,
      phone: formData.phone2,
      nationalNumber: formData.nationalNumber,
      password: formData.password,
      ownerType: formData.ownerType,
    };

    try {
      const licenseImage = isCompany ? idImages.license : null;
      const res = await addOwner(payload, idImages.front, idImages.back, licenseImage);
      if (res.status === 200 || res.ok) {
        const tempToken = res.data;
        if (tempToken) AuthService.addToken(tempToken);
        toast.success(res.message || t('register.accountCreated'), { duration: 4000 });

        const loginRes = await loginWithEmail(formData.email, formData.password);
        if (loginRes.status === 206) {
          const otpToken = loginRes.data;
          if (otpToken) AuthService.addToken(otpToken);
          toast(loginRes.message || t('register.otpSentEmail'), { icon: '📧' });
          setShowOtpModal(true);
        } else if (loginRes.status === 200) {
          const loginToken = loginRes.data;
          if (loginToken) AuthService.addToken(loginToken);
          toast.success(loginRes.message || t('register.loginSuccess'));
          router.push('/');
        }
      } else {
        toast.error(res.message || res.data?.message || t('register.accountCreationFailed'));
      }
    } catch (err) {
      toast.error(err.message || t('register.registrationError'));
    } finally {
      setIsLoading(false);
    }
  };

  const handleVerifyOTP = async () => {
    if (!otpCode || otpCode.length < 4) {
      toast.error(t('register.otpRequired'));
      return;
    }
    setIsLoading(true);
    try {
      const res = await verifyEmail(otpCode);
      if (res.status === 200) {
        AuthService.deleteToken();
        toast.success(res.message || t('register.emailVerified'));
        setShowOtpModal(false);
        setTimeout(() => router.push('/login'), 1500);
      } else {
        toast.error(res.message || res.data?.message || t('accountVerification.otpInvalid'));
      }
    } catch (err) {
      toast.error(err.message || t('register.verificationError'));
    } finally {
      setIsLoading(false);
    }
  };

  const handleResendOTP = async () => {
    setIsLoading(true);
    try {
      await sendEmailOTP();
      toast.success(t('register.newOtpSent'));
    } catch (err) {
      toast.error(t('register.resendOtpFailed'));
    } finally {
      setIsLoading(false);
    }
  };

  const fadeInUp = {
    initial: { opacity: 0, y: 20 },
    animate: { opacity: 1, y: 0 },
    transition: { duration: 0.5 }
  };

  const staggerContainer = {
    animate: { transition: { staggerChildren: 0.1 } }
  };

  const backgroundElements = useMemo(() => {
    const circles = [
      { style: { top: '20%', right: '20%', width: '256px', height: '256px' }, className: 'bg-amber-500/5' },
      { style: { bottom: '20%', left: '20%', width: '320px', height: '320px' }, className: 'bg-blue-500/5' },
      { style: { top: '50%', left: '50%', width: '384px', height: '384px', transform: 'translate(-50%, -50%)' }, className: 'bg-purple-500/5' },
    ];
    const dots = Array.from({ length: 10 }).map((_, i) => ({
      left: `${5 + i * 10}%`,
      top: `${10 + (i * 7) % 80}%`,
      size: `${80 + (i % 5) * 15}px`
    }));
    return (
      <>
        {circles.map((circle, i) => (
          <div key={`circle-${i}`} className={`absolute rounded-full ${circle.className}`} style={circle.style} />
        ))}
        {dots.map((dot, i) => (
          <div key={`dot-${i}`} className="absolute rounded-full bg-amber-500/10" style={{ left: dot.left, top: dot.top, width: dot.size, height: dot.size }} />
        ))}
      </>
    );
  }, []);

  return (
    <div className="min-h-screen bg-gradient-to-br from-gray-950 via-gray-900 to-gray-950 flex items-center justify-center p-4 relative overflow-hidden">
      <Toaster position="top-center" reverseOrder={false} />
      <div className="absolute inset-0 overflow-hidden">{backgroundElements}</div>

      <motion.div initial={{ opacity: 0, scale: 0.95 }} animate={{ opacity: 1, scale: 1 }} transition={{ duration: 0.5 }}
        className="relative z-10 w-full max-w-2xl">
        <div className="mb-8">
          <div className="flex items-center justify-between mb-4">
            <Link href="/auth/choose-role" className="flex items-center gap-2 text-gray-400 hover:text-white transition-colors group">
              <motion.div whileHover={{ x: -5 }}><ArrowLeft className="w-4 h-4" /></motion.div>
              <span>{t('forgotPassword.backToLogin')}</span>
            </Link>
            <span className="text-sm text-gray-400">{t('register.stepOf2', { step: step })}</span>
          </div>
          <div className="flex gap-2">
            {[1, 2].map((s) => (
              <motion.div key={s} className={`h-2 flex-1 rounded-full ${step >= s ? 'bg-amber-500' : 'bg-gray-700'}`} animate={{ scaleX: step >= s ? 1 : 0.5 }} />
            ))}
          </div>
        </div>

        <motion.div key={step} initial={{ opacity: 0, x: 20 }} animate={{ opacity: 1, x: 0 }} exit={{ opacity: 0, x: -20 }} transition={{ duration: 0.3 }}
          className="bg-white/5 backdrop-blur-xl rounded-3xl shadow-2xl border border-white/10 overflow-hidden">
          <div className="bg-gradient-to-r from-amber-500 to-amber-600 p-8 text-center relative overflow-hidden">
            <motion.div initial={{ scale: 0 }} animate={{ scale: 1 }} transition={{ delay: 0.2, type: "spring" }}
              className="absolute -top-10 -right-10 w-40 h-40 bg-white/10 rounded-full" />
            <motion.div initial={{ y: 20, opacity: 0 }} animate={{ y: 0, opacity: 1 }} className="relative z-10">
              <motion.div animate={{ rotate: [0, 10, -10, 0] }} transition={{ duration: 2, repeat: Infinity }}
                className="w-20 h-20 mx-auto mb-4 bg-white/20 rounded-2xl flex items-center justify-center backdrop-blur-sm">
                <Building className="w-10 h-10 text-white" />
              </motion.div>
              <h1 className="text-3xl font-bold text-white mb-2">{step === 1 ? t('register.owner.step1Title') : t('register.owner.step1bTitle')}</h1>
              <p className="text-amber-100">{step === 1 ? t('register.owner.step1Desc') : t('register.owner.step1bDesc')}</p>
            </motion.div>
          </div>

          <div className="p-8">
            <motion.form variants={staggerContainer} initial="initial" animate="animate"
              onSubmit={step === 1 ? (e) => { e.preventDefault(); handleNextStep(); } : handleSubmit}
              className="space-y-6">
              {step === 1 && (
                <>
                  <motion.div variants={fadeInUp} className="grid grid-cols-2 gap-3">
                    <div>
                      <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.firstName')} <span className="text-red-500">*</span></label>
                      <div className="relative group">
                        <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                          <User className={`w-5 h-5 ${errors.firstName ? 'text-red-500' : 'text-gray-400 group-focus-within:text-amber-500'}`} />
                        </div>
                        <input type="text" value={formData.firstName}
                          onChange={(e) => { setFormData({...formData, firstName: e.target.value}); setErrors({...errors, firstName: null}); }}
                          className={`w-full pr-12 pl-4 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.firstName ? 'border-red-500' : 'border-gray-700'}`}
                          placeholder={t('register.firstName')} />
                      </div>
                      {errors.firstName && <p className="text-red-500 text-sm mt-1">{errors.firstName}</p>}
                    </div>
                    <div>
                      <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.lastName')} <span className="text-red-500">*</span></label>
                      <input type="text" value={formData.lastName}
                        onChange={(e) => { setFormData({...formData, lastName: e.target.value}); setErrors({...errors, lastName: null}); }}
                        className={`w-full px-4 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.lastName ? 'border-red-500' : 'border-gray-700'}`}
                        placeholder={t('register.lastName')} />
                      {errors.lastName && <p className="text-red-500 text-sm mt-1">{errors.lastName}</p>}
                    </div>
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.email')} <span className="text-red-500">*</span></label>
                    <div className="relative group">
                      <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                        <Mail className={`w-5 h-5 ${errors.email ? 'text-red-500' : 'text-gray-400 group-focus-within:text-amber-500'}`} />
                      </div>
                      <input type="email" value={formData.email}
                        onChange={(e) => { setFormData({...formData, email: e.target.value}); setErrors({...errors, email: null}); }}
                        className={`w-full pr-12 pl-4 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.email ? 'border-red-500' : 'border-gray-700'}`}
                        placeholder={t('register.emailPlaceholder')} />
                    </div>
                    {errors.email && <p className="text-red-500 text-sm mt-1">{errors.email}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.phone')} <span className="text-gray-500">({t('register.optional')})</span></label>
                    <div className="relative group">
                      <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                        <Phone className="w-5 h-5 text-gray-400 group-focus-within:text-amber-500" />
                      </div>
                      <input type="tel" value={formData.phone}
                        onChange={(e) => { setFormData({...formData, phone: e.target.value}); setErrors({...errors, phone: null}); }}
                        className="w-full pr-12 pl-4 py-3 bg-white/5 border border-gray-700 rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all"
                        placeholder={t('register.phoneOptionalPlaceholder')} />
                    </div>
                    {errors.phone && <p className="text-red-500 text-sm mt-1">{errors.phone}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.whatsapp')} <span className="text-red-500">*</span></label>
                    <div className="relative group">
                      <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                        <MessageCircle className={`w-5 h-5 ${errors.whatsapp ? 'text-red-500' : 'text-gray-400 group-focus-within:text-amber-500'}`} />
                      </div>
                      <input type="tel" value={formData.whatsapp}
                        onChange={(e) => { setFormData({...formData, whatsapp: e.target.value}); setErrors({...errors, whatsapp: null}); }}
                        className={`w-full pr-12 pl-4 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.whatsapp ? 'border-red-500' : 'border-gray-700'}`}
                        placeholder={t('register.whatsappPlaceholder')} />
                    </div>
                    {errors.whatsapp && <p className="text-red-500 text-sm mt-1">{errors.whatsapp}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.phoneSecondary')} <span className="text-red-500">*</span></label>
                    <div className="relative group">
                      <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                        <Phone className={`w-5 h-5 ${errors.phone2 ? 'text-red-500' : 'text-gray-400 group-focus-within:text-amber-500'}`} />
                      </div>
                      <input type="tel" value={formData.phone2}
                        onChange={(e) => { setFormData({...formData, phone2: e.target.value}); setErrors({...errors, phone2: null}); }}
                        className={`w-full pr-12 pl-4 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.phone2 ? 'border-red-500' : 'border-gray-700'}`}
                        placeholder={t('register.phoneSecondaryPlaceholder')} maxLength={7} />
                    </div>
                    {errors.phone2 && <p className="text-red-500 text-sm mt-1">{errors.phone2}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.nationalNumber')} <span className="text-red-500">*</span></label>
                    <div className="relative group">
                      <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                        <User className={`w-5 h-5 ${errors.nationalNumber ? 'text-red-500' : 'text-gray-400 group-focus-within:text-amber-500'}`} />
                      </div>
                      <input type="text" value={formData.nationalNumber}
                        onChange={(e) => { setFormData({...formData, nationalNumber: e.target.value}); setErrors({...errors, nationalNumber: null}); }}
                        className={`w-full pr-12 pl-4 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.nationalNumber ? 'border-red-500' : 'border-gray-700'}`}
                        placeholder={t('register.nationalNumberPlaceholder')} />
                    </div>
                    {errors.nationalNumber && <p className="text-red-500 text-sm mt-1">{errors.nationalNumber}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.ownerType')} <span className="text-red-500">*</span></label>
                    <select value={formData.ownerType.toString()}
                      onChange={(e) => {
                        const selectedType = parseInt(e.target.value, 10);
                        setFormData((prev) => ({ ...prev, ownerType: selectedType }));
                      }}
                      className="w-full py-3 px-4 bg-white/5 border border-gray-700 rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white appearance-none cursor-pointer">
                      {Object.entries(OwnerTypeLabels).map(([value, label]) => (
                        <option key={value} value={value} className="bg-gray-900 text-white">{label}</option>
                      ))}
                    </select>
                  </motion.div>

                  <AnimatePresence>
                    {isCompany && (
                      <motion.div
                        key="license-upload-step1"
                        initial={{ opacity: 0, y: -10 }}
                        animate={{ opacity: 1, y: 0 }}
                        exit={{ opacity: 0, y: -10 }}
                        transition={{ duration: 0.2 }}
                      >
                        <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.owner.licenseImageLabel')} <span className="text-gray-400">({t('register.optional')})</span></label>
                        <div className="mb-2 text-xs text-gray-400">{t('register.owner.licenseHint')}</div>
                        <div onClick={() => fileInputLicenseRef.current?.click()}
                          className={`relative border-2 border-dashed rounded-xl p-6 text-center cursor-pointer transition-all ${idImagePreviews.license ? 'border-green-500 bg-green-500/10' : 'border-gray-700 hover:border-amber-500 hover:bg-white/5'}`}>
                          <input ref={fileInputLicenseRef} type="file" accept="image/*" onChange={(e) => handleImageUpload('license', e.target.files?.[0])} className="hidden" />
                          {idImagePreviews.license ? (
                            <div className="relative">
                              <Image src={idImagePreviews.license} alt={t('register.uploadLicenseAlt')} width={200} height={120} className="mx-auto rounded-lg object-cover" />
                              <button onClick={(e) => { e.stopPropagation(); setIdImages(prev => ({...prev, license: null})); setIdImagePreviews(prev => ({...prev, license: ''})); }}
                                className="absolute -top-2 -right-2 w-6 h-6 bg-red-500 rounded-full flex items-center justify-center hover:bg-red-600">
                                <X className="w-4 h-4 text-white" />
                              </button>
                            </div>
                          ) : (<><Camera className="w-12 h-12 text-gray-500 mx-auto mb-3" /><p className="text-gray-400">{t('register.owner.clickUploadLicense')}</p><p className="text-xs text-gray-500 mt-2">JPEG, PNG, JPG • {t('register.upTo5MB')}</p></>)}
                        </div>
                      </motion.div>
                    )}
                  </AnimatePresence>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.password')} <span className="text-red-500">*</span></label>
                    <div className="relative group">
                      <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                        <Lock className={`w-5 h-5 ${errors.password ? 'text-red-500' : 'text-gray-400 group-focus-within:text-amber-500'}`} />
                      </div>
                      <input type={showPassword ? "text" : "password"} value={formData.password}
                        onChange={(e) => { setFormData({...formData, password: e.target.value}); setErrors({...errors, password: null}); }}
                        className={`w-full pr-12 pl-12 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.password ? 'border-red-500' : 'border-gray-700'}`}
                        placeholder={t('register.passwordPlaceholder')} />
                      <button type="button" onClick={() => setShowPassword(!showPassword)} className="absolute inset-y-0 left-0 pl-3 flex items-center">
                        {showPassword ? <EyeOff className="w-5 h-5 text-gray-400" /> : <Eye className="w-5 h-5 text-gray-400" />}
                      </button>
                    </div>
                    {errors.password && <p className="text-red-500 text-sm mt-1">{errors.password}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.confirmPassword')} <span className="text-red-500">*</span></label>
                    <div className="relative group">
                      <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                        <Lock className={`w-5 h-5 ${errors.confirmPassword ? 'text-red-500' : 'text-gray-400 group-focus-within:text-amber-500'}`} />
                      </div>
                      <input type={showConfirmPassword ? "text" : "password"} value={formData.confirmPassword}
                        onChange={(e) => { setFormData({...formData, confirmPassword: e.target.value}); setErrors({...errors, confirmPassword: null}); }}
                        className={`w-full pr-12 pl-12 py-3 bg-white/5 border rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 focus:border-transparent text-white placeholder-gray-500 transition-all ${errors.confirmPassword ? 'border-red-500' : 'border-gray-700'}`}
                        placeholder={t('register.confirmPasswordPlaceholder')} />
                      <button type="button" onClick={() => setShowConfirmPassword(!showConfirmPassword)} className="absolute inset-y-0 left-0 pl-3 flex items-center">
                        {showConfirmPassword ? <EyeOff className="w-5 h-5 text-gray-400" /> : <Eye className="w-5 h-5 text-gray-400" />}
                      </button>
                      {formData.confirmPassword && (
                        <div className="absolute inset-y-0 left-12 flex items-center">
                          {formData.password === formData.confirmPassword ? <CheckCircle className="w-5 h-5 text-green-500" /> : <XCircle className="w-5 h-5 text-red-500" />}
                        </div>
                      )}
                    </div>
                    {errors.confirmPassword && <p className="text-red-500 text-sm mt-1">{errors.confirmPassword}</p>}
                  </motion.div>
                </>
              )}

              {step === 2 && (
                <>
                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.owner.frontIdLabel')} <span className="text-red-500">*</span></label>
                    <div onClick={() => fileInputFrontRef.current?.click()}
                      className={`relative border-2 border-dashed rounded-xl p-6 text-center cursor-pointer transition-all ${idImagePreviews.front ? 'border-green-500 bg-green-500/10' : errors.front ? 'border-red-500 bg-red-500/10' : 'border-gray-700 hover:border-amber-500 hover:bg-white/5'}`}>
                      <input ref={fileInputFrontRef} type="file" accept="image/*" onChange={(e) => handleImageUpload('front', e.target.files?.[0])} className="hidden" />
                      {idImagePreviews.front ? (
                        <div className="relative">
                          <Image src={idImagePreviews.front} alt={t('register.uploadFrontAlt')} width={200} height={120} className="mx-auto rounded-lg object-cover" />
                          <button onClick={(e) => { e.stopPropagation(); setIdImages(prev => ({...prev, front: null})); setIdImagePreviews(prev => ({...prev, front: ''})); }}
                            className="absolute -top-2 -right-2 w-6 h-6 bg-red-500 rounded-full flex items-center justify-center hover:bg-red-600">
                            <X className="w-4 h-4 text-white" />
                          </button>
                        </div>
                      ) : (<><Camera className="w-12 h-12 text-gray-500 mx-auto mb-3" /><p className="text-gray-400">{t('register.owner.clickUploadImage')}</p><p className="text-xs text-gray-500 mt-2">JPEG, PNG, JPG • {t('register.upTo5MB')}</p></>)}
                    </div>
                    {errors.front && <p className="text-red-500 text-sm mt-1">{errors.front}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp}>
                    <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.owner.backIdLabel')} <span className="text-red-500">*</span></label>
                    <div onClick={() => fileInputBackRef.current?.click()}
                      className={`relative border-2 border-dashed rounded-xl p-6 text-center cursor-pointer transition-all ${idImagePreviews.back ? 'border-green-500 bg-green-500/10' : errors.back ? 'border-red-500 bg-red-500/10' : 'border-gray-700 hover:border-amber-500 hover:bg-white/5'}`}>
                      <input ref={fileInputBackRef} type="file" accept="image/*" onChange={(e) => handleImageUpload('back', e.target.files?.[0])} className="hidden" />
                      {idImagePreviews.back ? (
                        <div className="relative">
                          <Image src={idImagePreviews.back} alt={t('register.uploadBackAlt')} width={200} height={120} className="mx-auto rounded-lg object-cover" />
                          <button onClick={(e) => { e.stopPropagation(); setIdImages(prev => ({...prev, back: null})); setIdImagePreviews(prev => ({...prev, back: ''})); }}
                            className="absolute -top-2 -right-2 w-6 h-6 bg-red-500 rounded-full flex items-center justify-center hover:bg-red-600">
                            <X className="w-4 h-4 text-white" />
                          </button>
                        </div>
                      ) : (<><Camera className="w-12 h-12 text-gray-500 mx-auto mb-3" /><p className="text-gray-400">{t('register.owner.clickUploadImage')}</p><p className="text-xs text-gray-500 mt-2">JPEG, PNG, JPG • {t('register.upTo5MB')}</p></>)}
                    </div>
                    {errors.back && <p className="text-red-500 text-sm mt-1">{errors.back}</p>}
                  </motion.div>

                  <motion.div variants={fadeInUp} className="flex items-center gap-2">
                    <input type="checkbox" id="terms" checked={formData.agreeTerms}
                      onChange={(e) => setFormData({...formData, agreeTerms: e.target.checked})}
                      className="w-4 h-4 rounded border-gray-600 bg-white/5 text-amber-500 focus:ring-amber-500" required />
                    <label htmlFor="terms" className="text-sm text-gray-300">
                      {t('register.agreeTerms')} <Link href="/terms" className="text-amber-400 hover:text-amber-300">{t('register.termsOfUse')}</Link> {t('register.and')} <Link href="/privacy" className="text-amber-400 hover:text-amber-300">{t('register.privacyPolicy')}</Link>
                    </label>
                  </motion.div>
                </>
              )}

              <motion.div variants={fadeInUp} className="flex gap-3 pt-4">
                {step === 1 ? (
                  <>
                    <button type="button" onClick={() => router.push('/auth/choose-role')}
                      className="flex-1 py-3 px-4 bg-white/5 border border-gray-700 rounded-xl text-gray-300 hover:bg-white/10 transition-colors">{t('register.cancel')}</button>
                    <button type="submit"
                      className="flex-1 bg-gradient-to-r from-amber-500 to-amber-600 text-white py-3 px-4 rounded-xl font-medium hover:from-amber-600 hover:to-amber-700 transition-all">{t('register.next')}</button>
                  </>
                ) : (
                  <>
                    <button type="button" onClick={() => setStep(1)}
                      className="flex-1 py-3 px-4 bg-white/5 border border-gray-700 rounded-xl text-gray-300 hover:bg-white/10 transition-colors">{t('register.previous')}</button>
                    <button type="submit" disabled={isLoading || !formData.agreeTerms}
                      className="flex-1 bg-gradient-to-r from-amber-500 to-amber-600 text-white py-3 px-4 rounded-xl font-medium hover:from-amber-600 hover:to-amber-700 transition-all disabled:opacity-50 disabled:cursor-not-allowed">
                      {isLoading ? (<div className="flex items-center justify-center gap-2"><Loader2 className="w-5 h-5 animate-spin" /><span>{t('register.registering')}</span></div>) : t('register.createAccount')}
                    </button>
                  </>
                )}
              </motion.div>
            </motion.form>
          </div>
        </motion.div>
      </motion.div>

      <AnimatePresence>
        {showOtpModal && (
          <motion.div initial={{ opacity: 0 }} animate={{ opacity: 1 }} exit={{ opacity: 0 }}
            className="fixed inset-0 bg-black/60 backdrop-blur-sm flex items-center justify-center p-4 z-50">
            <motion.div initial={{ scale: 0.9, y: 20 }} animate={{ scale: 1, y: 0 }} exit={{ scale: 0.9, y: 20 }}
              className="bg-gray-900 border border-white/10 rounded-2xl w-full max-w-md p-6 shadow-2xl">
              <div className="text-center mb-6">
                <div className="w-16 h-16 bg-amber-500/20 rounded-full flex items-center justify-center mx-auto mb-3">
                  <Shield className="w-8 h-8 text-amber-500" />
                </div>
                <h2 className="text-xl font-bold text-white">{t('register.otpTitle')}</h2>
                <p className="text-gray-400 text-sm mt-1">{t('register.otpSentTo')}</p>
                <p className="text-amber-400 font-medium text-sm">{formData.email}</p>
              </div>
              <div className="mb-6">
                <label className="block text-sm font-medium text-gray-300 mb-2">{t('register.otpLabel')}</label>
                <div className="relative">
                  <div className="absolute inset-y-0 right-0 pr-3 flex items-center pointer-events-none">
                    <KeyRound className="w-5 h-5 text-gray-400" />
                  </div>
                  <input type="text" value={otpCode} maxLength={6}
                    onChange={(e) => setOtpCode(e.target.value)}
                    className="w-full pr-12 pl-4 py-3 bg-white/5 border border-gray-700 rounded-xl focus:outline-none focus:ring-2 focus:ring-amber-500 text-white text-center tracking-[0.5em] text-xl"
                    placeholder="------" />
                </div>
              </div>
              <div className="flex gap-3">
                <button onClick={handleVerifyOTP} disabled={isLoading || !otpCode}
                  className="flex-1 bg-gradient-to-r from-amber-500 to-amber-600 text-white py-3 rounded-xl font-medium hover:from-amber-600 hover:to-amber-700 disabled:opacity-50 disabled:cursor-not-allowed flex items-center justify-center gap-2">
                  {isLoading ? <><Loader2 className="w-5 h-5 animate-spin" /><span>{t('register.verifying')}</span></> : t('register.verify')}
                </button>
              </div>
              <button onClick={handleResendOTP} disabled={isLoading}
                className="w-full text-center text-amber-400 hover:text-amber-300 text-sm mt-3 disabled:opacity-50">
                {t('register.resendOtp')}
              </button>
            </motion.div>
          </motion.div>
        )}
      </AnimatePresence>
    </div>
  );
}