ethanglide/groq-model-comparison
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Overhaul UI

Browse Source
This commit is contained in:
Ethan 2024-12-27 13:38:08 -05:00
parent 809ab9a979
commit ba333b928e
4 changed files with 80 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
rust-learning/src/gui/model_response.rs
View File

@ -5,16 +5,16 @@ use crate::groq::GROQ_CLIENT;
pub struct ModelResponse { pub struct ModelResponse {
pub name: String, pub name: String,
pub message: String, pub message: String,
pub status: i32, pub tokens: i32,
pub time: f32, pub time: f32,
} }
impl ModelResponse { impl ModelResponse {
pub fn new(name: String, message: String, status: i32, time: f32) -> Self { pub fn new(name: String, message: String, tokens: i32, time: f32) -> Self {
Self { Self {
name, name,
message, message,
status, tokens,
time, time,
} }
} }
@ -22,40 +22,43 @@ impl ModelResponse {
pub async fn chat_completion(&mut self, prompt: String) { pub async fn chat_completion(&mut self, prompt: String) {
let response = GROQ_CLIENT.chat_completion(self.name.clone(), prompt).await; let response = GROQ_CLIENT.chat_completion(self.name.clone(), prompt).await;
self.message = response.choices[0].message.content.clone(); self.message = response.choices[0].message.content.clone();
self.status = 200; self.tokens = response.usage.total_tokens;
self.time = response.usage.total_time; self.time = response.usage.total_time;
} }
} }
impl Widget for &ModelResponse { impl Widget for &ModelResponse {
fn ui(self, ui: &mut Ui) -> Response { fn ui(self, ui: &mut Ui) -> Response {
egui::Frame::none() ui.vertical(|ui| {
.inner_margin(8.0) egui::Frame::none()
.rounding(4.0) .inner_margin(8.0)
.fill(egui::Color32::DARK_GRAY) .rounding(4.0)
.show(ui, |ui| { .fill(egui::Color32::DARK_GRAY)
ui.label(RichText::new(&self.name).strong()); .show(ui, |ui| {
ui.label(RichText::new(&self.name).strong());
ui.horizontal(|ui| { ui.horizontal_wrapped(|ui| {
ui.horizontal(|ui| { ui.horizontal(|ui| {
ui.label("Status:"); ui.label("Tokens:");
ui.label(self.status.to_string()); ui.label(self.tokens.to_string());
});
ui.horizontal(|ui| {
ui.label("Time:");
ui.label(self.time.to_string());
});
}); });
ui.horizontal(|ui| { egui::Frame::none()
ui.label("Time:"); .inner_margin(8.0)
ui.label(self.time.to_string()); .rounding(4.0)
}); .fill(egui::Color32::WHITE)
}); .show(ui, |ui| {
ui.label(
egui::Frame::none() egui::RichText::new(&self.message).color(egui::Color32::BLACK),
.inner_margin(8.0) );
.rounding(4.0) });
.fill(egui::Color32::WHITE) })
.show(ui, |ui| { }).response
ui.label(egui::RichText::new(&self.message).color(egui::Color32::BLACK));
});
})
.response
} }
} }

35
rust-learning/src/gui/model_selection.rs
View File

@ -1,4 +1,4 @@
use eframe::egui::{self, Response, Ui, Widget}; use eframe::egui::{self, Response, RichText, Ui, Widget};
use super::{model_response::ModelResponse, state::AppState}; use super::{model_response::ModelResponse, state::AppState};
@ -33,28 +33,37 @@ impl<'a> Widget for ModelSelection<'a> {
self.app_state.selected_models.push(ModelResponse { self.app_state.selected_models.push(ModelResponse {
name: model.clone(), name: model.clone(),
message: "No message".to_string(), message: "No message".to_string(),
status: 0, tokens: 0,
time: 0.0, time: 0.0,
}); });
} }
} }
}); });
let selection_frame = egui::Frame::none()
.inner_margin(4.0)
.rounding(4.0)
.fill(egui::Color32::DARK_GRAY);
if self.app_state.selected_models.len() == 0 {
selection_frame.show(ui, |ui| {
ui.label(RichText::new("No models selected").strong());
});
return;
}
let mut model_to_remove: Option<String> = None; let mut model_to_remove: Option<String> = None;
for model in &self.app_state.selected_models { for model in &self.app_state.selected_models {
egui::Frame::none() selection_frame.show(ui, |ui| {
.inner_margin(8.0) ui.horizontal(|ui| {
.rounding(4.0) ui.label(RichText::new(model.name.clone()).strong());
.fill(egui::Color32::WHITE) if ui.button("X").clicked() {
.show(ui, |ui| { model_to_remove.replace(model.name.clone());
ui.horizontal(|ui| { }
ui.label(model.name.clone());
if ui.button("X").clicked() {
model_to_remove.replace(model.name.clone());
}
});
}); });
});
} }
if let Some(model) = model_to_remove { if let Some(model) = model_to_remove {

9
rust-learning/src/gui/prompt_input.rs
View File

@ -1,4 +1,4 @@
use eframe::egui::{Response, Ui, Widget}; use eframe::egui::{Response, TextEdit, Ui, Vec2, Widget};
use super::state::AppState; use super::state::AppState;
@ -15,8 +15,11 @@ impl<'a> PromptInput<'a> {
impl<'a> Widget for PromptInput<'a> { impl<'a> Widget for PromptInput<'a> {
fn ui(self, ui: &mut Ui) -> Response { fn ui(self, ui: &mut Ui) -> Response {
let label = ui.label("Prompt:"); let label = ui.label("Prompt:");
ui.text_edit_multiline(&mut self.app_state.prompt_input) ui.add_sized(
.labelled_by(label.id); Vec2::new(ui.available_width(), 100.0),
TextEdit::multiline(&mut self.app_state.prompt_input),
)
.labelled_by(label.id);
ui.horizontal(|ui| { ui.horizontal(|ui| {
if ui.button("Submit").clicked() { if ui.button("Submit").clicked() {

25
rust-learning/src/gui/state.rs
View File

@ -1,4 +1,7 @@
use eframe::{egui, CreationContext}; use eframe::{
egui::{self, Layout, Separator},
CreationContext,
};
use super::{model_response::ModelResponse, model_selection, prompt_input}; use super::{model_response::ModelResponse, model_selection, prompt_input};
@ -38,10 +41,22 @@ impl eframe::App for AppState {
ui.add(prompt_input::PromptInput::new(self)); ui.add(prompt_input::PromptInput::new(self));
ui.horizontal(|ui| { ui.add(Separator::default().spacing(16.0));
for model in &self.selected_models {
ui.vertical(|ui| ui.add(model)); let size = ui.available_size_before_wrap();
} let widget_size = egui::Vec2 {
x: size.x / 4.0,
y: size.y,
};
ui.allocate_ui_with_layout(size, Layout::left_to_right(egui::Align::Min), |ui| {
egui::ScrollArea::both().show(ui, |ui| {
ui.horizontal_top(|ui| {
for model in &self.selected_models {
ui.add_sized(widget_size, model);
}
});
});
}); });
}); });
} }