backend: separate App and server, add websocket endpoint, accept data directory command argument. initialize sqlalchemy database and models in new module. frontend: create backend class for managing ws connection, move ui to chat_screen, create scaffold chat and agent classes.

This commit is contained in:
Shivering Ocelot committed 2025-05-06 17:50:57 -05:00
1 parent 1e82796db1
commit 44f0b3589c
19 files changed
+936 -122

No files matched your search

+9
View File
@@ -0,0 +1,9 @@
import 'chat.dart';
class Agent {
void receivedChatMessage(Chat chat, String message) {
// Simple response from the agent
String responseMessage = "Agent received: " + message;
chat.addAgentMessage(responseMessage);
}
}
+49
View File
@@ -0,0 +1,49 @@
import 'dart:html';
import 'dart:convert';
class Backend {
static WebSocket? _socket;
static String? _url;
static void setup(String url) {
_url = url;
connect();
}
static void auth(username) {
_socket!.send(
json.encode({'username': username})
);
}
static void connect() {
_socket = WebSocket(_url!);
_socket!.onOpen.listen((event) {
print('Connected to the WebSocket server');
auth("the-user");
ping();
});
_socket!.onError.listen((event) {
print('Error: $event');
_reconnect();
});
_socket!.onClose.listen((event) {
print('Disconnected. Attempting to reconnect...');
_reconnect();
});
}
static void _reconnect() {
Future.delayed(Duration(seconds: 5), () {
connect();
});
}
static void ping() {
if (_socket != null && _socket!.readyState == WebSocket.OPEN) {
final data = json.encode({'path': 'ping'});
_socket!.send(data);
print('Ping sent: $data');
}
}
}
+31
View File
@@ -0,0 +1,31 @@
import 'agent.dart'; // Import the Agent class
class Message {
final String content;
final String sender;
Message(this.content, this.sender);
}
class Chat {
final String id;
final List<Message> _messages = [];
final Agent agent = Agent(); // Create an instance of Agent
Chat(this.id);
List<Message> get messages => List.unmodifiable(_messages);
void addUserMessage(String message) {
_messages.add(Message(message, 'user'));
agent.receivedChatMessage(this, message); // Call agent's method
}
void addAgentMessage(String message) {
_messages.add(Message(message, 'agent'));
}
void clearMessages() {
_messages.clear();
}
}
+80
View File
@@ -0,0 +1,80 @@
import 'package:flutter/material.dart';
import 'chat.dart'; // Import the new Chat class
class ChatScreen extends StatefulWidget {
const ChatScreen({super.key});
@override
State<ChatScreen> createState() => _ChatScreenState();
}
class _ChatScreenState extends State<ChatScreen> {
final Chat _chat = Chat('chat1'); // Create an instance of Chat
final TextEditingController _controller = TextEditingController();
void _sendMessage() {
if (_controller.text.isNotEmpty) {
setState(() {
_chat.addUserMessage(_controller.text); // Use the Chat instance to add the message
_controller.clear();
});
}
}
@override
Widget build(BuildContext context) {
final width = MediaQuery.of(context).size.width;
final isBigScreen = width > 600; // Define threshold for small screens
return Scaffold(
appBar: AppBar(
title: const Text('Chat'),
),
body: Column(
children: [
Expanded(
child: ListView.builder(
itemCount: _chat.messages.length,
itemBuilder: (context, index) {
final message = _chat.messages[index];
// Align all messages to left for big screens
final align = message.sender == 'user' && !isBigScreen ? CrossAxisAlignment.end : CrossAxisAlignment.start;
return Row(
mainAxisAlignment: align == CrossAxisAlignment.end ? MainAxisAlignment.end : MainAxisAlignment.start,
children: [
Container(
padding: const EdgeInsets.all(8.0),
margin: const EdgeInsets.symmetric(vertical: 4.0),
child: Text(
'${message.content}',
style: const TextStyle(color: Colors.white),
),
),
],
);
},
),
),
Padding(
padding: const EdgeInsets.all(8.0),
child: Row(
children: [
Expanded(
child: TextField(
controller: _controller,
decoration: const InputDecoration(hintText: 'Send a message'),
onSubmitted: (_) => _sendMessage(),
),
),
IconButton(
icon: const Icon(Icons.send),
onPressed: _sendMessage,
),
],
),
),
],
),
);
}
}
+29 -108
View File
@@ -1,122 +1,43 @@
import 'dart:html' as html;
import 'dart:io' as io;
import 'package:flutter/material.dart';
import 'chat_screen.dart';
import 'backend.dart';
class Config {
static String? backendUrl; // Declare it as nullable
static void loadConfig() {
// Check the platform for web or other
if (html.window.location.origin.isNotEmpty) {
// For web applications
backendUrl = html.window.location.origin; // Get the server URL from window.location
} else if (io.Platform.environment.containsKey('BACKEND_URL')) {
// For Android or other platforms, check for the environment variable
backendUrl = io.Platform.environment['BACKEND_URL'];
} else {
// If no value is determined, set it to null
backendUrl = null;
}
}
}
void main() {
Config.loadConfig(); // Load the configuration before running the app
Backend.setup('${Config.backendUrl}/ws');
runApp(const MyApp());
}
class MyApp extends StatelessWidget {
const MyApp({super.key});
// This widget is the root of your application.
@override
Widget build(BuildContext context) {
return MaterialApp(
title: 'Flutter Demo',
theme: ThemeData(
// This is the theme of your application.
//
// TRY THIS: Try running your application with "flutter run". You'll see
// the application has a purple toolbar. Then, without quitting the app,
// try changing the seedColor in the colorScheme below to Colors.green
// and then invoke "hot reload" (save your changes or press the "hot
// reload" button in a Flutter-supported IDE, or press "r" if you used
// the command line to start the app).
//
// Notice that the counter didn't reset back to zero; the application
// state is not lost during the reload. To reset the state, use hot
// restart instead.
//
// This works for code too, not just values: Most code changes can be
// tested with just a hot reload.
colorScheme: ColorScheme.fromSeed(seedColor: Colors.deepPurple),
),
home: const MyHomePage(title: 'Flutter Demo Home Page'),
title: 'Flutter Chat',
theme: ThemeData.dark(),
home: const ChatScreen(),
);
}
}
class MyHomePage extends StatefulWidget {
const MyHomePage({super.key, required this.title});
// This widget is the home page of your application. It is stateful, meaning
// that it has a State object (defined below) that contains fields that affect
// how it looks.
// This class is the configuration for the state. It holds the values (in this
// case the title) provided by the parent (in this case the App widget) and
// used by the build method of the State. Fields in a Widget subclass are
// always marked "final".
final String title;
@override
State<MyHomePage> createState() => _MyHomePageState();
}
class _MyHomePageState extends State<MyHomePage> {
int _counter = 0;
void _incrementCounter() {
setState(() {
// This call to setState tells the Flutter framework that something has
// changed in this State, which causes it to rerun the build method below
// so that the display can reflect the updated values. If we changed
// _counter without calling setState(), then the build method would not be
// called again, and so nothing would appear to happen.
_counter++;
});
}
@override
Widget build(BuildContext context) {
// This method is rerun every time setState is called, for instance as done
// by the _incrementCounter method above.
//
// The Flutter framework has been optimized to make rerunning build methods
// fast, so that you can just rebuild anything that needs updating rather
// than having to individually change instances of widgets.
return Scaffold(
appBar: AppBar(
// TRY THIS: Try changing the color here to a specific color (to
// Colors.amber, perhaps?) and trigger a hot reload to see the AppBar
// change color while the other colors stay the same.
backgroundColor: Theme.of(context).colorScheme.inversePrimary,
// Here we take the value from the MyHomePage object that was created by
// the App.build method, and use it to set our appbar title.
title: Text(widget.title),
),
body: Center(
// Center is a layout widget. It takes a single child and positions it
// in the middle of the parent.
child: Column(
// Column is also a layout widget. It takes a list of children and
// arranges them vertically. By default, it sizes itself to fit its
// children horizontally, and tries to be as tall as its parent.
//
// Column has various properties to control how it sizes itself and
// how it positions its children. Here we use mainAxisAlignment to
// center the children vertically; the main axis here is the vertical
// axis because Columns are vertical (the cross axis would be
// horizontal).
//
// TRY THIS: Invoke "debug painting" (choose the "Toggle Debug Paint"
// action in the IDE, or press "p" in the console), to see the
// wireframe for each widget.
mainAxisAlignment: MainAxisAlignment.center,
children: <Widget>[
const Text('You have pushed the button this many times:'),
Text(
'$_counter',
style: Theme.of(context).textTheme.headlineMedium,
),
],
),
),
floatingActionButton: FloatingActionButton(
onPressed: _incrementCounter,
tooltip: 'Increment',
child: const Icon(Icons.add),
), // This trailing comma makes auto-formatting nicer for build methods.
);
}
}
}