page.dart

import 'package:flutter/material.dart';

import 'package:provider/provider.dart';

import 'package:omi/backend/schema/conversation.dart';
import 'package:omi/pages/capture/widgets/widgets.dart';
import 'package:omi/providers/conversation_provider.dart';
import 'package:omi/utils/l10n_extensions.dart';

class ProcessingConversationPage extends StatefulWidget {
  final ServerConversation conversation;

  const ProcessingConversationPage({super.key, required this.conversation});

  @override
  State<ProcessingConversationPage> createState() => _ProcessingConversationPageState();
}

class _ProcessingConversationPageState extends State<ProcessingConversationPage> with TickerProviderStateMixin {
  final scaffoldKey = GlobalKey<ScaffoldState>();
  TabController? _controller;

  @override
  void initState() {
    _controller = TabController(length: 2, vsync: this, initialIndex: 0);
    _controller!.addListener(() => setState(() {}));
    super.initState();
  }

  @override
  Widget build(BuildContext context) {
    return Consumer<ConversationProvider>(
      builder: (context, provider, child) {
        // Conversation source
        var convoSource = widget.conversation.source;
        bool hasPhotos = widget.conversation.photos.isNotEmpty;
        String contentTabLabel;
        if (convoSource == ConversationSource.openglass) {
          contentTabLabel = context.l10n.photos;
        } else if (convoSource == ConversationSource.screenpipe) {
          contentTabLabel = context.l10n.rawData;
        } else {
          contentTabLabel = context.l10n.content;
        }

        return PopScope(
          canPop: true,
          child: Scaffold(
            key: scaffoldKey,
            backgroundColor: Theme.of(context).colorScheme.primary,
            appBar: AppBar(
              automaticallyImplyLeading: false,
              backgroundColor: Theme.of(context).colorScheme.primary,
              title: Row(
                mainAxisSize: MainAxisSize.max,
                mainAxisAlignment: MainAxisAlignment.spaceBetween,
                crossAxisAlignment: CrossAxisAlignment.center,
                children: [
                  IconButton(
                    key: const ValueKey('processing_conversation_back_button'),
                    onPressed: () {
                      Navigator.pop(context);
                      return;
                    },
                    icon: const Icon(Icons.arrow_back_rounded, size: 24.0),
                  ),
                  const SizedBox(width: 4),
                  Text(hasPhotos ? "📸" : "🎙️"),
                  const SizedBox(width: 4),
                  Expanded(child: Text(context.l10n.inProgress)),
                ],
              ),
            ),
            body: Column(
              children: [
                TabBar(
                  key: const ValueKey('processing_conversation_tab_bar'),
                  indicatorSize: TabBarIndicatorSize.label,
                  isScrollable: false,
                  padding: EdgeInsets.zero,
                  indicatorPadding: EdgeInsets.zero,
                  controller: _controller,
                  labelStyle: Theme.of(context).textTheme.titleLarge!.copyWith(fontSize: 18),
                  tabs: [
                    Tab(text: contentTabLabel),
                    Tab(text: context.l10n.summary),
                  ],
                  indicator: BoxDecoration(color: Colors.transparent, borderRadius: BorderRadius.circular(16)),
                ),
                Expanded(
                  child: Padding(
                    padding: const EdgeInsets.symmetric(horizontal: 16),
                    child: TabBarView(
                      controller: _controller,
                      physics: const NeverScrollableScrollPhysics(),
                      children: [
                        ListView(
                          shrinkWrap: true,
                          children: [
                            if (widget.conversation.transcriptSegments.isNotEmpty ||
                                widget.conversation.photos.isNotEmpty)
                              getTranscriptWidget(
                                false,
                                widget.conversation.transcriptSegments,
                                widget.conversation.photos,
                                null,
                              ),
                            if (!hasPhotos && widget.conversation.transcriptSegments.isEmpty)
                              Column(
                                children: [
                                  const SizedBox(height: 80),
                                  Center(child: Text(context.l10n.noContentToDisplay)),
                                ],
                              ),
                            const SizedBox(height: 32),
                          ],
                        ),
                        ListView(
                          shrinkWrap: true,
                          children: [
                            const SizedBox(height: 80),
                            Center(
                              child: Padding(
                                padding: const EdgeInsets.symmetric(horizontal: 16.0),
                                child: Text(
                                  widget.conversation.transcriptSegments.isEmpty
                                      ? context.l10n.noSummary
                                      : context.l10n.statusProcessing,
                                  textAlign: TextAlign.center,
                                  style: const TextStyle(fontSize: 16),
                                ),
                              ),
                            ),
                          ],
                        ),
                      ],
                    ),
                  ),
                ),
              ],
            ),
          ),
        );
      },
    );
  }
}