Fix #508 (#518)

* add requirements dialog

* make storage op dialog futurebuilder

lib/classes/google_gemini_model.dart

@@ -26,6 +26,21 @@ class GoogleGeminiModel extends LargeLanguageModel {
  fromMap(json);
  }
 
+ @override
+ List<String> get missingRequirements {
+ List<String> missing = [];
+
+ if (name.isEmpty) {
+ missing.add('- A model option is required for prompting.\n');
+ }
+
+ if (token.isEmpty) {
+ missing.add('- An authentication token is required for prompting.\n');
+ } 
+
+ return missing;
+ }
+
  @override
  Stream<String> prompt(List<ChatNode> messages) async* {
  try {

lib/classes/large_language_model.dart

@@ -325,6 +325,24 @@ class LargeLanguageModel extends ChangeNotifier {
  };
  }
 
+ List<String> get missingRequirements {
+ List<String> missing = [];
+
+ if (_name.isEmpty) {
+ missing.add('- A model option is required for prompting.\n');
+ } 
+
+ if (_uri.isEmpty) {
+ missing.add('- A compatible URL is required for prompting.\n');
+ }
+
+ if (_token.isEmpty) {
+ missing.add('- An authentication token is required for prompting.\n');
+ } 
+
+ return missing;
+ }
+
  Stream<String> prompt(List<ChatNode> messages) {
  throw UnimplementedError();
  }

lib/classes/llama_cpp_model.dart

@@ -24,6 +24,21 @@ class LlamaCppModel extends LargeLanguageModel {
  _promptFormat = value;
  notifyListeners();
  }
+
+ @override
+ List<String> get missingRequirements {
+ List<String> missing = [];
+
+ if (uri.isEmpty) {
+ missing.add('- A path to the model file is required.\n');
+ } 
+
+ if (!File(uri).existsSync()) {
+ missing.add('- The file provided does not exist.\n');
+ }
+
+ return missing;
+ }
 
  LlamaCppModel({
  super.listener, 
@@ -145,7 +160,7 @@ class LlamaCppModel extends LargeLanguageModel {
  uri = file.path;
  notifyListeners();
  } catch (e) {
- return "Error: $e";
+ return e.toString();
  }
 
  return "Model Successfully Loaded";

lib/classes/ollama_model.dart

@@ -20,6 +20,21 @@ class OllamaModel extends LargeLanguageModel {
 
  String _ip = '';
 
+ @override
+ List<String> get missingRequirements {
+ List<String> missing = [];
+
+ if (name.isEmpty) {
+ missing.add('- A model option is required for prompting.\n');
+ } 
+
+ if (uri.isEmpty) {
+ missing.add('- A compatible URL is required for prompting.\n');
+ }
+
+ return missing;
+ }
+
  OllamaModel({
  super.listener, 
  super.name,

lib/ui/mobile/pages/platforms/llama_cpp_page.dart

@@ -87,9 +87,11 @@ class _LlamaCppPageState extends State<LlamaCppPage> {
  mainAxisAlignment: MainAxisAlignment.center,
  children: [
  FilledButton(
- onPressed: () async {
- await storageOperationDialog(
- context, (session.model as LlamaCppModel).loadModel);
+ onPressed: () {
+ storageOperationDialog(
+ context, 
+ (session.model as LlamaCppModel).loadModel
+ );
  session.notify();
  },
  child: Text(

lib/ui/mobile/widgets/chat_widgets/chat_field.dart

@@ -2,6 +2,7 @@ import 'dart:io';
 import 'dart:async';
 
 import 'package:flutter/material.dart';
+import 'package:maid/ui/mobile/widgets/dialogs.dart';
 import 'package:maid_llm/src/chat_node.dart';
 import 'package:maid/classes/large_language_model.dart';
 import 'package:maid/providers/session.dart';
@@ -109,7 +110,10 @@ class _ChatFieldState extends State<ChatField> {
  ),
  IconButton(
  onPressed: () {
- if (session.chat.tail.finalised ) {
+ if (session.model.missingRequirements.isNotEmpty) {
+ showMissingRequirementsDialog(context);
+ }
+ else if (session.chat.tail.finalised) {
  send();
  }
  },

lib/ui/mobile/widgets/chat_widgets/chat_message.dart

@@ -1,4 +1,5 @@
 import 'package:flutter/material.dart';
+import 'package:maid/ui/mobile/widgets/dialogs.dart';
 import 'package:maid_llm/maid_llm.dart';
 import 'package:maid/providers/character.dart';
 import 'package:maid/providers/session.dart';
@@ -145,9 +146,15 @@ class _ChatMessageState extends State<ChatMessage> with SingleTickerProviderStat
  IconButton(
  onPressed: () {
  if (!context.read<Session>().chat.tail.finalised) return;
- setState(() {
- editing = true;
- });
+
+ if (context.read<Session>().model.missingRequirements.isNotEmpty) {
+ showMissingRequirementsDialog(context);
+ }
+ else {
+ setState(() {
+ editing = true;
+ });
+ }
  },
  icon: const Icon(Icons.edit),
  ),
@@ -159,8 +166,13 @@ class _ChatMessageState extends State<ChatMessage> with SingleTickerProviderStat
  IconButton(
  onPressed: () {
  if (!context.read<Session>().chat.tail.finalised) return;
- context.read<Session>().regenerate(node.key, context);
- setState(() {});
+
+ if (context.read<Session>().model.missingRequirements.isNotEmpty) {
+ showMissingRequirementsDialog(context);
+ } else {
+ context.read<Session>().regenerate(node.key, context);
+ setState(() {});
+ }
  },
  icon: const Icon(Icons.refresh),
  ),
@@ -189,10 +201,16 @@ class _ChatMessageState extends State<ChatMessage> with SingleTickerProviderStat
  padding: const EdgeInsets.all(0),
  onPressed: () {
  if (!context.read<Session>().chat.tail.finalised) return;
- setState(() {
- editing = false;
- });
- context.read<Session>().edit(node.key, messageController.text, context);
+
+ if (context.read<Session>().model.missingRequirements.isNotEmpty) {
+ showMissingRequirementsDialog(context);
+ } 
+ else {
+ setState(() {
+ editing = false;
+ });
+ context.read<Session>().edit(node.key, messageController.text, context);
+ }
  },
  icon: const Icon(Icons.done)),
  IconButton(

lib/ui/mobile/widgets/dialogs.dart

@@ -1,34 +1,77 @@
 
 import 'package:flutter/material.dart';
+import 'package:maid/providers/session.dart';
+import 'package:provider/provider.dart';
 
-Future<void> storageOperationDialog(BuildContext context, Future<String> Function(BuildContext context) storageFunction) async {
- String ret = await storageFunction(context);
- // Ensure that the context is still valid before attempting to show the dialog.
- if (context.mounted) {
- showDialog(
- context: context,
- builder: (BuildContext context) {
- return AlertDialog(
- title: Text(ret),
- alignment: Alignment.center,
- actionsAlignment: MainAxisAlignment.center,
- backgroundColor: Theme.of(context).colorScheme.background,
- shape: const RoundedRectangleBorder(
- borderRadius: BorderRadius.all(Radius.circular(20.0)),
- ),
- actions: [
- FilledButton(
- onPressed: () {
- Navigator.of(context).pop();
- },
- child: Text(
- "Close",
- style: Theme.of(context).textTheme.labelLarge,
+void storageOperationDialog(BuildContext context, Future<String> Function(BuildContext context) storageFunction) {
+ showDialog(
+ context: context,
+ builder: (BuildContext context) {
+ return FutureBuilder<String>(
+ future: storageFunction(context),
+ builder: (BuildContext context, AsyncSnapshot<String> snapshot) {
+ if (snapshot.connectionState == ConnectionState.done) {
+ return AlertDialog(
+ title: Text(
+ snapshot.data!,
+ textAlign: TextAlign.center
+ ),
+ actionsAlignment: MainAxisAlignment.center,
+ actions: [
+ FilledButton(
+ onPressed: () {
+ Navigator.of(context).pop();
+ },
+ child: Text(
+ "Close",
+ style: Theme.of(context).textTheme.labelLarge,
+ ),
+ ),
+ ],
+ );
+ } else {
+ return const AlertDialog(
+ title: Text(
+ "Storage Operation Pending",
+ textAlign: TextAlign.center
  ),
+ content: Center(
+ heightFactor: 1.0,
+ child: CircularProgressIndicator(),
+ )
+ );
+ }
+ },
+ );
+ },
+ );
+}
+
+void showMissingRequirementsDialog(BuildContext context) {
+ showDialog(
+ context: context,
+ builder: (context) {
+ final requirement = context.read<Session>().model.missingRequirements;
+
+ return AlertDialog(
+ title: const Text(
+ "Missing Requirements",
+ textAlign: TextAlign.center
+ ),
+ actionsAlignment: MainAxisAlignment.center,
+ content: Text(
+ requirement.join()
+ ),
+ actions: [
+ FilledButton(
+ onPressed: () => Navigator.of(context).pop(),
+ child: Text(
+ "OK",
+ style: Theme.of(context).textTheme.labelLarge
  ),
- ],
- );
- },
- );
- }
+ ),
+ ],
+ );
+ },
+ );
 }