Merge pull request #31 from openvinotoolkit/arendjan/tti_draft

TTI feature

lib/inference/inference_page.dart

@@ -2,6 +2,7 @@ import 'package:flutter/material.dart';
 import 'package:inference/inference/download_page.dart';
 import 'package:inference/inference/image_inference_page.dart';
 import 'package:inference/inference/text_inference_page.dart';
+import 'package:inference/inference/text_to_image_inference_page.dart';
 import 'package:inference/project.dart';
 import 'package:inference/providers/download_provider.dart';
 import 'package:provider/provider.dart';
@@ -23,6 +24,8 @@ class _InferencePageState extends State<InferencePage> {
           return ImageInferencePage(widget.project);
         case ProjectType.text:
           return TextInferencePage(widget.project);
+        case ProjectType.textToImage:
+          return TextToImageInferencePage(widget.project);
         case ProjectType.speech:
           return Container();
       }

lib/inference/textToImage/tti_metric_widgets.dart

@@ -0,0 +1,33 @@
+import 'package:flutter/material.dart';
+import 'package:inference/inference/text/metric_widgets.dart';
+import 'package:inference/interop/openvino_bindings.dart';
+import 'package:intl/intl.dart';
+
+class TTICirclePropRow extends StatelessWidget {
+  final TTIMetrics metrics;
+
+  const TTICirclePropRow({super.key, required this.metrics});
+
+  @override
+  Widget build(BuildContext context) {
+    Locale locale = Localizations.localeOf(context);
+    final nf = NumberFormat.decimalPatternDigits(
+        locale: locale.languageCode, decimalDigits: 0);
+
+    return Row(
+      mainAxisAlignment: MainAxisAlignment.spaceBetween,
+      children: [
+        CircleProp(
+          header: "Time to load model",
+          value: nf.format(metrics.load_time),
+          unit: "ms",
+        ),
+        CircleProp(
+          header: "Time to generate image",
+          value: nf.format(metrics.generate_time),
+          unit: "ms",
+        )
+      ],
+    );
+  }
+}

lib/inference/textToImage/tti_performance_metrics.dart

@@ -0,0 +1,71 @@
+import 'package:flutter/material.dart';
+import 'package:inference/inference/text/metric_widgets.dart';
+import 'package:inference/inference/textToImage/tti_metric_widgets.dart';
+import 'package:inference/providers/text_to_image_inference_provider.dart';
+import 'package:inference/theme.dart';
+import 'package:intl/intl.dart';
+import 'package:provider/provider.dart';
+
+class TTIPerformanceMetricsPage extends StatefulWidget {
+  const TTIPerformanceMetricsPage({super.key});
+
+  @override
+  State<TTIPerformanceMetricsPage> createState() => _TTIPerformanceMetricsPageState();
+}
+
+class _TTIPerformanceMetricsPageState extends State<TTIPerformanceMetricsPage> {
+
+  @override
+  void initState() {
+    super.initState();
+    final provider = Provider.of<TextToImageInferenceProvider>(context, listen: false);
+    if (provider.metrics == null) {
+      provider.loaded.future.then((_) {
+          provider.message("Generate OpenVINO logo");
+      });
+    }
+  }
+
+  @override
+  Widget build(BuildContext context) {
+    return Consumer<TextToImageInferenceProvider>(builder: (context, inference, child) {
+        if (inference.metrics == null) {
+          return Center(
+            child: Column(
+              mainAxisAlignment: MainAxisAlignment.center,
+              children: [
+                Image.asset('images/intel-loading.gif', width: 100),
+                const Text("Running benchmark prompt...")
+              ],
+            )
+          );
+        }
+
+        Locale locale = Localizations.localeOf(context);
+        final nf = NumberFormat.decimalPatternDigits(
+            locale: locale.languageCode, decimalDigits: 0);
+
+        final metrics = inference.metrics!;
+
+        return Container(
+          decoration: BoxDecoration(
+            shape: BoxShape.rectangle,
+            borderRadius: const BorderRadius.all(Radius.circular(8)),
+            color: intelGray,
+          ),
+          child: Padding(
+            padding: const EdgeInsets.all(30.0),
+            child: Column(
+              mainAxisAlignment: MainAxisAlignment.start,
+              crossAxisAlignment: CrossAxisAlignment.stretch,
+              children: [
+                TTICirclePropRow(metrics: metrics),
+              ],
+            ),
+          ),
+        );
+    });
+  }
+}
+
+