385 lines
12 KiB
Dart
385 lines
12 KiB
Dart
import 'dart:convert';
|
|
|
|
import 'package:Mzansi_Innovation_Hub/main.dart';
|
|
import 'package:Mzansi_Innovation_Hub/mih_components/mih_inputs_and_buttons/mih_dropdown_input.dart';
|
|
import 'package:Mzansi_Innovation_Hub/mih_components/mih_package/mih-app_tool_body.dart';
|
|
import 'package:Mzansi_Innovation_Hub/mih_components/mih_package/mih_app_window.dart';
|
|
import 'package:Mzansi_Innovation_Hub/mih_components/mih_pop_up_messages/mih_loading_circle.dart';
|
|
import 'package:Mzansi_Innovation_Hub/mih_env/env.dart';
|
|
import 'package:Mzansi_Innovation_Hub/mih_objects/app_user.dart';
|
|
import 'package:flutter/material.dart';
|
|
import 'package:flutter_chat_ui/flutter_chat_ui.dart';
|
|
import 'package:flutter_chat_types/flutter_chat_types.dart' as types;
|
|
import 'package:flutter/services.dart' show rootBundle;
|
|
import 'package:ollama_dart/ollama_dart.dart' as ollama;
|
|
import 'package:uuid/uuid.dart';
|
|
|
|
class AiChat extends StatefulWidget {
|
|
final AppUser signedInUser;
|
|
const AiChat({
|
|
super.key,
|
|
required this.signedInUser,
|
|
});
|
|
|
|
@override
|
|
State<AiChat> createState() => _AiChatState();
|
|
}
|
|
|
|
class _AiChatState extends State<AiChat> {
|
|
TextEditingController _modelCopntroller = TextEditingController();
|
|
final ValueNotifier<bool> _showModelOptions = ValueNotifier(false);
|
|
List<types.Message> _messages = [];
|
|
late types.User _user;
|
|
late types.User _mihAI;
|
|
String systemPromt =
|
|
"You are a helpful and friendly AI assistant. You are running on a system called MIH which was created by \"Mzansi Innovation Hub\" a South African based company.";
|
|
final client = ollama.OllamaClient(
|
|
baseUrl: "${AppEnviroment.baseAiUrl}/api",
|
|
);
|
|
List<ollama.Message> _chatHistory = [];
|
|
|
|
void _addMessage(types.Message message) {
|
|
setState(() {
|
|
_messages.insert(0, message);
|
|
});
|
|
}
|
|
|
|
void _handleSendPressed(types.PartialText message) {
|
|
final textMessage = types.TextMessage(
|
|
author: _user,
|
|
createdAt: DateTime.now().millisecondsSinceEpoch,
|
|
id: const Uuid().v4(),
|
|
text: message.text,
|
|
);
|
|
//Add user prompt to history
|
|
setState(() {
|
|
_chatHistory.add(
|
|
ollama.Message(
|
|
role: ollama.MessageRole.user,
|
|
content: message.text,
|
|
),
|
|
);
|
|
});
|
|
|
|
_addMessage(textMessage);
|
|
|
|
_handleMessageBack(message.text);
|
|
}
|
|
|
|
void _handleMessageBack(String userMessage) async {
|
|
// types.TextMessage textMessage;
|
|
// String aiResponse = "";
|
|
// final aiResponseStream =
|
|
showDialog(
|
|
context: context,
|
|
builder: (context) {
|
|
return const Mihloadingcircle();
|
|
},
|
|
);
|
|
_generateChatCompletionWithHistoryStream(userMessage, client)
|
|
.listen((response) {
|
|
//aiResponse = response; //.split("</think>").last.trim();
|
|
});
|
|
Navigator.of(context).pop();
|
|
showDialog(
|
|
context: context,
|
|
builder: (context) {
|
|
return responseWindow();
|
|
},
|
|
);
|
|
// setState(() {
|
|
// _chatHistory.add(
|
|
// ollama.Message(
|
|
// role: ollama.MessageRole.assistant,
|
|
// content: aiResponse,
|
|
// ),
|
|
// );
|
|
// });
|
|
// textMessage = types.TextMessage(
|
|
// author: _mihAI,
|
|
// createdAt: DateTime.now().millisecondsSinceEpoch,
|
|
// id: const Uuid().v4(),
|
|
// text: aiResponse //message.text,
|
|
// );
|
|
|
|
// _addMessage(textMessage);
|
|
// print(_chatHistory.toString());
|
|
// Navigator.of(context).pop();
|
|
}
|
|
|
|
Widget responseWindow() {
|
|
types.TextMessage textMessage;
|
|
return StreamBuilder(
|
|
stream: _generateChatCompletionWithHistoryStream("Hello", client),
|
|
builder: (context, snapshot) {
|
|
if (snapshot.hasData) {
|
|
return MihAppWindow(
|
|
fullscreen: false,
|
|
windowTitle: 'Mzansi AI Thoughts',
|
|
windowTools: [
|
|
IconButton(
|
|
onPressed: () {
|
|
setState(() {
|
|
_chatHistory.add(
|
|
ollama.Message(
|
|
role: ollama.MessageRole.assistant,
|
|
content: snapshot.requireData,
|
|
),
|
|
);
|
|
});
|
|
textMessage = types.TextMessage(
|
|
author: _mihAI,
|
|
createdAt: DateTime.now().millisecondsSinceEpoch,
|
|
id: const Uuid().v4(),
|
|
// metadata: <String, dynamic>{
|
|
// "thoughts": snapshot.requireData
|
|
// },
|
|
text: snapshot.requireData
|
|
.replaceAll("<think>\n\n", "Thinking:\n")
|
|
.replaceAll(
|
|
"</think>\n\n", "Answer:\n"), //message.text,
|
|
);
|
|
|
|
_addMessage(textMessage);
|
|
Navigator.of(context).pop();
|
|
},
|
|
icon: const Icon(Icons.arrow_back),
|
|
),
|
|
],
|
|
onWindowTapClose: () {
|
|
Navigator.of(context).pop();
|
|
},
|
|
windowBody: [
|
|
Text(
|
|
snapshot.requireData,
|
|
textAlign: TextAlign.left,
|
|
style: TextStyle(
|
|
color:
|
|
MzanziInnovationHub.of(context)!.theme.secondaryColor(),
|
|
fontSize: 12,
|
|
fontWeight: FontWeight.bold,
|
|
),
|
|
),
|
|
],
|
|
);
|
|
} else {
|
|
return MihAppWindow(
|
|
fullscreen: false,
|
|
windowTitle: 'Mzansi AI Thoughts',
|
|
windowTools: [],
|
|
onWindowTapClose: () {
|
|
Navigator.of(context).pop();
|
|
},
|
|
windowBody: const [
|
|
Mihloadingcircle(),
|
|
],
|
|
);
|
|
}
|
|
},
|
|
);
|
|
}
|
|
|
|
void _loadMessages() async {
|
|
final response = await rootBundle.loadString('assets/messages.json');
|
|
final messages = (jsonDecode(response) as List)
|
|
.map((e) => types.Message.fromJson(e as Map<String, dynamic>))
|
|
.toList();
|
|
|
|
setState(() {
|
|
_messages = messages;
|
|
});
|
|
}
|
|
|
|
// Future<String> _generateChatCompletionWithHistory(
|
|
// String userMessage,
|
|
// final ollama.OllamaClient client,
|
|
// ) async {
|
|
// final generated = await client.generateChatCompletion(
|
|
// request: ollama.GenerateChatCompletionRequest(
|
|
// model: _modelCopntroller.text,
|
|
// messages: _chatHistory,
|
|
// ),
|
|
// );
|
|
// return generated.message.content;
|
|
// }
|
|
|
|
Stream<String> _generateChatCompletionWithHistoryStream(
|
|
String userMessage,
|
|
final ollama.OllamaClient client,
|
|
) async* {
|
|
final aiStream = client.generateChatCompletionStream(
|
|
request: ollama.GenerateChatCompletionRequest(
|
|
model: _modelCopntroller.text,
|
|
messages: _chatHistory,
|
|
),
|
|
);
|
|
String text = '';
|
|
await for (final res in aiStream) {
|
|
text += (res.message.content);
|
|
yield text;
|
|
}
|
|
// print(text);
|
|
}
|
|
|
|
void _resetChat() {
|
|
setState(() {
|
|
_messages = [];
|
|
_chatHistory = [];
|
|
|
|
_loadMessages();
|
|
});
|
|
// Navigator.of(context).popAndPushNamed(
|
|
// '/mzansi-ai',
|
|
// arguments: widget.signedInUser,
|
|
// );
|
|
}
|
|
|
|
ChatTheme getChatTheme() {
|
|
return DarkChatTheme(
|
|
backgroundColor: MzanziInnovationHub.of(context)!.theme.primaryColor(),
|
|
inputBackgroundColor:
|
|
MzanziInnovationHub.of(context)!.theme.secondaryColor(),
|
|
inputTextColor: MzanziInnovationHub.of(context)!.theme.primaryColor(),
|
|
inputTextCursorColor:
|
|
MzanziInnovationHub.of(context)!.theme.primaryColor(),
|
|
primaryColor: MzanziInnovationHub.of(context)!.theme.secondaryColor(),
|
|
secondaryColor: MzanziInnovationHub.of(context)!.theme.successColor(),
|
|
errorColor: MzanziInnovationHub.of(context)!.theme.errorColor(),
|
|
sentMessageBodyTextStyle: TextStyle(
|
|
color: MzanziInnovationHub.of(context)!.theme.primaryColor(),
|
|
fontSize: 17,
|
|
fontWeight: FontWeight.w500,
|
|
fontFamily: 'Segoe UI',
|
|
),
|
|
receivedMessageBodyTextStyle: TextStyle(
|
|
color: MzanziInnovationHub.of(context)!.theme.primaryColor(),
|
|
fontSize: 17,
|
|
fontWeight: FontWeight.w500,
|
|
fontFamily: 'Segoe UI',
|
|
),
|
|
emptyChatPlaceholderTextStyle: TextStyle(
|
|
color: MzanziInnovationHub.of(context)!.theme.messageTextColor(),
|
|
fontSize: 17,
|
|
fontWeight: FontWeight.w500,
|
|
fontFamily: 'Segoe UI',
|
|
),
|
|
);
|
|
}
|
|
|
|
@override
|
|
void dispose() {
|
|
// TODO: implement dispose
|
|
super.dispose();
|
|
_modelCopntroller.dispose();
|
|
client.endSession();
|
|
}
|
|
|
|
@override
|
|
void initState() {
|
|
super.initState();
|
|
_user = types.User(
|
|
firstName: widget.signedInUser.fname,
|
|
id: widget.signedInUser.app_id, //'82091008-a484-4a89-ae75-a22bf8d6f3ac',
|
|
);
|
|
_mihAI = types.User(
|
|
firstName: "Mzansi AI",
|
|
id: const Uuid().v4(),
|
|
);
|
|
_modelCopntroller.text = 'deepseek-r1:1.5b';
|
|
// _chatHistory.add(
|
|
// ollama.Message(
|
|
// role: ollama.MessageRole.system,
|
|
// content: systemPromt,
|
|
// ),
|
|
// );
|
|
_loadMessages();
|
|
}
|
|
|
|
@override
|
|
Widget build(BuildContext context) {
|
|
return MihAppToolBody(
|
|
borderOn: false,
|
|
bodyItem: Column(
|
|
mainAxisAlignment: MainAxisAlignment.start,
|
|
mainAxisSize: MainAxisSize.max,
|
|
children: [
|
|
Row(
|
|
mainAxisAlignment: MainAxisAlignment.center,
|
|
children: [
|
|
IconButton(
|
|
onPressed: () {
|
|
if (_showModelOptions.value == true) {
|
|
setState(() {
|
|
_showModelOptions.value = false;
|
|
});
|
|
} else {
|
|
setState(() {
|
|
_showModelOptions.value = true;
|
|
});
|
|
}
|
|
},
|
|
icon: const Icon(Icons.settings),
|
|
),
|
|
Text(
|
|
"Mzansi AI",
|
|
textAlign: TextAlign.center,
|
|
style: TextStyle(
|
|
fontSize: 25,
|
|
fontWeight: FontWeight.bold,
|
|
color:
|
|
MzanziInnovationHub.of(context)!.theme.secondaryColor(),
|
|
),
|
|
),
|
|
IconButton(
|
|
onPressed: () {
|
|
_resetChat();
|
|
},
|
|
icon: const Icon(Icons.refresh),
|
|
),
|
|
],
|
|
),
|
|
ValueListenableBuilder(
|
|
valueListenable: _showModelOptions,
|
|
builder: (BuildContext context, bool value, Widget? child) {
|
|
return Visibility(
|
|
visible: value,
|
|
child: Row(
|
|
mainAxisAlignment: MainAxisAlignment.start,
|
|
children: [
|
|
Padding(
|
|
padding: const EdgeInsets.symmetric(horizontal: 25),
|
|
child: SizedBox(
|
|
width: 300,
|
|
child: MIHDropdownField(
|
|
controller: _modelCopntroller,
|
|
hintText: "AI Model",
|
|
dropdownOptions: const ['deepseek-r1:1.5b'],
|
|
required: true,
|
|
editable: true,
|
|
),
|
|
),
|
|
)
|
|
],
|
|
),
|
|
);
|
|
},
|
|
),
|
|
Expanded(
|
|
child: Chat(
|
|
messages: _messages,
|
|
// onAttachmentPressed: _handleAttachmentPressed,
|
|
// onMessageTap: _handleMessageTap,
|
|
// onPreviewDataFetched: _handlePreviewDataFetched,
|
|
onSendPressed: _handleSendPressed,
|
|
showUserAvatars: false,
|
|
showUserNames: false,
|
|
user: _user,
|
|
theme: getChatTheme(),
|
|
),
|
|
)
|
|
],
|
|
),
|
|
);
|
|
}
|
|
}
|