Native: Add configuration window in Native UI

Author: sudeeptarlekar

application/apps/indexer/gui/application/src/host/ui/menu.rs

@@ -1,7 +1,8 @@
-use egui::{MenuBar, Theme, Ui};
+use egui::{MenuBar, Popup, Theme, Ui};
 use stypes::FileFormat;
 use tokio::sync::mpsc::Sender;
 
+use super::state::HostState;
 use crate::host::{
     command::HostCommand,
     common::{parsers::ParserNames, sources::StreamNames},
@@ -24,7 +25,7 @@ impl MainMenuBar {
         Self { cmd_tx }
     }
 
-    pub fn render(&mut self, ui: &mut Ui, actions: &mut UiActions) {
+    pub fn render(&mut self, ui: &mut Ui, actions: &mut UiActions, state: &mut HostState) {
         self.handle_file_dialog(actions);
 
         MenuBar::new().ui(ui, |ui| {
@@ -154,6 +155,12 @@ impl MainMenuBar {
                     ui.ctx().set_theme(Theme::Light);
                 }
             });
+
+            ui.menu_button("Settings", |ui| {
+                if ui.button("Ai Configuration").clicked() {
+                    state.open_configuration();
+                }
+            });
         });
     }
 

application/apps/indexer/gui/application/src/host/ui/mod.rs

@@ -22,6 +22,7 @@ use crate::{
                 parsers::ParserConfig,
                 sources::{ByteSourceConfig, ProcessConfig, StreamConfig},
             },
+            settings::SystemSettingsView,
             tabs::TabType,
         },
     },
@@ -37,6 +38,7 @@ mod menu;
 pub mod multi_setup;
 mod notification;
 pub mod session_setup;
+mod settings;
 mod state;
 mod tabs;
 
@@ -48,6 +50,7 @@ pub struct Host {
     senders: UiSenders,
     menu: MainMenuBar,
     notifications: NotificationUi,
+    settings: SystemSettingsView,
     state: HostState,
     ui_actions: UiActions,
 }
@@ -78,6 +81,7 @@ impl Host {
                     senders: ui_comm.senders,
                     notifications: NotificationUi::default(),
                     state: HostState::default(),
+                    settings: SystemSettingsView::default(),
                     ui_actions: UiActions::new(tokio_handle),
                 };
 
@@ -186,9 +190,12 @@ impl Host {
 
     fn render_menu(&mut self, ui: &mut Ui) {
         let Self {
-            menu, ui_actions, ..
+            menu,
+            ui_actions,
+            state,
+            ..
         } = self;
-        menu.render(ui, ui_actions);
+        menu.render(ui, ui_actions, state);
     }
 
     fn render_tabs(&mut self, ui: &mut Ui) {
@@ -233,6 +240,8 @@ impl Host {
                 .expect("Multiple files setups with provided ID from active tab must exist")
                 .render_content(ui_actions, ui),
         }
+
+        self.settings.render_content(&mut self.state, ui);
     }
 
     fn handle_ui_actions(&mut self, ctx: &Context) {

application/apps/indexer/gui/application/src/host/ui/settings/ai_config.rs

@@ -0,0 +1,49 @@
+use crate::host::ui::state::HostState;
+use eframe::egui::Ui;
+
+#[derive(Debug)]
+pub struct AiConfiguration {
+    pub model: String,
+    pub url: String,
+    pub api_token: Option<String>,
+}
+
+impl Default for AiConfiguration {
+    fn default() -> Self {
+        Self {
+            model: "llama3.2".to_string(),
+            url: "http://localhost:11434".to_string(),
+            api_token: None,
+        }
+    }
+}
+
+impl AiConfiguration {
+    pub fn render_content(&mut self, state: &mut HostState, ui: &mut Ui) {
+        if state.configuration_open {
+            egui::Window::new("Configuration")
+                .collapsible(false)
+                .title_bar(true)
+                .open(&mut state.configuration_open)
+                .show(ui.ctx(), |ui| {
+                    ui.horizontal(|ui| {
+                        let label = ui.label("Model Name: ");
+                        ui.text_edit_singleline(&mut self.model)
+                            .labelled_by(label.id);
+                    });
+
+                    ui.horizontal(|ui| {
+                        let label = ui.label("URL: ");
+                        ui.text_edit_singleline(&mut self.url).labelled_by(label.id);
+                    });
+
+                    ui.horizontal(|ui| {
+                        let label = ui.label("API Token: ");
+                        let mut token = self.api_token.clone().unwrap_or_default();
+                        ui.text_edit_singleline(&mut token).labelled_by(label.id);
+                        self.api_token = if token.is_empty() { None } else { Some(token) };
+                    });
+                });
+        }
+    }
+}

application/apps/indexer/gui/application/src/host/ui/settings/mod.rs

@@ -0,0 +1,25 @@
+use egui::Ui;
+
+use crate::host::ui::settings::ai_config::AiConfiguration;
+use crate::host::ui::state::HostState;
+
+pub mod ai_config;
+
+#[derive(Debug)]
+pub struct SystemSettingsView {
+    pub configuration: AiConfiguration,
+}
+
+impl Default for SystemSettingsView {
+    fn default() -> Self {
+        Self {
+            configuration: AiConfiguration::default(),
+        }
+    }
+}
+
+impl SystemSettingsView {
+    pub fn render_content(&mut self, state: &mut HostState, ui: &mut Ui) {
+        self.configuration.render_content(state, ui);
+    }
+}

application/apps/indexer/gui/application/src/host/ui/state.rs

@@ -9,6 +9,7 @@ use crate::{
         ui::{
             multi_setup::{MultiFileSetup, state::MultiFileState},
             session_setup::{SessionSetup, state::SessionSetupState},
+            settings::ai_config::AiConfiguration,
             tabs::TabType,
         },
     },
@@ -20,10 +21,12 @@ pub const HOME_TAB_IDX: usize = 0;
 #[derive(Debug)]
 pub struct HostState {
     pub active_tab_idx: usize,
+    pub configuration_open: bool,
     pub tabs: Vec<TabType>,
     pub sessions: HashMap<Uuid, Session>,
     pub session_setups: HashMap<Uuid, SessionSetup>,
     pub multi_setups: HashMap<Uuid, MultiFileSetup>,
+    pub ai_configuration: AiConfiguration,
 }
 
 impl HostState {
@@ -161,6 +164,10 @@ impl HostState {
         self.multi_setups.remove(&setup_id);
     }
 
+    pub fn open_configuration(&mut self) {
+        self.configuration_open = true;
+    }
+
     fn update_current_tab_on_close(&mut self, removed_idx: usize) {
         if self.active_tab_idx == removed_idx {
             self.active_tab_idx = HOME_TAB_IDX;
@@ -175,10 +182,12 @@ impl Default for HostState {
     fn default() -> Self {
         Self {
             active_tab_idx: 0,
+            configuration_open: false,
             tabs: vec![TabType::Home],
             sessions: HashMap::new(),
             session_setups: HashMap::new(),
             multi_setups: HashMap::new(),
+            ai_configuration: AiConfiguration::default(),
         }
     }
 }

application/apps/indexer/gui/application/src/host/ui/tabs.rs

@@ -24,10 +24,12 @@ pub enum TabType {
 pub fn render_all_tabs(state: &mut HostState, actions: &mut UiActions, ui: &mut Ui) {
     let HostState {
         active_tab_idx,
+        configuration_open,
         tabs,
         sessions,
         session_setups,
         multi_setups,
+        ai_configuration,
     } = state;
 
     for (idx, tab) in tabs.iter().enumerate() {