xiaji/flomo-ai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: 包含 Kotlin 原始代码和 Rust Windows 桌面版

Browse Source 
- flomo-ai/: Android Kotlin 原始项目
- flomo-ai-desktop/: Rust + egui Windows 桌面移植版
  - LLM API 调用、提示词管理、主题切换、配置持久化
  - MinGW 工具链编译，无控制台窗口
This commit is contained in:
xiaji
2026-04-04 17:32:49 +08:00
commit b4908fcc2d
138 changed files with 6815 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

217
flomo-ai-desktop/src/pages/main_page.rs Normal file
View File

@@ -0,0 +1,217 @@
use egui::{Ui, Color32, RichText, TextEdit, Button, ScrollArea};
use crate::config::AppSettings;
use crate::api::call_llm;
pub enum OutputStatus {
Waiting,
Connecting,
Completed,
Error(String),
Stopped,
}
pub struct MainPage {
pub input_text: String,
pub output_text: String,
pub status: OutputStatus,
pub selected_prompt_index: usize,
pub is_loading: bool,
}
impl Default for MainPage {
fn default() -> Self {
Self {
input_text: String::new(),
output_text: String::new(),
status: OutputStatus::Waiting,
selected_prompt_index: 0,
is_loading: false,
}
}
}
impl MainPage {
pub fn new() -> Self {
Self::default()
}
pub fn ui(&mut self, ui: &mut Ui, settings: &AppSettings, ctx: &egui::Context) {
ScrollArea::vertical().show(ui, |ui| {
self.render_content(ui, settings, ctx);
});
}
fn render_content(&mut self, ui: &mut Ui, settings: &AppSettings, ctx: &egui::Context) {
ui.add_space(16.0);
ui.label(RichText::new("快速操作").size(11.0).color(Color32::GRAY));
ui.add_space(6.0);
ui.horizontal(|ui| {
let buttons = vec![
("🔍", "检查错别字"),
("📋", "总结"),
("🌐", "翻译"),
("✨", "润色"),
];
for (emoji, name) in buttons {
let btn = Button::new(RichText::new(emoji).size(16.0))
.fill(Color32::from_rgb(245, 245, 247))
.stroke(egui::Stroke::new(1.0, Color32::from_rgb(210, 210, 220)))
.rounding(6.0)
.min_size(egui::vec2(36.0, 36.0));
if ui.add(btn).on_hover_text(name).clicked() {
self.select_prompt_by_name(name, settings, ctx);
}
}
});
ui.add_space(14.0);
ui.label(RichText::new("提示词").size(11.0).color(Color32::GRAY));
ui.add_space(6.0);
let selected_text = if self.selected_prompt_index == 0 {
"无系统提示词".to_string()
} else if self.selected_prompt_index <= settings.prompt_configs.len() {
settings.prompt_configs[self.selected_prompt_index - 1].title.clone()
} else {
"无系统提示词".to_string()
};
egui::ComboBox::from_id_salt("prompt_selector")
.selected_text(&selected_text)
.show_ui(ui, |ui| {
ui.selectable_value(&mut self.selected_prompt_index, 0, "无系统提示词");
for (i, prompt) in settings.prompt_configs.iter().enumerate() {
ui.selectable_value(&mut self.selected_prompt_index, i + 1, &prompt.title);
}
});
ui.separator();
let prompt_content = if self.selected_prompt_index == 0 {
"无特殊指令".to_string()
} else if self.selected_prompt_index <= settings.prompt_configs.len() {
settings.prompt_configs[self.selected_prompt_index - 1].content.clone()
} else {
"无特殊指令".to_string()
};
ui.label(RichText::new(&prompt_content).size(11.0).color(Color32::GRAY));
ui.add_space(14.0);
ui.label(RichText::new("优化结果").size(11.0).color(Color32::GRAY));
ui.add_space(6.0);
let status_text = match &self.status {
OutputStatus::Waiting => "等待发送".to_string(),
OutputStatus::Connecting => "连接中…".to_string(),
OutputStatus::Completed => "已完成".to_string(),
OutputStatus::Error(_) => "发生错误".to_string(),
OutputStatus::Stopped => "已停止".to_string(),
};
let status_color = match &self.status {
OutputStatus::Waiting => Color32::from_rgb(100, 100, 255),
OutputStatus::Connecting => Color32::from_rgb(255, 165, 0),
OutputStatus::Completed => Color32::from_rgb(0, 180, 0),
OutputStatus::Error(_) => Color32::RED,
OutputStatus::Stopped => Color32::GRAY,
};
ui.label(RichText::new(&status_text).size(11.0).color(status_color));
let card_frame = egui::Frame::none()
.fill(ui.style().visuals.widgets.inactive.bg_fill)
.stroke(egui::Stroke::new(1.0, Color32::from_rgb(200, 200, 210)))
.inner_margin(egui::Margin::same(12.0))
.rounding(6.0);
card_frame.show(ui, |ui: &mut Ui| {
ui.add_sized([ui.available_width(), 2.0], egui::Separator::default());
ui.add_space(8.0);
if self.output_text.is_empty() {
ui.label(RichText::new("发送消息后结果将在此显示").size(13.0).color(Color32::GRAY));
} else {
ui.add_sized(
[ui.available_width(), 150.0],
TextEdit::multiline(&mut self.output_text.clone())
.desired_width(f32::INFINITY)
.desired_rows(6),
);
}
ui.add_space(8.0);
ui.horizontal(|ui: &mut Ui| {
let copy_btn = Button::new("复制结果")
.fill(Color32::TRANSPARENT)
.stroke(egui::Stroke::new(1.0, Color32::from_rgb(100, 100, 255)))
.rounding(4.0);
if ui.add(copy_btn).clicked() && !self.output_text.is_empty() {
self.copy_to_clipboard();
}
});
});
ui.add_space(16.0);
}
pub fn send_request(&mut self, settings: &AppSettings, ctx: egui::Context) {
if self.input_text.trim().is_empty() || self.is_loading {
return;
}
self.status = OutputStatus::Connecting;
self.output_text = "正在生成...".to_string();
self.is_loading = true;
let prompt = if self.selected_prompt_index == 0 {
None
} else if self.selected_prompt_index <= settings.prompt_configs.len() {
Some(settings.prompt_configs[self.selected_prompt_index - 1].content.clone())
} else {
None
};
let input = self.input_text.clone();
let settings = settings.clone();
let handle = std::thread::spawn(move || {
let result = call_llm(&settings, input, prompt);
ctx.request_repaint();
result
});
std::thread::spawn(move || {
let _ = handle.join();
});
}
fn select_prompt_by_name(&mut self, name: &str, settings: &AppSettings, ctx: &egui::Context) {
let mut found = false;
for (i, prompt) in settings.prompt_configs.iter().enumerate() {
if prompt.title == name {
self.selected_prompt_index = i + 1;
found = true;
break;
}
}
if !self.input_text.is_empty() && found {
self.send_request(settings, ctx.clone());
}
}
fn copy_to_clipboard(&self) {
if let Ok(mut clipboard) = arboard::Clipboard::new() {
let _ = clipboard.set_text(&self.output_text);
}
}
}