541 lines
19 KiB
Dart
541 lines
19 KiB
Dart
import 'package:flutter/material.dart';
|
|
import 'package:ai_chat_lab/services/convai_service.dart';
|
|
import 'package:ai_chat_lab/services/storage_service.dart';
|
|
import 'package:ai_chat_lab/screens/settings_screen.dart';
|
|
import 'package:ai_chat_lab/models/chat_message.dart';
|
|
import 'dart:async';
|
|
|
|
class ChatScreen extends StatefulWidget {
|
|
const ChatScreen({super.key});
|
|
|
|
@override
|
|
State<ChatScreen> createState() => _ChatScreenState();
|
|
}
|
|
|
|
class _ChatScreenState extends State<ChatScreen> {
|
|
final TextEditingController _messageController = TextEditingController();
|
|
final TextEditingController _participantController = TextEditingController();
|
|
final ScrollController _scrollController = ScrollController();
|
|
final ConvaiService _convaiService = ConvaiService();
|
|
final StorageService _storageService = StorageService();
|
|
|
|
List<ChatMessage> messages = [];
|
|
bool isLoading = false;
|
|
String participantId = '';
|
|
bool _supervisedMode = false;
|
|
bool _initialTurnPending = false;
|
|
String _initialPrompt = '';
|
|
bool _timedExperiment = false;
|
|
int _experimentDuration = 5;
|
|
DateTime? _experimentStartTime;
|
|
Timer? _experimentTimer;
|
|
bool _experimentExpired = false;
|
|
Duration _timeRemaining = Duration.zero;
|
|
|
|
@override
|
|
void initState() {
|
|
super.initState();
|
|
_loadSettings();
|
|
}
|
|
|
|
Future<void> _loadSettings() async {
|
|
await _convaiService.loadSettings();
|
|
final supervised = await _convaiService.isSupervisedModeEnabled();
|
|
final initPrompt = await _convaiService.getCharacterInitialPrompt(_convaiService.characterId);
|
|
final timedExperiment = await _convaiService.isTimedExperimentEnabled();
|
|
final duration = await _convaiService.getExperimentDurationMinutes();
|
|
setState(() {
|
|
_supervisedMode = supervised;
|
|
_initialPrompt = initPrompt;
|
|
_initialTurnPending = supervised; // first message gated
|
|
_timedExperiment = timedExperiment;
|
|
_experimentDuration = duration;
|
|
});
|
|
}
|
|
|
|
void _startExperiment() {
|
|
if (_timedExperiment && _experimentStartTime == null) {
|
|
_experimentStartTime = DateTime.now();
|
|
_startExperimentTimer();
|
|
}
|
|
}
|
|
|
|
void _startExperimentTimer() {
|
|
_experimentTimer?.cancel();
|
|
_experimentTimer = Timer.periodic(const Duration(seconds: 1), (timer) {
|
|
final now = DateTime.now();
|
|
final elapsed = now.difference(_experimentStartTime!);
|
|
final totalDuration = Duration(minutes: _experimentDuration);
|
|
|
|
if (elapsed >= totalDuration) {
|
|
setState(() {
|
|
_experimentExpired = true;
|
|
_timeRemaining = Duration.zero;
|
|
});
|
|
_experimentTimer?.cancel();
|
|
} else {
|
|
setState(() {
|
|
_timeRemaining = totalDuration - elapsed;
|
|
});
|
|
}
|
|
});
|
|
}
|
|
|
|
Future<void> _sendMessage() async {
|
|
final text = _messageController.text.trim();
|
|
if (text.isEmpty || isLoading) return;
|
|
|
|
// Start experiment timer on first message
|
|
if (messages.isEmpty) {
|
|
_startExperiment();
|
|
}
|
|
|
|
setState(() {
|
|
messages.add(ChatMessage(text: text, isUser: true));
|
|
isLoading = true;
|
|
});
|
|
|
|
_messageController.clear();
|
|
_scrollToBottom();
|
|
|
|
try {
|
|
final response = await _convaiService.sendMessage(text);
|
|
|
|
setState(() {
|
|
messages.add(ChatMessage(text: response, isUser: false));
|
|
isLoading = false;
|
|
});
|
|
|
|
if (await _storageService.isSavingEnabled()) {
|
|
await _storageService.saveConversation(participantId, text, response);
|
|
}
|
|
} catch (e) {
|
|
setState(() {
|
|
messages.add(ChatMessage(text: 'Error: ${e.toString()}', isUser: false));
|
|
isLoading = false;
|
|
});
|
|
}
|
|
|
|
_scrollToBottom();
|
|
}
|
|
|
|
Future<void> _sendInitialPrompt() async {
|
|
// If no initial prompt, just allow typing without sending anything
|
|
if (_initialPrompt.trim().isEmpty) {
|
|
setState(() {
|
|
_initialTurnPending = false;
|
|
});
|
|
_startExperiment();
|
|
return;
|
|
}
|
|
|
|
if (isLoading) return;
|
|
setState(() {
|
|
isLoading = true;
|
|
_initialTurnPending = false;
|
|
});
|
|
|
|
_startExperiment();
|
|
|
|
try {
|
|
final response = await _convaiService.sendMessage(_initialPrompt);
|
|
setState(() {
|
|
// Do not attribute initial prompt as a user message in the chat transcript
|
|
messages.add(ChatMessage(text: response, isUser: false));
|
|
isLoading = false;
|
|
});
|
|
|
|
if (await _storageService.isSavingEnabled()) {
|
|
await _storageService.saveConversation(participantId, _initialPrompt, response);
|
|
}
|
|
} catch (e) {
|
|
setState(() {
|
|
messages.add(ChatMessage(text: 'Error: ${e.toString()}', isUser: false));
|
|
isLoading = false;
|
|
});
|
|
}
|
|
|
|
_scrollToBottom();
|
|
}
|
|
|
|
void _scrollToBottom() {
|
|
WidgetsBinding.instance.addPostFrameCallback((_) {
|
|
if (_scrollController.hasClients) {
|
|
_scrollController.animateTo(
|
|
_scrollController.position.maxScrollExtent,
|
|
duration: const Duration(milliseconds: 300),
|
|
curve: Curves.easeOut,
|
|
);
|
|
}
|
|
});
|
|
}
|
|
|
|
void _clearChat() async {
|
|
final bool? shouldClear = await showDialog<bool>(
|
|
context: context,
|
|
builder: (BuildContext context) {
|
|
return AlertDialog(
|
|
title: const Text('Clear Chat'),
|
|
content: const Text(
|
|
'This will clear all messages and start a new session with the character. Are you sure?',
|
|
),
|
|
actions: [
|
|
TextButton(
|
|
onPressed: () => Navigator.of(context).pop(false),
|
|
child: const Text('Cancel'),
|
|
),
|
|
TextButton(
|
|
onPressed: () => Navigator.of(context).pop(true),
|
|
style: TextButton.styleFrom(
|
|
foregroundColor: Colors.red,
|
|
),
|
|
child: const Text('Clear'),
|
|
),
|
|
],
|
|
);
|
|
},
|
|
);
|
|
|
|
if (shouldClear == true) {
|
|
setState(() {
|
|
messages.clear();
|
|
_initialTurnPending = _supervisedMode; // reset gating
|
|
_experimentStartTime = null;
|
|
_experimentExpired = false;
|
|
_timeRemaining = Duration.zero;
|
|
});
|
|
_experimentTimer?.cancel();
|
|
_convaiService.resetSession();
|
|
ScaffoldMessenger.of(context).showSnackBar(
|
|
const SnackBar(
|
|
content: Text('Chat cleared and new session started'),
|
|
duration: Duration(seconds: 2),
|
|
),
|
|
);
|
|
}
|
|
}
|
|
|
|
String _formatDuration(Duration duration) {
|
|
String twoDigits(int n) => n.toString().padLeft(2, '0');
|
|
final minutes = twoDigits(duration.inMinutes.remainder(60));
|
|
final seconds = twoDigits(duration.inSeconds.remainder(60));
|
|
return '$minutes:$seconds';
|
|
}
|
|
|
|
@override
|
|
Widget build(BuildContext context) {
|
|
return Scaffold(
|
|
backgroundColor: const Color(0xFFF8FAFC),
|
|
appBar: AppBar(
|
|
title: const Text(
|
|
'Convai Chat',
|
|
style: TextStyle(
|
|
fontWeight: FontWeight.w600,
|
|
letterSpacing: 1,
|
|
),
|
|
),
|
|
backgroundColor: Colors.white,
|
|
elevation: 0,
|
|
foregroundColor: const Color(0xFF2D3748),
|
|
actions: [
|
|
IconButton(
|
|
icon: const Icon(Icons.clear_all),
|
|
onPressed: _clearChat,
|
|
tooltip: 'Clear Chat',
|
|
),
|
|
IconButton(
|
|
icon: const Icon(Icons.settings),
|
|
onPressed: () {
|
|
Navigator.push(
|
|
context,
|
|
MaterialPageRoute(
|
|
builder: (context) => const SettingsScreen(),
|
|
),
|
|
).then((_) => _loadSettings());
|
|
},
|
|
tooltip: 'Settings',
|
|
),
|
|
],
|
|
),
|
|
body: Column(
|
|
children: [
|
|
Container(
|
|
padding: const EdgeInsets.all(16),
|
|
decoration: const BoxDecoration(
|
|
color: Colors.white,
|
|
border: Border(
|
|
bottom: BorderSide(color: Color(0xFFE0E7EF)),
|
|
),
|
|
),
|
|
child: Column(
|
|
crossAxisAlignment: CrossAxisAlignment.start,
|
|
children: [
|
|
TextField(
|
|
controller: _participantController,
|
|
onChanged: (value) => participantId = value,
|
|
decoration: InputDecoration(
|
|
labelText: 'Participant ID',
|
|
hintText: 'Enter participant ID for this session',
|
|
border: OutlineInputBorder(
|
|
borderRadius: BorderRadius.circular(8),
|
|
borderSide: const BorderSide(color: Color(0xFFCBD5E1)),
|
|
),
|
|
focusedBorder: OutlineInputBorder(
|
|
borderRadius: BorderRadius.circular(8),
|
|
borderSide: const BorderSide(color: Color(0xFF4F8CFF), width: 2),
|
|
),
|
|
contentPadding: const EdgeInsets.symmetric(horizontal: 16, vertical: 12),
|
|
filled: true,
|
|
fillColor: const Color(0xFFF8FAFC),
|
|
),
|
|
),
|
|
if (_timedExperiment && _experimentStartTime != null && !_experimentExpired)
|
|
Padding(
|
|
padding: const EdgeInsets.only(top: 8.0),
|
|
child: Row(
|
|
children: [
|
|
const Icon(Icons.timer, size: 16, color: Color(0xFF4F8CFF)),
|
|
const SizedBox(width: 6),
|
|
Text(
|
|
'Time remaining: ${_formatDuration(_timeRemaining)}',
|
|
style: const TextStyle(fontSize: 12, color: Color(0xFF64748B), fontWeight: FontWeight.w500),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
if (_supervisedMode)
|
|
Padding(
|
|
padding: const EdgeInsets.only(top: 8.0),
|
|
child: Row(
|
|
children: [
|
|
const Icon(Icons.flag, size: 16, color: Color(0xFF4F8CFF)),
|
|
const SizedBox(width: 6),
|
|
Expanded(
|
|
child: Text(
|
|
_initialTurnPending
|
|
? 'Supervised Mode: Press "Start Experiment" to begin.'
|
|
: 'Supervised Mode: You may now type.',
|
|
style: const TextStyle(fontSize: 12, color: Color(0xFF64748B)),
|
|
),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
|
|
if (_timedExperiment && _experimentExpired)
|
|
Expanded(
|
|
child: Container(
|
|
margin: const EdgeInsets.all(16),
|
|
decoration: BoxDecoration(
|
|
color: Colors.white,
|
|
borderRadius: BorderRadius.circular(12),
|
|
border: Border.all(color: const Color(0xFFE0E7EF)),
|
|
boxShadow: [
|
|
BoxShadow(
|
|
color: Colors.black.withOpacity(0.05),
|
|
blurRadius: 10,
|
|
offset: const Offset(0, 2),
|
|
),
|
|
],
|
|
),
|
|
child: const Center(
|
|
child: Column(
|
|
mainAxisAlignment: MainAxisAlignment.center,
|
|
children: [
|
|
Icon(Icons.timer_off, size: 64, color: Color(0xFF64748B)),
|
|
SizedBox(height: 16),
|
|
Text(
|
|
'Experiment Time Expired',
|
|
style: TextStyle(
|
|
fontSize: 20,
|
|
fontWeight: FontWeight.w600,
|
|
color: Color(0xFF2D3748),
|
|
),
|
|
),
|
|
SizedBox(height: 8),
|
|
Text(
|
|
'The chat is now hidden as the time limit has been reached.',
|
|
style: TextStyle(
|
|
fontSize: 14,
|
|
color: Color(0xFF64748B),
|
|
),
|
|
textAlign: TextAlign.center,
|
|
),
|
|
],
|
|
),
|
|
),
|
|
),
|
|
)
|
|
else
|
|
Expanded(
|
|
child: Container(
|
|
margin: const EdgeInsets.all(16),
|
|
decoration: BoxDecoration(
|
|
color: Colors.white,
|
|
borderRadius: BorderRadius.circular(12),
|
|
border: Border.all(color: const Color(0xFFE0E7EF)),
|
|
boxShadow: [
|
|
BoxShadow(
|
|
color: Colors.black.withOpacity(0.05),
|
|
blurRadius: 10,
|
|
offset: const Offset(0, 2),
|
|
),
|
|
],
|
|
),
|
|
child: Column(
|
|
children: [
|
|
Expanded(
|
|
child: ListView.builder(
|
|
controller: _scrollController,
|
|
padding: const EdgeInsets.all(16),
|
|
itemCount: messages.length + (isLoading ? 1 : 0),
|
|
itemBuilder: (context, index) {
|
|
if (index == messages.length && isLoading) {
|
|
return const Padding(
|
|
padding: EdgeInsets.all(8.0),
|
|
child: Row(
|
|
children: [
|
|
SizedBox(
|
|
width: 20,
|
|
height: 20,
|
|
child: CircularProgressIndicator(strokeWidth: 2),
|
|
),
|
|
SizedBox(width: 12),
|
|
Text('Thinking...'),
|
|
],
|
|
),
|
|
);
|
|
}
|
|
|
|
final message = messages[index];
|
|
return Container(
|
|
margin: const EdgeInsets.only(bottom: 8),
|
|
child: Row(
|
|
mainAxisAlignment: message.isUser
|
|
? MainAxisAlignment.end
|
|
: MainAxisAlignment.start,
|
|
children: [
|
|
Container(
|
|
constraints: BoxConstraints(
|
|
maxWidth: MediaQuery.of(context).size.width * 0.7,
|
|
),
|
|
padding: const EdgeInsets.symmetric(
|
|
horizontal: 16,
|
|
vertical: 12,
|
|
),
|
|
decoration: BoxDecoration(
|
|
color: message.isUser
|
|
? const Color(0xFF4F8CFF)
|
|
: const Color(0xFFF1F5FB),
|
|
borderRadius: BorderRadius.circular(16).copyWith(
|
|
bottomRight: message.isUser
|
|
? const Radius.circular(4)
|
|
: const Radius.circular(16),
|
|
bottomLeft: message.isUser
|
|
? const Radius.circular(16)
|
|
: const Radius.circular(4),
|
|
),
|
|
),
|
|
child: Text(
|
|
message.text,
|
|
style: TextStyle(
|
|
color: message.isUser
|
|
? Colors.white
|
|
: const Color(0xFF2D3748),
|
|
fontSize: 16,
|
|
height: 1.4,
|
|
),
|
|
),
|
|
),
|
|
],
|
|
),
|
|
);
|
|
},
|
|
),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
),
|
|
|
|
if (!(_timedExperiment && _experimentExpired))
|
|
Container(
|
|
padding: const EdgeInsets.fromLTRB(16, 0, 16, 16),
|
|
child: Row(
|
|
children: [
|
|
if (_supervisedMode && _initialTurnPending)
|
|
Expanded(
|
|
child: SizedBox(
|
|
height: 48,
|
|
child: ElevatedButton.icon(
|
|
onPressed: isLoading ? null : _sendInitialPrompt,
|
|
icon: const Icon(Icons.play_arrow),
|
|
label: const Text('Start Experiment'),
|
|
style: ElevatedButton.styleFrom(
|
|
backgroundColor: const Color(0xFF4F8CFF),
|
|
foregroundColor: Colors.white,
|
|
shape: RoundedRectangleBorder(
|
|
borderRadius: BorderRadius.circular(12),
|
|
),
|
|
),
|
|
),
|
|
),
|
|
)
|
|
else ...[
|
|
Expanded(
|
|
child: TextField(
|
|
controller: _messageController,
|
|
decoration: InputDecoration(
|
|
hintText: 'Type your message...',
|
|
border: OutlineInputBorder(
|
|
borderRadius: BorderRadius.circular(24),
|
|
borderSide: const BorderSide(color: Color(0xFFCBD5E1)),
|
|
),
|
|
focusedBorder: OutlineInputBorder(
|
|
borderRadius: BorderRadius.circular(24),
|
|
borderSide: const BorderSide(color: Color(0xFF4F8CFF), width: 2),
|
|
),
|
|
contentPadding: const EdgeInsets.symmetric(horizontal: 20, vertical: 12),
|
|
filled: true,
|
|
fillColor: Colors.white,
|
|
),
|
|
maxLines: 5,
|
|
minLines: 1,
|
|
textCapitalization: TextCapitalization.sentences,
|
|
textInputAction: TextInputAction.send,
|
|
onSubmitted: (_) => _sendMessage(),
|
|
),
|
|
),
|
|
const SizedBox(width: 8),
|
|
Container(
|
|
decoration: const BoxDecoration(
|
|
color: Color(0xFF4F8CFF),
|
|
shape: BoxShape.circle,
|
|
),
|
|
child: IconButton(
|
|
icon: const Icon(Icons.send, color: Colors.white),
|
|
onPressed: isLoading ? null : _sendMessage,
|
|
),
|
|
),
|
|
],
|
|
],
|
|
),
|
|
),
|
|
],
|
|
),
|
|
);
|
|
}
|
|
|
|
@override
|
|
void dispose() {
|
|
_experimentTimer?.cancel();
|
|
_messageController.dispose();
|
|
_participantController.dispose();
|
|
_scrollController.dispose();
|
|
super.dispose();
|
|
}
|
|
}
|