JJ_MKVMaker/lib/ui/ai_dialog.dart

import 'package:flutter/material.dart';

import '../app/app_controller.dart';
import '../core/languages.dart';
import '../core/models.dart';
import '../services/model_store.dart';
import 'theme.dart';

/// AI 자막 만들기 설정 창
/// [targets] 를 주면 그 동영상들만 (탐색기 "자막 만들기" 메뉴)
Future<void> showAiDialog(BuildContext context, AppController c, VideoItem current,
    {List<VideoItem>? targets}) async {
  final list = targets ?? [current];
  // 환경 설정에서 "매번 묻기" 를 끄면 마지막 설정으로 바로 시작
  if (!c.settings.askAiOptions) {
    await c.generateAiSubtitles(list, c.aiOptions);
    return;
  }
  final result = await showDialog<(AiOptions, bool)>(
    context: context,
    builder: (_) => _AiDialog(c: c, videoCount: targets == null ? c.videos.length : 1),
  );
  if (result == null) return;
  final (opts, all) = result;
  await c.generateAiSubtitles(all ? List.of(c.videos) : list, opts);
}

class _AiDialog extends StatefulWidget {
  final AppController c;
  final int videoCount;
  const _AiDialog({required this.c, required this.videoCount});

  @override
  State<_AiDialog> createState() => _AiDialogState();
}

class _AiDialogState extends State<_AiDialog> {
  late AiOptions _o = widget.c.aiOptions;
  bool _all = false;
  final Map<String, bool> _installed = {};

  @override
  void initState() {
    super.initState();
    for (final m in [...whisperModels, nllbModel]) {
      widget.c.services.models.isInstalled(m).then((ok) {
        if (mounted) setState(() => _installed[m.id] = ok);
      });
    }
  }

  Widget _status(ModelSpec m) {
    final ok = _installed[m.id];
    if (ok == null) return const SizedBox();
    return Text(ok ? '설치됨' : '처음 사용 시 내려받기 ${m.sizeLabel}',
        style: TextStyle(fontSize: 11, color: ok ? JjColors.success : JjColors.textDim));
  }

  @override
  Widget build(BuildContext context) {
    final extra = languages.where((l) => !_o.targets.contains(l)).toList();
    return AlertDialog(
      title: const Text('AI 자막 만들기'),
      content: SizedBox(
        width: 520,
        child: SingleChildScrollView(
          child: Column(
            crossAxisAlignment: CrossAxisAlignment.start,
            mainAxisSize: MainAxisSize.min,
            children: [
              const Text('영상의 음성을 인식하고 선택한 언어로 번역합니다. 모든 처리는 이 PC 에서 이루어집니다.',
                  style: TextStyle(fontSize: 12, color: JjColors.textDim)),
              const SizedBox(height: 16),
              _label('원어 (영상 속 언어)'),
              DropdownButton<Language>(
                value: _o.source,
                isExpanded: true,
                items: [
                  const DropdownMenuItem(value: undetermined, child: Text('자동 감지')),
                  for (final l in languages)
                    DropdownMenuItem(value: l, child: Text('${l.name} (${l.code})')),
                ],
                onChanged: (l) => setState(() => _o = _o.copyWith(source: l)),
              ),
              const SizedBox(height: 16),
              _label('만들 자막 언어 (파일명_언어코드.srt)'),
              Wrap(
                spacing: 6,
                runSpacing: 6,
                children: [
                  for (final l in _o.targets)
                    InputChip(
                      label: Text('${l.name} (${l.code})'),
                      onDeleted: _o.targets.length > 1
                          ? () => setState(() => _o = _o.copyWith(targets: {..._o.targets}..remove(l)))
                          : null,
                    ),
                  PopupMenuButton<Language>(
                    tooltip: '언어 추가',
                    itemBuilder: (_) => [
                      for (final l in extra)
                        PopupMenuItem(value: l, child: Text('${l.name} (${l.code})')),
                    ],
                    onSelected: (l) => setState(() => _o = _o.copyWith(targets: {..._o.targets, l})),
                    child: const Chip(avatar: Icon(Icons.add, size: 16), label: Text('언어 추가')),
                  ),
                ],
              ),
              const SizedBox(height: 16),
              _label('음성인식 모델'),
              RadioGroup<String>(
                groupValue: _o.whisper.id,
                onChanged: (id) => setState(() => _o =
                    _o.copyWith(whisper: whisperModels.firstWhere((m) => m.id == id))),
                child: Column(children: [
                  for (final m in whisperModels)
                    RadioListTile<String>(
                      dense: true,
                      contentPadding: EdgeInsets.zero,
                      value: m.id,
                      title: Text(m.label),
                      subtitle: _status(m),
                    ),
                ]),
              ),
              Row(children: [
                const Text('번역 모델: NLLB-200 (로컬)  ', style: TextStyle(fontSize: 13)),
                _status(nllbModel),
              ]),
              if (widget.videoCount > 1) ...[
                const SizedBox(height: 8),
                CheckboxListTile(
                  dense: true,
                  contentPadding: EdgeInsets.zero,
                  value: _all,
                  onChanged: (v) => setState(() => _all = v ?? false),
                  title: Text('목록의 동영상 ${widget.videoCount}개 모두'),
                ),
              ],
              const SizedBox(height: 8),
              const Text('※ 시간이 오래 걸립니다 (영상 1시간 ≈ 음성인식 15~30분 + 언어당 번역 10~20분)',
                  style: TextStyle(fontSize: 11, color: JjColors.textDim)),
            ],
          ),
        ),
      ),
      actions: [
        TextButton(onPressed: () => Navigator.pop(context), child: const Text('취소')),
        FilledButton.icon(
          onPressed: () => Navigator.pop(context, (_o, _all)),
          icon: const Icon(Icons.auto_awesome, size: 18),
          label: const Text('시작'),
        ),
      ],
    );
  }

  Widget _label(String t) => Padding(
        padding: const EdgeInsets.only(bottom: 4),
        child: Text(t, style: const TextStyle(fontSize: 13, fontWeight: FontWeight.w600)),
      );
}