import 'package:flutter/material.dart';
import 'package:she_healthy_desktop/presentation/components/table/table_header_custom_cnn.dart';
import 'package:she_healthy_desktop/presentation/components/table/table_row_custom_cnn.dart';

import '../../../utils/tooltip_message.dart';
import '../table/table_header_custom.dart';
import '../table/table_row_custom.dart';

Widget buildCnnLayerBreast(){
  return ListView.separated(
      shrinkWrap: true,
      itemBuilder: (ctx, index) {
        if (index == 0) {
          return const TableHeaderCustomCnn(
              title1: 'Layer (Type)',
              title2: 'Output Shape',
              title3: 'Param #');
        }

        if (index == 1){
          return const TableRowCustomCnn(
            val1: "resnet50v2 (Functional)",
            val2: "(None, 8, 8, 2048)",
            val3: "23564800",
          );
        }

        if (index == 2){
          return const TableRowCustomCnn(
            val1: 'flatten_16 (Flatten)',
            val2: "(None, 131072)",
            val3: "0",
          );
        }

        if (index == 3){
          return const TableRowCustomCnn(
            val1: "dense_49 (Dense)",
            val2: "(None, 316)",
            val3: "41419068"
          );
        }

        if (index == 4){
          return const TableRowCustomCnn(
            val1: "dropout_35 (Dropout)",
            val2: "(None, 316)",
            val3: "0",
          );
        }

        if (index == 5){
          return const TableRowCustomCnn(
              val1: "dense_50 (Dense)",
              val2: "(None, 4)",
              val3: "1268"
          );
        }

        return const SizedBox();
      },
      separatorBuilder: (ctx, index) => const Divider(color: Colors.black87),
      itemCount: 6);
}

Widget buildCnnLayerCervix(){
  return ListView.separated(
      shrinkWrap: true,
      itemBuilder: (ctx, index) {
        if (index == 0) {
          return const TableHeaderCustomCnn(
              title1: 'Layer (Type)',
              title2: 'Output Shape',
              title3: 'Param #');
        }

        if (index == 1){
          return const TableRowCustomCnn(
            val1: "resnet50v2 (Functional)",
            val2: "(None, 8, 8, 2048)",
            val3: "23564800",
          );
        }

        if (index == 2){
          return const TableRowCustomCnn(
            val1: 'flatten_10 (Flatten)',
            val2: "(None, 131072)",
            val3: "0",
          );
        }

        if (index == 3){
          return const TableRowCustomCnn(
              val1: "dense_28 (Dense)",
              val2: "(None, 141)",
              val3: "18481293"
          );
        }

        if (index == 4){
          return const TableRowCustomCnn(
            val1: "dropout_18 (Dropout)",
            val2: "(None, 141)",
            val3: "0",
          );
        }

        if (index == 5){
          return const TableRowCustomCnn(
              val1: "dense_29 (Dense)",
              val2: "(None, 269)",
              val3: "38198"
          );
        }

        if (index == 6){
          return const TableRowCustomCnn(
              val1: "dropout_19 (Dense)",
              val2: "(None, 269)",
              val3: "0"
          );
        }

        if (index == 7){
          return const TableRowCustomCnn(
              val1: "dense_30 (Dense)",
              val2: "(None, 415)",
              val3: "112050"
          );
        }

        if (index == 8){
          return const TableRowCustomCnn(
              val1: "dropout_20 (Dropout)",
              val2: "(None, 415)",
              val3: "0"
          );
        }

        if (index == 9){
          return const TableRowCustomCnn(
              val1: "dense_31 (Dense)",
              val2: "(None, 4)",
              val3: "1664"
          );
        }

        return const SizedBox();
      },
      separatorBuilder: (ctx, index) => const Divider(color: Colors.black87),
      itemCount: 10);
}