Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

User page and provider #385

Merged
merged 3 commits into from
Mar 1, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
7 changes: 5 additions & 2 deletions lib/main.dart
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import 'dart:io';

import 'package:flutter/material.dart';
import 'package:maid/providers/user.dart';
import 'package:maid/ui/mobile/pages/home_page.dart';
import 'package:maid/providers/ai_platform.dart';
import 'package:maid/providers/session.dart';
Expand All @@ -23,6 +24,7 @@ void main() {
providers: [
ChangeNotifierProvider(create: (context) => MainProvider()),
ChangeNotifierProvider(create: (context) => AiPlatform()),
ChangeNotifierProvider(create: (context) => User()),
ChangeNotifierProvider(create: (context) => Character()),
ChangeNotifierProvider(create: (context) => Session()),
],
Expand Down Expand Up @@ -67,11 +69,12 @@ class MaidApp extends StatefulWidget {
class MaidAppState extends State<MaidApp> {
@override
Widget build(BuildContext context) {
return Consumer4<MainProvider, AiPlatform, Character, Session>(
builder: (context, mainProvider, ai, character, session, child) {
return Consumer5<MainProvider, AiPlatform, User, Character, Session>(
builder: (context, mainProvider, ai, user, character, session, child) {
if (!mainProvider.initialised) {
mainProvider.init();
ai.init();
user.init();
character.init();
session.init();
}
Expand Down
10 changes: 5 additions & 5 deletions lib/providers/ai_platform.dart
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ class AiPlatform extends ChangeNotifier {
void newPreset() {
final key = UniqueKey().toString();
_preset = "New Preset $key";
resetAll();
reset();
}

void notify() {
Expand All @@ -64,7 +64,7 @@ class AiPlatform extends ChangeNotifier {
fromMap(lastModel);
Logger.log(lastModel.toString());
} else {
resetAll();
reset();
}
}

Expand Down Expand Up @@ -255,7 +255,7 @@ class AiPlatform extends ChangeNotifier {

void fromMap(Map<String, dynamic> inputJson) {
if (inputJson.isEmpty) {
resetAll();
reset();
} else {
_promptFormat = PromptFormatType.values[
inputJson["prompt_promptFormat"] ?? PromptFormatType.alpaca.index];
Expand Down Expand Up @@ -328,7 +328,7 @@ class AiPlatform extends ChangeNotifier {
return outputJson;
}

void resetAll() {
void reset() {
rootBundle.loadString('assets/default_parameters.json').then((jsonString) {
Map<String, dynamic> assetJson = json.decode(jsonString);

Expand Down Expand Up @@ -370,7 +370,7 @@ class AiPlatform extends ChangeNotifier {

fromMap(inputJson);
} catch (e) {
resetAll();
reset();
return "Error: $e";
}

Expand Down
31 changes: 7 additions & 24 deletions lib/providers/character.dart
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@ import 'package:flutter/material.dart';
import 'package:maid/static/file_manager.dart';
import 'package:maid/static/logger.dart';
import 'package:image/image.dart';
import 'package:maid/static/user.dart';
import 'package:path_provider/path_provider.dart';
import 'package:shared_preferences/shared_preferences.dart';

Expand All @@ -28,7 +27,7 @@ class Character extends ChangeNotifier {
void newCharacter() {
final key = UniqueKey().toString();
_name = "New Character $key";
resetAll();
reset();
}

void notify() {
Expand All @@ -47,7 +46,7 @@ class Character extends ChangeNotifier {
Logger.log(lastCharacter.toString());
fromMap(lastCharacter);
} else {
resetAll();
reset();
}
}

Expand All @@ -72,7 +71,7 @@ class Character extends ChangeNotifier {
_name = inputJson["name"] ?? "Unknown";

if (inputJson.isEmpty) {
resetAll();
reset();
}

_description = inputJson["description"] ?? "";
Expand Down Expand Up @@ -234,7 +233,7 @@ class Character extends ChangeNotifier {

List<Map<String, dynamic>> get examples => _examples;

void resetAll() {
void reset() {
// Reset all the internal state to the defaults
rootBundle.loadString('assets/default_character.json').then((jsonString) {
Map<String, dynamic> jsonCharacter = json.decode(jsonString);
Expand Down Expand Up @@ -276,14 +275,14 @@ class Character extends ChangeNotifier {
Map<String, dynamic> jsonCharacter = json.decode(jsonString);

if (jsonCharacter.isEmpty) {
resetAll();
reset();
return "Failed to decode character";
}

fromMap(jsonCharacter);
return "Character Successfully Loaded";
} catch (e) {
resetAll();
reset();
Logger.log("Error: $e");
return "Error: $e";
}
Expand Down Expand Up @@ -377,7 +376,7 @@ class Character extends ChangeNotifier {
notifyListeners();
return "Character Successfully Loaded";
} catch (e) {
resetAll();
reset();
Logger.log("Error: $e");
return "Error: $e";
}
Expand Down Expand Up @@ -429,20 +428,4 @@ class Character extends ChangeNotifier {

return buffer.toString();
}

String formatPlaceholders(String input) {
input = _replaceCaseInsensitive(input, "{{char}}", _name);
input = _replaceCaseInsensitive(input, "<BOT>", _name);
input = _replaceCaseInsensitive(input, "{{user}}", User.name);
input = _replaceCaseInsensitive(input, "<USER>", User.name);

return input;
}

String _replaceCaseInsensitive(
String original, String from, String replaceWith) {
// This creates a regular expression that ignores case (case-insensitive)
RegExp exp = RegExp(RegExp.escape(from), caseSensitive: false);
return original.replaceAll(exp, replaceWith);
}
}
64 changes: 64 additions & 0 deletions lib/providers/user.dart
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
import 'dart:io';
import 'dart:async';
import 'dart:convert';

import 'package:flutter/material.dart';
import 'package:shared_preferences/shared_preferences.dart';

class User extends ChangeNotifier {
File _profile = File("assets/chadUser.png");
String _name = "User";

File get profile => _profile;

String get name => _name;

set profile(File value) {
_profile = value;
notifyListeners();
}

set name(String value) {
_name = value;
notifyListeners();
}

void init() async {
final prefs = await SharedPreferences.getInstance();

Map<String, dynamic> lastUser =
json.decode(prefs.getString("last_user") ?? "{}") ?? {};

if (lastUser.isNotEmpty) {
fromMap(lastUser);
} else {
reset();
}

notifyListeners();
}

void fromMap(Map<String, dynamic> inputJson) {
if (inputJson["profile"] != null) {
_profile = File(inputJson["profile"]);
} else {
_profile = File("assets/chadUser.png");
}

_name = inputJson["name"];
notifyListeners();
}

Map<String, dynamic> toMap() {
return {
"profile": _profile.path,
"name": _name,
};
}

void reset() {
_profile = File("assets/chadUser.png");
_name = "User";
notifyListeners();
}
}
40 changes: 30 additions & 10 deletions lib/static/generation_manager.dart
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ import 'package:langchain_openai/langchain_openai.dart';
import 'package:llama_cpp_dart/llama_cpp_dart.dart';
import 'package:maid/providers/character.dart';
import 'package:maid/providers/session.dart';
import 'package:maid/providers/user.dart';
import 'package:maid/static/logger.dart';
import 'package:maid/providers/ai_platform.dart';
import 'package:lan_scanner/lan_scanner.dart';
Expand All @@ -39,6 +40,7 @@ class GenerationManager {
_completer = Completer();

final ai = context.read<AiPlatform>();
final user = context.read<User>();
final character = context.read<Character>();
final session = context.read<Session>();

Expand Down Expand Up @@ -76,10 +78,10 @@ class GenerationManager {
{
'role': 'system',
'content': '''
${character.formatPlaceholders(character.description)}\n\n
${character.formatPlaceholders(character.personality)}\n\n
${character.formatPlaceholders(character.scenario)}\n\n
${character.formatPlaceholders(character.system)}\n\n
${formatPlaceholders(character.description, user.name, character.name)}\n\n
${formatPlaceholders(character.personality, user.name, character.name)}\n\n
${formatPlaceholders(character.scenario, user.name, character.name)}\n\n
${formatPlaceholders(character.system, user.name, character.name)}\n\n
'''
}
];
Expand Down Expand Up @@ -107,6 +109,7 @@ class GenerationManager {
static void remotePrompt(String input, BuildContext context,
void Function(String?) callback) async {
final ai = context.read<AiPlatform>();
final user = context.read<User>();
final character = context.read<Character>();
final session = context.read<Session>();

Expand All @@ -118,10 +121,10 @@ class GenerationManager {
List<ChatMessage> chatMessages = [];

final prePrompt = '''
${character.formatPlaceholders(character.description)}\n\n
${character.formatPlaceholders(character.personality)}\n\n
${character.formatPlaceholders(character.scenario)}\n\n
${character.formatPlaceholders(character.system)}\n\n
${formatPlaceholders(character.description, user.name, character.name)}\n\n
${formatPlaceholders(character.personality, user.name, character.name)}\n\n
${formatPlaceholders(character.scenario, user.name, character.name)}\n\n
${formatPlaceholders(character.system, user.name, character.name)}\n\n
''';

List<Map<String, dynamic>> messages = [
Expand Down Expand Up @@ -152,8 +155,8 @@ class GenerationManager {
break;
}

chatMessages.add(
ChatMessage.system(character.formatPlaceholders(character.system)));
chatMessages.add(ChatMessage.system(
formatPlaceholders(character.system, user.name, character.name)));
}

chatMessages.add(ChatMessage.humanText(input));
Expand Down Expand Up @@ -399,4 +402,21 @@ class GenerationManager {
return false;
}
}

static String formatPlaceholders(
String input, String userName, String characterName) {
input = _replaceCaseInsensitive(input, "{{char}}", characterName);
input = _replaceCaseInsensitive(input, "<BOT>", characterName);
input = _replaceCaseInsensitive(input, "{{user}}", userName);
input = _replaceCaseInsensitive(input, "<USER>", userName);

return input;
}

static String _replaceCaseInsensitive(
String original, String from, String replaceWith) {
// This creates a regular expression that ignores case (case-insensitive)
RegExp exp = RegExp(RegExp.escape(from), caseSensitive: false);
return original.replaceAll(exp, replaceWith);
}
}
3 changes: 0 additions & 3 deletions lib/static/user.dart

This file was deleted.

5 changes: 2 additions & 3 deletions lib/ui/mobile/pages/character_page.dart
Original file line number Diff line number Diff line change
Expand Up @@ -102,8 +102,7 @@ class _CharacterPageState extends State<CharacterPage> {
children: [
const SizedBox(height: 10.0),
CircleAvatar(
backgroundImage:
const AssetImage("assets/maid.png"),
backgroundImage: const AssetImage("assets/maid.png"),
foregroundImage: Image.file(character.profile).image,
radius: 75,
),
Expand Down Expand Up @@ -236,7 +235,7 @@ class _CharacterPageState extends State<CharacterPage> {
},
rightText: "Reset All",
rightOnPressed: () {
character.resetAll();
character.reset();
}),
const SizedBox(height: 15.0),
DoubleButtonRow(
Expand Down