import 'dart:io';
import 'package:flutter/material.dart';
import 'package:camera/camera.dart';
import 'package:image/image.dart' as img;
import 'package:shared_preferences/shared_preferences.dart';
import 'yolo_detector.dart';
import 'preview_screen.dart';
import '../utils/translations.dart';

class CameraScreen extends StatefulWidget {
  const CameraScreen({super.key});

  @override
  State<CameraScreen> createState() => _CameraScreenState();
}

class _CameraScreenState extends State<CameraScreen> {
  CameraController? _controller;
  Future<void>? _initializeControllerFuture;
  late YOLODetector _yoloDetector;
  Color _bulletColor = Colors.blue; // default fallback

  // The variable to hold the number of bullets entered by the user (default set to 5)
  int _numberOfBullets = 5;
  String _selectedLanguage = 'English';

  // Brightness and contrast controls
  double _brightness = 0.0; // -1.0 to 1.0
  double _contrast = 1.0;    // 0.0 to 2.0

  @override
  void initState() {
    super.initState();
    _loadBulletCount();
    _initCamera();
    _loadBulletColor(); // 👈
    _loadLanguage();
    _loadBrightnessContrast();
    YOLODetector.create().then((detector) => _yoloDetector = detector);
  }

  Future<void> _loadBrightnessContrast() async {
    final prefs = await SharedPreferences.getInstance();
    setState(() {
      _brightness = prefs.getDouble('brightness') ?? 0.0;
      _contrast = prefs.getDouble('contrast') ?? 1.0;
    });
  }

  Future<void> _saveBrightnessContrast() async {
    final prefs = await SharedPreferences.getInstance();
    await prefs.setDouble('brightness', _brightness);
    await prefs.setDouble('contrast', _contrast);
  }

  String _t(String key) {
    return AppTranslations.translate(key, _selectedLanguage);
  }

  Future<void> _loadLanguage() async {
    final prefs = await SharedPreferences.getInstance();
    setState(() {
      _selectedLanguage = prefs.getString('language') ?? 'English';
    });
  }

  Future<void> _loadBulletColor() async {
  final prefs = await SharedPreferences.getInstance();
  final selected = prefs.getString('bulletColor') ?? 'Blue';
  setState(() {
    _bulletColor = _getColorFromName(selected);
  });
}

Color _getColorFromName(String colorName) {
  switch (colorName.toLowerCase()) {
    case 'blue':
      return Colors.blue;
    case 'green':
      return Colors.green;
    case 'yellow':
      return Colors.yellow;
    case 'purple':
      return Colors.purple;
    case 'cyan':
      return Colors.cyan;
    case 'teal':
      return Colors.teal;
    case 'indigo':
      return Colors.indigo;
    case 'pink':
      return Colors.pink;
    case 'lime':
      return Colors.lime;
    case 'mint':
      return Colors.greenAccent;
    case 'orange':
      return Colors.orange;
    case 'violet':
      return Colors.deepPurpleAccent;
    case 'amber':
      return Colors.amber;
    case 'brown':
      return Colors.brown;
    case 'grey':
      return Colors.grey;
    default:
      return Colors.blue; // fallback
  }
}



  Future<void> _loadBulletCount() async {
    final prefs = await SharedPreferences.getInstance();
    setState(() {
      _numberOfBullets = prefs.getInt('bulletCount') ?? 5; // Load stored bullet count, default to 5
    });
  }

  Future<void> _saveBulletCount(int count) async {
    final prefs = await SharedPreferences.getInstance();
    await prefs.setInt('bulletCount', count); // Save the bullet count
  }

  Future<void> _initCamera() async {
    final cameras = await availableCameras();
    final camera = cameras.first;

    _controller = CameraController(
      camera,
      ResolutionPreset.high,
      enableAudio: false,
    );

    _initializeControllerFuture = _controller!.initialize();
    if (mounted) setState(() {});
  }

  @override
  void dispose() {
    _controller?.dispose();
    _yoloDetector.close();
    super.dispose();
  }

  Future<void> _takePicture() async {
    try {
      await _initializeControllerFuture;
      final file = await _controller!.takePicture();
      final imageFile = File(file.path);
      final imageBytes = await imageFile.readAsBytes();
      final decodedImage = img.decodeImage(imageBytes)!;

      // Crop center square from full image
      final width = decodedImage.width;
      final height = decodedImage.height;
      final size = width < height ? width : height;
      final offsetX = (width - size) ~/ 2;
      final offsetY = (height - size) ~/ 2;

      final cropped = img.copyCrop(
        decodedImage,
        x: offsetX,
        y: offsetY,
        width: size,
        height: size,
      );

      final croppedFile = File('${file.path}_square.jpg')
        ..writeAsBytesSync(img.encodeJpg(cropped));

      final detections = await _yoloDetector.detect(croppedFile);

      if (!mounted) return;

      Navigator.push(
        context,
        MaterialPageRoute(
          builder: (context) => PreviewScreen(
            imagePath: croppedFile.path,
            detections: detections,
            bulletCount: _numberOfBullets,  // Pass the bullet count
          ),
        ),
      );
    } catch (e) {
      print('Error taking picture: $e');
    }
  }

  // Method to show the dialog for selecting the number of bullets
  void _showBulletDialog() {
    showDialog(
      context: context,
      builder: (BuildContext context) {
        return Dialog(
          backgroundColor: Colors.grey[900],
          shape: RoundedRectangleBorder(borderRadius: BorderRadius.circular(16)),
          child: Padding(
            padding: const EdgeInsets.all(24.0), // Increased padding for better spacing
            child: Column(
              mainAxisSize: MainAxisSize.min,
              crossAxisAlignment: CrossAxisAlignment.center,
              children: [
                Padding(
                  padding: const EdgeInsets.only(bottom: 16.0),
                  child: Text(
                    _t('Select the number of bullets'),
                    textAlign: TextAlign.center,
                    style: const TextStyle(
                      color: Colors.white,
                      fontSize: 20,
                      fontWeight: FontWeight.bold,
                    ),
                  ),
                ),
                Center(
                  child: Wrap(
                    alignment: WrapAlignment.center, 
                    spacing: 12,
                    runSpacing: 12,
                    children: List.generate(5, (index) {
                      final bulletNum = index + 1;
                      final isSelected = _numberOfBullets == bulletNum;
                      return ElevatedButton(
                        onPressed: () {
                          _selectBulletCount(bulletNum);
                          Navigator.of(context).pop();
                        },
                        style: ElevatedButton.styleFrom(
                          backgroundColor:
                              isSelected ? _bulletColor : Colors.grey[800],
                          foregroundColor: Colors.white,
                          padding:
                              const EdgeInsets.symmetric(horizontal: 20, vertical: 12),
                          shape: RoundedRectangleBorder(
                            borderRadius: BorderRadius.circular(12),
                          ),
                        ),
                        child: Text(
                          '$bulletNum',
                          style: TextStyle(
                            fontSize: 18,
                            fontWeight:
                                isSelected ? FontWeight.bold : FontWeight.normal,
                          ),
                        ),
                      );
                    }),
                  ),
                ),
              ],
            ),
          ),
        );
      },
    );
  }


  void _showHelpDialog() {
    showDialog(
      context: context,
      builder: (BuildContext context) {
        return Dialog(
          backgroundColor: Colors.grey[900],
          shape: RoundedRectangleBorder(borderRadius: BorderRadius.circular(16)),
          child: Padding(
            padding: const EdgeInsets.all(24.0),
            child: Column(
              mainAxisSize: MainAxisSize.min,
              crossAxisAlignment: CrossAxisAlignment.start,
              children: [
                Text(
                  _t('Help'),
                  style: const TextStyle(color: Colors.white, fontSize: 24, fontWeight: FontWeight.bold),
                ),
                const SizedBox(height: 20),
                _buildHelpItem(Icons.camera_alt, _t('Tap camera button to capture and detect bullets')),
                const SizedBox(height: 12),
                _buildHelpItem(Icons.filter_9_plus, _t('Bottom left number: Set expected bullet count (1-5)')),
                const SizedBox(height: 12),
                _buildHelpItem(Icons.brightness_6, _t('Sun icon: Adjust brightness and contrast for better visibility')),
                const SizedBox(height: 12),
                _buildHelpItem(Icons.crop_square, _t('Align target inside the square frame')),
                const SizedBox(height: 20),
                Center(
                  child: ElevatedButton(
                    onPressed: () => Navigator.of(context).pop(),
                    style: ElevatedButton.styleFrom(
                      backgroundColor: _bulletColor,
                      foregroundColor: Colors.white,
                    ),
                    child: Text(_t('Got it!')),
                  ),
                ),
              ],
            ),
          ),
        );
      },
    );
  }

  Widget _buildHelpItem(IconData icon, String text) {
    return Row(
      crossAxisAlignment: CrossAxisAlignment.start,
      children: [
        Icon(icon, color: _bulletColor, size: 20),
        const SizedBox(width: 12),
        Expanded(
          child: Text(
            text,
            style: const TextStyle(color: Colors.white, fontSize: 14),
          ),
        ),
      ],
    );
  }

  void _showBrightnessContrastDialog() {
    showDialog(
      context: context,
      builder: (BuildContext context) {
        return StatefulBuilder(
          builder: (context, setDialogState) {
            return Dialog(
              backgroundColor: Colors.grey[900],
              shape: RoundedRectangleBorder(borderRadius: BorderRadius.circular(16)),
              child: Padding(
                padding: const EdgeInsets.all(24.0),
                child: Column(
                  mainAxisSize: MainAxisSize.min,
                  children: [
                    Text(
                      _t('Brightness & Contrast'),
                      style: const TextStyle(color: Colors.white, fontSize: 20, fontWeight: FontWeight.bold),
                    ),
                    const SizedBox(height: 24),
                    // Brightness slider
                    Column(
                      crossAxisAlignment: CrossAxisAlignment.start,
                      children: [
                        Text(
                          '${_t('Brightness')}: ${(_brightness * 100).toInt()}',
                          style: const TextStyle(color: Colors.white, fontSize: 16),
                        ),
                        Slider(
                          value: _brightness,
                          min: -1.0,
                          max: 1.0,
                          divisions: 40,
                          activeColor: _bulletColor,
                          onChanged: (value) {
                            setState(() {
                              _brightness = value;
                            });
                            setDialogState(() {});
                            _saveBrightnessContrast();
                          },
                        ),
                      ],
                    ),
                    const SizedBox(height: 16),
                    // Contrast slider
                    Column(
                      crossAxisAlignment: CrossAxisAlignment.start,
                      children: [
                        Text(
                          '${_t('Contrast')}: ${(_contrast * 100).toInt()}%',
                          style: const TextStyle(color: Colors.white, fontSize: 16),
                        ),
                        Slider(
                          value: _contrast,
                          min: 0.5,
                          max: 2.0,
                          divisions: 30,
                          activeColor: _bulletColor,
                          onChanged: (value) {
                            setState(() {
                              _contrast = value;
                            });
                            setDialogState(() {});
                            _saveBrightnessContrast();
                          },
                        ),
                      ],
                    ),
                    const SizedBox(height: 16),
                    Row(
                      mainAxisAlignment: MainAxisAlignment.spaceEvenly,
                      children: [
                        ElevatedButton(
                          onPressed: () {
                            setState(() {
                              _brightness = 0.0;
                              _contrast = 1.0;
                            });
                            setDialogState(() {});
                            _saveBrightnessContrast();
                          },
                          style: ElevatedButton.styleFrom(
                            backgroundColor: Colors.grey[800],
                            foregroundColor: Colors.white,
                          ),
                          child: Text(_t('Reset')),
                        ),
                        ElevatedButton(
                          onPressed: () {
                            Navigator.of(context).pop();
                          },
                          style: ElevatedButton.styleFrom(
                            backgroundColor: _bulletColor,
                            foregroundColor: Colors.white,
                          ),
                          child: Text(_t('Done')),
                        ),
                      ],
                    ),
                  ],
                ),
              ),
            );
          },
        );
      },
    );
  }

  // Update the bullet count and save it
  void _selectBulletCount(int number) {
    setState(() {
      _numberOfBullets = number; // Update the selected number
    });
    _saveBulletCount(number); // Save the selected number
  }

  @override
  Widget build(BuildContext context) {
    if (_controller == null || _initializeControllerFuture == null) {
      return const Scaffold(
        body: Center(child: CircularProgressIndicator()),
      );
    }

    return Scaffold(
      extendBody: true,
      appBar: AppBar(
        backgroundColor: Colors.grey[900],
        title: Text(
          _t("Align the target inside the square"),
          style: const TextStyle(
            fontSize: 16,
            fontWeight: FontWeight.bold,
            color: Colors.white,
          ),
        ),
        leading: IconButton(
          icon: const Icon(Icons.arrow_back, color: Colors.white),
          onPressed: () => Navigator.pop(context),
        ),
        centerTitle: true,
        actions: [
          IconButton(
            icon: const Icon(Icons.help_outline, color: Colors.white),
            onPressed: _showHelpDialog,
          ),
          const SizedBox(width: 16),
        ],
      ),
      bottomNavigationBar: BottomAppBar(
        color: Colors.grey[900],
        shape: const CircularNotchedRectangle(),
        notchMargin: 8.0,
        child: SizedBox(
          height: 60,
          child: Row(
            children: [
              // Left - Bullet count (just number)
              Expanded(
                child: Align(
                  alignment: Alignment.centerLeft,
                  child: Padding(
                    padding: const EdgeInsets.only(left: 16.0),
                    child: InkWell(
                      onTap: _showBulletDialog,
                      child: Text(
                        '$_numberOfBullets',
                        style: const TextStyle(
                          fontSize: 24,
                          fontWeight: FontWeight.bold,
                          color: Colors.white,
                        ),
                      ),
                    ),
                  ),
                ),
              ),
              // Center space for FAB
              const SizedBox(width: 80),
              // Right section - Brightness
              Expanded(
                child: Row(
                  mainAxisAlignment: MainAxisAlignment.end,
                  children: [
                    IconButton(
                      icon: const Icon(Icons.brightness_6, color: Colors.white, size: 24),
                      onPressed: _showBrightnessContrastDialog,
                    ),
                  ],
                ),
              ),
            ],
          ),
        ),
      ),
      floatingActionButtonLocation: FloatingActionButtonLocation.centerDocked,
      floatingActionButton: FloatingActionButton(
        backgroundColor: Colors.white,
        onPressed: _takePicture,
        child: const Icon(Icons.camera_alt, color: Colors.black),
      ),
      body: FutureBuilder(
        future: _initializeControllerFuture,
        builder: (context, snapshot) {
          if (snapshot.connectionState == ConnectionState.done) {
            final screenSize = MediaQuery.of(context).size;
          return Stack(
            children: [
              Positioned.fill(
                child: Container(
                  color: Colors.black, // covers any empty space
                  child: FittedBox(
                    fit: BoxFit.cover,
                    child: SizedBox(
                      width: _controller!.value.previewSize!.height,
                      height: _controller!.value.previewSize!.width,
                      child: CameraPreview(_controller!),
                    ),
                  ),
                ),
              ),
              Center(
                child: Container(
                  width: screenSize.width * 0.9,
                  height: screenSize.width * 0.9,
                  decoration: BoxDecoration(
                    border: Border.all(
                      color: _bulletColor.withValues(alpha: 0.9),
                      width: 3,
                    ),
                  ),
                ),
              ),
            ],
          );
          } else {
            return const Center(child: CircularProgressIndicator());
          }
        },
      ),
    );
  }
}