JJ_MKVMaker/lib/ui/ai_image_page.dart

import 'dart:async';
import 'dart:convert';
import 'dart:io';

import 'package:flutter/material.dart';
import 'package:path/path.dart' as p;

import '../app/ai_jobs.dart';
import '../app/ai_local.dart';
import '../app/ai_local_engine.dart';
import '../app/app_controller.dart';
import '../core/reader_sources.dart';
import '../core/sd_cli.dart';
import '../l10n/tr.dart';
import '../services/image_ai.dart';
import 'ai_settings.dart';
import 'app_actions.dart';
import 'browser_page.dart' show browserRouteObserver;
import 'reader_page.dart';
import 'theme.dart';
import 'notice_band.dart';
import 'jj_field.dart';

/// 123: AI 그림 - 글 → 그림 · 그림 → 그림. 기기 안 (SD1.5 + LCM) 이 기본이고, 사용자가 추가한 서버 · 서비스도 같은 화면에서.
/// 결과는 새 파일로 저장하고 (원본을 덮지 않음), 옆에 프롬프트 · 시드 · 모델을 json 으로 남긴다.
class AiImagePage extends StatefulWidget {
  const AiImagePage({super.key, required this.c, this.initImage});
  final AppController c;

  /// 보기 화면 · 탐색기의 [AI 그림으로] - 그림 → 그림의 바탕 그림
  final String? initImage;

  static const routeName = 'aiimage';

  static Future<void> open(NavigatorState nav, {required AppController c, String? initImage}) async {
    var found = false;
    if (initImage == null) {
      nav.popUntil((r) {
        if (r.settings.name == routeName) found = true;
        return found || r.isFirst;
      });
      if (found) return;
    }
    await nav.push(MaterialPageRoute<void>(
      settings: const RouteSettings(name: routeName),
      builder: (_) => AiImagePage(c: c, initImage: initImage),
    ));
  }

  @override
  State<AiImagePage> createState() => _AiImagePageState();
}

/// 결과를 저장할 폴더 (비어 있으면 사진 폴더의 JJ_MKVMaker_AI)
String aiSaveDir(String setting) {
  if (setting.isNotEmpty) return setting;
  if (Platform.isAndroid) return '/storage/emulated/0/Pictures/JJ_MKVMaker_AI';
  final home = Platform.environment['USERPROFILE'] ?? Platform.environment['HOME'] ?? Directory.systemTemp.path;
  return p.join(home, 'Pictures', 'JJ_MKVMaker_AI');
}

class _AiImagePageState extends State<AiImagePage> with RouteAware {
  AppController get c => widget.c;
  late final _prompt = TextEditingController(text: c.settings.aiImage.prompt);
  final _promptFocus = FocusNode();
  late final _negative = TextEditingController(text: c.settings.aiImage.negative);
  final _seed = TextEditingController(text: '-1');
  late String? _init = widget.initImage;
  /// 159: 지난 결과를 되살린다 (설정의 목록 + 그림 옆 json 의 시드). 지워진 파일은 빼고 보여 준다
  late final _results = <GeneratedImage>[
    for (final path in c.settings.aiImage.recent)
      if (File(path).existsSync()) GeneratedImage(path, _seedOf(path)),
  ];
  String? _error;

  static int _seedOf(String path) {
    try {
      final j = jsonDecode(File('${p.withoutExtension(path)}.json').readAsStringSync()) as Map;
      return (j['seed'] as num?)?.toInt() ?? -1;
    } catch (_) {
      return -1;
    }
  }

  /// 목록을 설정에 남긴다 (최근 100장)
  void _saveRecent() {
    final paths = [for (final g in _results) g.path].take(100).toList();
    c.updateSettings((x) => x.aiImage.recent = paths);
  }

  /// 150: 엔진 원문 ([자세히] 를 눌러야 보임) · 취소 (회색 한 줄, 오류 아님)
  String _detail = '';
  bool _showDetail = false;
  bool _cancelled = false;

  @override
  void didChangeDependencies() {
    super.didChangeDependencies();
    final route = ModalRoute.of(context);
    if (route != null) browserRouteObserver.subscribe(this, route);
  }

  /// 152: 다른 화면으로 가면 글 칸의 포커스를 풀어 둔다 (돌아올 때 자판이 저절로 올라와 결과를 가리지 않게)
  @override
  void didPushNext() => FocusManager.instance.primaryFocus?.unfocus();
  String _step = '';

  AiStore get store => AiStore.instance ??= AiStore(Directory.systemTemp.path);

  @override
  void dispose() {
    browserRouteObserver.unsubscribe(this);
    _prompt.dispose();
    _promptFocus.dispose();
    _negative.dispose();
    _seed.dispose();
    super.dispose();
  }

  AiService? get _service => c.settings.aiImage.services.where((x) => x.id == c.settings.aiImage.engine).firstOrNull;

  bool get _local => _service == null;

  bool get _modelsReady => store.sd15Paths(taesd: false) != null;

  Future<void> _generate() async {
    final s = c.settings;
    if (_prompt.text.trim().isEmpty) {
      // 159: 아래 오류 줄은 화면 밖일 수 있다 - 바로 보이게 알리고 글 칸으로
      setState(() => _error = tr('그릴 내용을 적어 주세요'));
      ScaffoldMessenger.maybeOf(context)?.showSnackBar(SnackBar(content: Text(tr('그릴 내용을 적어 주세요'))));
      _promptFocus.requestFocus();
      return;
    }
    if (_local && !_modelsReady) {
      await openAiModels(context, c);
      if (!_modelsReady) return;
    }
    await c.updateSettings((x) => x
      ..aiImage.negative = _negative.text
      ..aiImage.prompt = _prompt.text);
    final req = ImageGenRequest(
      prompt: _prompt.text.trim(),
      negative: _negative.text.trim(),
      width: s.aiImage.width,
      height: s.aiImage.height,
      steps: s.aiImage.steps,
      cfg: s.aiImage.cfg,
      seed: int.tryParse(_seed.text.trim()) ?? -1,
      count: s.aiImage.count,
      initImage: _init,
      strength: s.aiImage.strength,
      model: _local ? 'sd15-lcm' : (_service!.model.isEmpty ? _service!.kind : _service!.model),
    );
    setState(() {
      _error = null;
      _detail = '';
      _showDetail = false;
      _cancelled = false;
      _step = '';
    });
    try {
      final ImageAiEngine engine;
      if (_local) {
        final device = await pickAiDevice(c, store, onStep: (t) => mounted ? setState(() => _step = t) : null);
        if (device == null) throw ImageAiException(tr('AI 엔진이 없습니다. 환경 설정 > AI 그림에서 받아 주세요.'));
        engine = LocalSdEngine(device: device, models: store.sd15Paths(taesd: s.aiImage.taesd)!);
      } else {
        engine = RemoteImageEngine(_service!);
      }
      final made = await AiJobs.instance.run(engine, req, outDir: aiSaveDir(s.aiImage.saveDir));
      if (!mounted) return;
      setState(() => _results.insertAll(0, made));
      _saveRecent();
    } on ImageAiException catch (e) {
      if (mounted) {
        setState(() {
          _cancelled = e.cancelled;
          _error = e.message;
          _detail = e.detail;
        });
      }
    } catch (e) {
      if (mounted) setState(() => _error = '$e');
    }
  }

  Future<void> _pickInit() async {
    final files = await c.services.storage.pickFiles(title: tr('바탕 그림 고르기'), extensions: c.settings.reader.imageExts);
    if (files.isNotEmpty && mounted) setState(() => _init = files.first);
  }

  /// 결과 옆의 json 을 읽어 같은 조건으로 (시드 그대로)
  void _reuse(GeneratedImage g) {
    try {
      final j = File('${p.withoutExtension(g.path)}.json').readAsStringSync();
      final r = ImageGenRequest.fromJson(Map<String, Object?>.from(jsonDecode(j) as Map));
      setState(() {
        _prompt.text = r.prompt;
        _negative.text = r.negative;
        _seed.text = '${g.seed}';
        _init = r.initImage;
      });
    } catch (_) {
      setState(() => _seed.text = '${g.seed}');
    }
  }

  @override
  Widget build(BuildContext context) {
    return ListenableBuilder(
      listenable: Listenable.merge([c, AiJobs.instance, store]),
      builder: (context, _) {
        final s = c.settings;
        final jobs = AiJobs.instance;
        final compact = isCompact(context);
        return Scaffold(
          body: SwipeNav(
            current: 'aiimage',
            child: Column(children: [
              AppTopBar(
                nav: const AppNavButtons(onAiPage: true),
                actions: AppActions(c: c),
                middle: Text(tr('AI 그림'), style: const TextStyle(fontSize: 16, fontWeight: FontWeight.w600)),
              ),
              Expanded(
                child: ListView(padding: EdgeInsets.all(compact ? 8 : 16), children: [
                  _engineRow(),
                  if (!_local)
                    // 인터넷 서비스 · 자기 서버를 쓸 때는 늘 알린다
                    _notice(Icons.public, JjColors.warning,
                        trf('그림 · 프롬프트가 기기 밖으로 나갑니다 ({0})', [_service!.label])),
                  if (_local && !_modelsReady)
                    _notice(Icons.download_outlined, JjColors.accent,
                        trf('기기 안에서 그리려면 모델을 받아야 합니다 ({0})', [AiStore.sizeText(aiMissingSize(store))]),
                        action: TextButton(onPressed: () => openAiModels(context, c), child: Text(tr('받기 화면으로')))),
                  const SizedBox(height: 8),
                  TextField(
                    controller: _prompt,
                    focusNode: _promptFocus,
                    minLines: 2,
                    maxLines: 5,
                    decoration: InputDecoration(labelText: tr('그릴 것 (프롬프트, 영어가 잘 됩니다)'), border: const OutlineInputBorder()),
                  ),
                  const SizedBox(height: 8),
                  TextField(
                    controller: _negative,
                    decoration: InputDecoration(labelText: tr('빼고 싶은 것 (선택)'), border: const OutlineInputBorder()),
                  ),
                  const SizedBox(height: 8),
                  Wrap(spacing: 12, runSpacing: 8, crossAxisAlignment: WrapCrossAlignment.center, children: [
                    _drop<String>(tr('크기'), '${s.aiImage.width}x${s.aiImage.height}', const ['512x512', '512x768', '768x512', '768x768', '1024x1024'],
                        (v) => c.updateSettings((x) {
                              final wh = v.split('x');
                              x.aiImage.width = int.parse(wh[0]);
                              x.aiImage.height = int.parse(wh[1]);
                            })),
                    _drop<int>(tr('단계'), s.aiImage.steps, const [2, 4, 6, 8, 12, 20, 30], (v) => c.updateSettings((x) => x.aiImage.steps = v)),
                    _drop<int>(tr('개수'), s.aiImage.count, const [1, 2, 4, 8, 16], (v) => c.updateSettings((x) => x.aiImage.count = v)),
                    // 141: 이름이 잘리지 않게 짧게 · 넉넉히 (설명은 아래 줄)
                    SizedBox(
                      width: 190,
                      child: JjField(controller: _seed, keyboardType: TextInputType.number, label: tr('시드'), helper: tr('-1 은 아무렇게나'), isDense: true),
                    ),
                  ]),
                  const SizedBox(height: 8),
                  _initRow(s.aiImage.strength),
                  const SizedBox(height: 12),
                  if (jobs.busy) ...[
                    LinearProgressIndicator(value: jobs.progress),
                    const SizedBox(height: 4),
                    Row(children: [
                      Expanded(child: Text(jobs.statusLine ?? '', style: const TextStyle(fontSize: 12))),
                      OutlinedButton(onPressed: jobs.cancel, child: Text(tr('취소'))),
                    ]),
                  ] else
                    Row(children: [
                      FilledButton.icon(
                        onPressed: _generate,
                        icon: const Icon(Icons.auto_awesome),
                        label: Text(_init == null ? tr('만들기') : tr('그림을 바꿔 만들기')),
                      ),
                      const SizedBox(width: 12),
                      if (_step.isNotEmpty) Expanded(child: Text(_step, style: const TextStyle(fontSize: 12))),
                    ]),
                  if (_error != null)
                    Padding(
                      padding: const EdgeInsets.only(top: 8),
                      child: Column(crossAxisAlignment: CrossAxisAlignment.start, children: [
                        Text(_error!, style: TextStyle(color: _cancelled ? JjColors.textDim : JjColors.alert)),
                        if (_detail.isNotEmpty)
                          TextButton(
                            onPressed: () => setState(() => _showDetail = !_showDetail),
                            child: Text(_showDetail ? tr('자세히 닫기') : tr('자세히')),
                          ),
                        if (_showDetail) SelectableText(_detail, style: TextStyle(fontSize: 11, color: JjColors.textDim)),
                      ]),
                    ),
                  const SizedBox(height: 12),
                  if (_results.isNotEmpty) Text(trf('만든 그림 · 저장: {0}', [aiSaveDir(s.aiImage.saveDir)]), style: const TextStyle(fontSize: 12)),
                  const SizedBox(height: 6),
                  Wrap(spacing: 8, runSpacing: 8, children: [for (final g in _results) _thumb(g)]),
                ]),
              ),
            ]),
          ),
        );
      },
    );
  }

  Widget _engineRow() {
    final s = c.settings;
    return Row(children: [
      Text(tr('엔진'), style: const TextStyle(fontSize: 13)),
      const SizedBox(width: 8),
      Flexible(
        child: DropdownButton<String>(
          isExpanded: true,
          value: _service == null ? 'local' : s.aiImage.engine,
          items: [
            DropdownMenuItem(value: 'local', child: Text(tr('이 기기 (SD1.5 + LCM)'))),
            for (final x in s.aiImage.services) DropdownMenuItem(value: x.id, child: Text('☁ ${x.label}')),
          ],
          onChanged: (v) => c.updateSettings((x) => x.aiImage.engine = v ?? 'local'),
        ),
      ),
      IconButton(
        tooltip: tr('AI 설정 · 모델 받기 · 서비스 추가'),
        icon: const Icon(Icons.tune),
        onPressed: () => openAiModels(context, c),
      ),
    ]);
  }

  Widget _initRow(double strength) => Wrap(spacing: 8, runSpacing: 4, crossAxisAlignment: WrapCrossAlignment.center, children: [
        // 바탕 그림이 무엇인지 바로 보이게 작은 그림으로
        if (_init != null)
          ClipRRect(
            borderRadius: BorderRadius.circular(6),
            child: Image.file(File(_init!), width: 64, height: 64, fit: BoxFit.cover,
                errorBuilder: (_, _, _) => const SizedBox(width: 64, height: 64, child: Icon(Icons.broken_image_outlined))),
          ),
        OutlinedButton.icon(
          onPressed: _pickInit,
          icon: const Icon(Icons.image_outlined, size: 18),
          label: Text(_init == null ? tr('바탕 그림 고르기 (그림 → 그림)') : p.basename(_init!)),
        ),
        if (_init != null) ...[
          IconButton(tooltip: tr('바탕 그림 빼기'), icon: const Icon(Icons.close), onPressed: () => setState(() => _init = null)),
          Text(trf('바꾸는 정도 {0}%', [(strength * 100).round()]), style: const TextStyle(fontSize: 12)),
          SizedBox(
            width: 200,
            child: Slider(
              value: strength,
              min: 0.05,
              max: 1,
              onChanged: (v) => c.updateSettings((x) => x.aiImage.strength = v),
            ),
          ),
        ],
      ]);

  Widget _thumb(GeneratedImage g) {
    // 153: 갤러리 등에서 지웠으면 빈칸 대신 알리고 목록에서 뺄 수 있게
    if (!File(g.path).existsSync()) {
      return SizedBox(
        width: 160,
        child: Column(mainAxisSize: MainAxisSize.min, children: [
          Container(
            width: 160,
            height: 160,
            color: JjColors.panel,
            alignment: Alignment.center,
            padding: const EdgeInsets.all(8),
            child: Text(tr('파일이 없습니다 (지워졌거나 옮겨짐)'),
                textAlign: TextAlign.center, style: TextStyle(fontSize: 12, color: JjColors.textDim)),
          ),
          TextButton(
            onPressed: () {
              setState(() => _results.remove(g));
              _saveRecent();
            },
            child: Text(tr('목록에서 빼기'))),
        ]),
      );
    }
    final shown = [for (final r in _results) if (File(r.path).existsSync()) r];
    return Column(mainAxisSize: MainAxisSize.min, children: [
        InkWell(
          onTap: () => openReader(context, c, ImageFilesSource([for (final r in shown) r.path], tempDir: Directory.systemTemp.path, title: tr('AI 그림')),
              start: shown.indexOf(g)),
          child: Image.file(File(g.path), width: 160, height: 160, fit: BoxFit.cover),
        ),
        Row(mainAxisSize: MainAxisSize.min, children: [
          Text(trf('시드 {0}', [g.seed]), style: TextStyle(fontSize: 11, color: JjColors.textDim)),
          IconButton(
            tooltip: tr('같은 조건으로 다시 (시드 그대로)'),
            visualDensity: VisualDensity.compact,
            icon: const Icon(Icons.replay, size: 16),
            onPressed: () => _reuse(g),
          ),
          IconButton(
            tooltip: tr('이 그림을 바탕으로 (그림 → 그림)'),
            visualDensity: VisualDensity.compact,
            icon: const Icon(Icons.edit_outlined, size: 16),
            onPressed: () => setState(() => _init = g.path),
          ),
        ]),
      ]);
  }

  Widget _drop<T>(String label, T value, List<T> items, void Function(T) on) => Row(mainAxisSize: MainAxisSize.min, children: [
        Text(label, style: const TextStyle(fontSize: 13)),
        const SizedBox(width: 6),
        DropdownButton<T>(
          value: items.contains(value) ? value : null,
          hint: Text('$value'),
          items: [for (final x in items) DropdownMenuItem(value: x, child: Text('$x'))],
          onChanged: (v) => v == null ? null : on(v),
        ),
      ]);

  Widget _notice(IconData icon, Color color, String text, {Widget? action}) =>
      NoticeBand(icon: icon, color: color, title: text, margin: const EdgeInsets.only(top: 8), actions: [?action]);
}