{if let Some(mode) = &props.message.mode {
@@ -60,14 +60,12 @@ pub fn message_bubble(props: &MessageBubbleProps) -> Html {
href={url}
target="_blank"
rel="noopener noreferrer"
- class="inline-flex items-center gap-1 text-xs px-2 py-1 rounded-full \
- bg-vect-violet/10 text-vect-violet-light border border-vect-violet/20 \
- hover:bg-vect-violet/20 hover:border-vect-violet/40 \
+ class="inline-flex items-center gap-1 text-xs px-2 py-1 rounded-md \
+ bg-vect-elevated text-vect-muted border border-vect-border \
+ hover:bg-vect-border hover:text-vect-text \
transition-all duration-150 max-w-[180px] truncate"
>
-
+
{title}
}
diff --git a/examples/chat/src/components/mode_badge.rs b/examples/chat/src/components/mode_badge.rs
index e235c07..3aa8dc8 100644
--- a/examples/chat/src/components/mode_badge.rs
+++ b/examples/chat/src/components/mode_badge.rs
@@ -10,7 +10,7 @@ pub struct ModeBadgeProps {
pub fn mode_badge(props: &ModeBadgeProps) -> Html {
html! {
-
+
{" "}{props.mode.label()}
}
diff --git a/examples/chat/src/components/sidebar.rs b/examples/chat/src/components/sidebar.rs
index 7c9bc27..5260068 100644
--- a/examples/chat/src/components/sidebar.rs
+++ b/examples/chat/src/components/sidebar.rs
@@ -76,7 +76,7 @@ pub fn sidebar(props: &SidebarProps) -> Html {
id={format!("mode-{}", m.label().to_lowercase())}
title={m.description()}
>
-
+
{m.label()}
{m.description()}
@@ -106,7 +106,7 @@ pub fn sidebar(props: &SidebarProps) -> Html {
class={classes!(
"toggle-track",
"shrink-0",
- if props.stochastic { "bg-vect-violet" } else { "bg-vect-border" }
+ if props.stochastic { "bg-vect-accent" } else { "bg-vect-border" }
)}
>
Html {
if props.web_search {
-
diff --git a/examples/chat/src/lmm_bridge.rs b/examples/chat/src/lmm_bridge.rs
index 5b4f388..70628bd 100644
--- a/examples/chat/src/lmm_bridge.rs
+++ b/examples/chat/src/lmm_bridge.rs
@@ -1,7 +1,5 @@
use crate::types::SearchLink;
-use lmm::net::{
- SearchAggregator, corpus_from_response, corpus_from_results, corpus_from_results_raw,
-};
+use lmm::net::{SearchAggregator, corpus_from_response};
use lmm::predict::TextPredictor;
use lmm::stochastic::StochasticEnhancer;
use lmm::text::{EssayGenerator, ParagraphGenerator, SentenceGenerator, TextSummarizer};
@@ -14,6 +12,20 @@ fn enhance(text: &str, stochastic: bool, prob: f64) -> String {
}
}
+fn local_fallback(prompt: &str, sentences: usize, stochastic: bool, prob: f64) -> String {
+ match ParagraphGenerator::new(sentences.max(2), 50, 5).generate(prompt) {
+ Ok(p) => enhance(&p, stochastic, prob),
+ Err(_) => match SentenceGenerator::new(50, 5).generate(prompt) {
+ Ok(s) => enhance(&s, stochastic, prob),
+ Err(_) => format!(
+ "I could not retrieve web results for \"{prompt}\" \
+ and the local model lacks enough context to generate a meaningful response. \
+ Please try a more descriptive query."
+ ),
+ },
+ }
+}
+
pub fn generate_sentence(seed: &str, stochastic: bool, prob: f64) -> String {
match SentenceGenerator::new(50, 5).generate(seed) {
Ok(s) => enhance(&s, stochastic, prob),
@@ -77,7 +89,9 @@ pub async fn ask(
stochastic: bool,
prob: f64,
) -> (String, Vec) {
+ let prompt = prompt.trim();
let aggregator = SearchAggregator::new();
+
let api_result = aggregator.get_response(prompt).await;
let corpus = api_result
@@ -85,7 +99,7 @@ pub async fn ask(
.map(corpus_from_response)
.unwrap_or_default();
- let mut links: Vec = api_result
+ let links: Vec = api_result
.as_ref()
.map(|resp| {
resp.related_topics
@@ -116,19 +130,9 @@ pub async fn ask(
let final_corpus = if corpus.trim().is_empty() {
#[cfg(not(target_arch = "wasm32"))]
{
+ use lmm::net::{corpus_from_results, corpus_from_results_raw};
match aggregator.fetch(prompt, 6).await {
Ok(results) => {
- if links.is_empty() {
- links = results
- .iter()
- .filter(|r| !r.url.is_empty())
- .map(|r| SearchLink {
- title: r.title.clone(),
- url: r.url.clone(),
- })
- .take(10)
- .collect();
- }
let quality = corpus_from_results(&results);
if quality.trim().is_empty() {
corpus_from_results_raw(&results)
@@ -148,12 +152,7 @@ pub async fn ask(
};
if final_corpus.trim().is_empty() {
- return (
- "The DuckDuckGo Instant Answer API returned no content for this query. \
- Try a more specific well-known topic."
- .to_string(),
- links,
- );
+ return (local_fallback(prompt, sentences, stochastic, prob), links);
}
let summary = match TextSummarizer::new(sentences.max(2), 50, 5)
@@ -164,7 +163,7 @@ pub async fn ask(
.map(|s| enhance(s, stochastic, prob))
.collect::>()
.join(" "),
- Err(_) => "Could not summarize search results.".to_string(),
+ Err(_) => local_fallback(prompt, sentences, stochastic, prob),
};
(summary, links)
diff --git a/examples/chat/src/tailwind.css b/examples/chat/src/tailwind.css
index f108ce7..315f098 100644
--- a/examples/chat/src/tailwind.css
+++ b/examples/chat/src/tailwind.css
@@ -29,7 +29,7 @@
}
::-webkit-scrollbar-track {
- @apply bg-vect-surface;
+ @apply bg-transparent;
}
::-webkit-scrollbar-thumb {
@@ -38,11 +38,11 @@
}
::-webkit-scrollbar-thumb:hover {
- @apply bg-vect-violet;
+ @apply bg-vect-subtle;
}
::selection {
- @apply bg-vect-violet text-white;
+ @apply bg-vect-accent text-white;
}
}
@@ -52,41 +52,41 @@
}
.vect-btn-primary {
- @apply inline-flex items-center justify-center gap-2 bg-vect-violet hover:bg-vect-violet-hover
- text-white font-semibold px-4 py-2 rounded-xl transition-all duration-200
- focus:outline-none focus:ring-2 focus:ring-vect-violet focus:ring-offset-2 focus:ring-offset-vect-bg
- active:scale-95 disabled:opacity-50 disabled:cursor-not-allowed;
+ @apply inline-flex items-center justify-center gap-2 bg-vect-accent hover:bg-vect-accent-hover
+ text-white font-medium px-4 py-2 rounded-full transition-all duration-200
+ focus:outline-none focus:ring-2 focus:ring-vect-accent focus:ring-offset-2 focus:ring-offset-vect-bg
+ active:scale-95 disabled:opacity-40 disabled:cursor-not-allowed;
}
.vect-btn-ghost {
@apply inline-flex items-center justify-center gap-2 bg-transparent hover:bg-vect-elevated
- text-vect-muted hover:text-vect-text font-medium px-3 py-2 rounded-xl transition-all duration-200
- focus:outline-none focus:ring-2 focus:ring-vect-violet;
+ text-vect-muted hover:text-vect-text font-medium px-3 py-2 rounded-lg transition-all duration-200
+ focus:outline-none focus:ring-2 focus:ring-vect-accent;
}
.vect-input {
- @apply bg-vect-elevated border border-vect-border text-vect-text rounded-xl px-4 py-3
- focus:outline-none focus:ring-2 focus:ring-vect-violet focus:border-vect-violet
+ @apply bg-vect-elevated border border-vect-border text-vect-text rounded-lg px-4 py-3
+ focus:outline-none focus:ring-2 focus:ring-vect-accent focus:border-vect-accent
transition-all duration-200 placeholder-vect-subtle w-full;
}
.vect-textarea {
@apply bg-vect-elevated border border-vect-border text-vect-text rounded-xl px-4 py-3
- focus:outline-none focus:ring-2 focus:ring-vect-violet focus:border-vect-violet
+ focus:outline-none focus:ring-2 focus:ring-vect-accent focus:border-vect-accent
transition-all duration-200 placeholder-vect-subtle w-full resize-none;
- min-height: 80px;
+ min-height: 52px;
max-height: 200px;
overflow-y: auto;
field-sizing: content;
}
.mode-btn {
- @apply flex items-center gap-2.5 w-full text-left px-3 py-2.5 rounded-xl text-sm font-medium
+ @apply flex items-center gap-2.5 w-full text-left px-3 py-2.5 rounded-lg text-sm font-medium
transition-all duration-150 cursor-pointer select-none;
}
.mode-btn-active {
- @apply bg-vect-violet text-white shadow-glow-violet;
+ @apply bg-vect-elevated text-vect-text;
}
.mode-btn-inactive {
@@ -102,17 +102,17 @@
}
.message-user {
- @apply self-end max-w-[85%] md:max-w-[70%] bg-gradient-to-br from-violet-700 to-vect-violet
- text-white rounded-2xl rounded-br-sm px-5 py-3 shadow-glow-violet animate-fade-in;
+ @apply self-end max-w-[85%] md:max-w-[70%] bg-vect-elevated
+ text-vect-text rounded-2xl rounded-br-sm px-5 py-3 animate-fade-in;
}
.message-ai {
- @apply self-start max-w-[85%] md:max-w-[70%] bg-vect-elevated border border-vect-border
+ @apply self-start max-w-[85%] md:max-w-[70%]
text-vect-text rounded-2xl rounded-bl-sm px-5 py-3 animate-fade-in;
}
.badge {
- @apply inline-flex items-center gap-1 text-xs font-semibold px-2 py-0.5 rounded-full;
+ @apply inline-flex items-center gap-1.5 text-xs font-medium px-2.5 py-1 rounded-md;
}
.sidebar-label {
diff --git a/examples/chat/src/types.rs b/examples/chat/src/types.rs
index bdd1d9c..1c7fd46 100644
--- a/examples/chat/src/types.rs
+++ b/examples/chat/src/types.rs
@@ -20,14 +20,14 @@ impl GenerationMode {
}
}
- pub fn icon(&self) -> &'static str {
+ pub fn icon_class(&self) -> &'static str {
match self {
- Self::Sentence => "✍️",
- Self::Paragraph => "📄",
- Self::Essay => "📖",
- Self::Summarize => "✂️",
- Self::Predict => "🔮",
- Self::Ask => "🌐",
+ Self::Sentence => "fa-solid fa-pen-nib",
+ Self::Paragraph => "fa-solid fa-align-left",
+ Self::Essay => "fa-solid fa-book-open",
+ Self::Summarize => "fa-solid fa-scissors",
+ Self::Predict => "fa-solid fa-wand-magic-sparkles",
+ Self::Ask => "fa-solid fa-globe",
}
}
@@ -44,12 +44,12 @@ impl GenerationMode {
pub fn color_class(&self) -> &'static str {
match self {
- Self::Sentence => "bg-vect-violet text-white",
- Self::Paragraph => "bg-violet-700 text-white",
- Self::Essay => "bg-violet-600 text-white",
- Self::Summarize => "bg-vect-cyan text-vect-bg",
- Self::Predict => "bg-indigo-600 text-white",
- Self::Ask => "bg-teal-600 text-white",
+ Self::Sentence => "bg-vect-elevated text-vect-text",
+ Self::Paragraph => "bg-vect-elevated text-vect-text",
+ Self::Essay => "bg-vect-elevated text-vect-text",
+ Self::Summarize => "bg-vect-elevated text-vect-text",
+ Self::Predict => "bg-vect-elevated text-vect-text",
+ Self::Ask => "bg-vect-accent/20 text-vect-accent",
}
}
diff --git a/examples/chat/tailwind.config.js b/examples/chat/tailwind.config.js
index 4b02518..3d612f4 100644
--- a/examples/chat/tailwind.config.js
+++ b/examples/chat/tailwind.config.js
@@ -5,19 +5,15 @@ module.exports = {
extend: {
colors: {
vect: {
- bg: "#07071a",
- surface: "#0d0d24",
- elevated: "#12122e",
- border: "#1e1e4a",
- violet: "#7c3aed",
- "violet-hover": "#6d28d9",
- "violet-light": "#a78bfa",
- cyan: "#06b6d4",
- "cyan-hover": "#0891b2",
- "cyan-light": "#67e8f9",
- text: "#e2e8f0",
- muted: "#94a3b8",
- subtle: "#475569",
+ bg: "#212121",
+ surface: "#171717",
+ elevated: "#2f2f2f",
+ border: "#424242",
+ accent: "#10a37f",
+ "accent-hover": "#0d8a6a",
+ text: "#ececec",
+ muted: "#b4b4b4",
+ subtle: "#6e6e6e",
},
},
fontFamily: {
@@ -25,27 +21,18 @@ module.exports = {
mono: ["JetBrains Mono", "monospace"],
},
animation: {
- "fade-in": "fadeIn 0.35s ease-out",
- "slide-up": "slideUp 0.3s ease-out",
- "pulse-glow": "pulseGlow 2s ease-in-out infinite",
+ "fade-in": "fadeIn 0.3s ease-out",
+ "slide-up": "slideUp 0.25s ease-out",
},
keyframes: {
fadeIn: {
- "0%": { opacity: "0", transform: "translateY(6px)" },
+ "0%": { opacity: "0", transform: "translateY(4px)" },
"100%": { opacity: "1", transform: "translateY(0)" },
},
slideUp: {
- "0%": { opacity: "0", transform: "translateY(10px)" },
+ "0%": { opacity: "0", transform: "translateY(8px)" },
"100%": { opacity: "1", transform: "translateY(0)" },
},
- pulseGlow: {
- "0%, 100%": { boxShadow: "0 0 0 0 rgba(124,58,237,0)" },
- "50%": { boxShadow: "0 0 16px 4px rgba(124,58,237,0.35)" },
- },
- },
- boxShadow: {
- "glow-violet": "0 0 24px rgba(124,58,237,0.4)",
- "glow-cyan": "0 0 24px rgba(6,182,212,0.35)",
},
},
},
diff --git a/lmm/src/imagen.rs b/lmm/src/imagen.rs
index c792ec7..adfac42 100644
--- a/lmm/src/imagen.rs
+++ b/lmm/src/imagen.rs
@@ -35,7 +35,7 @@
//! println!("Saved to {path}");
//! ```
-#![cfg_attr(target_arch = "wasm32", allow(dead_code))]
+#![cfg_attr(target_arch = "wasm32", allow(unused_imports), allow(dead_code))]
use crate::error::{LmmError, Result};
use phf::{Map, phf_map};
use strum_macros::{AsRefStr, Display, EnumIter, EnumString};
-
+
{format!("{}%", props.probability)}
@@ -129,7 +129,7 @@ pub fn sidebar(props: &SidebarProps) -> Html {
min="1"
max="100"
value={props.probability.to_string()}
- class="w-full h-1.5 rounded-full cursor-pointer accent-vect-violet bg-vect-border"
+ class="w-full h-1.5 rounded-full cursor-pointer accent-vect-accent bg-vect-border"
oninput={on_prob}
aria-label="Synonym replacement probability"
aria-valuemin="1"
@@ -167,7 +167,7 @@ pub fn sidebar(props: &SidebarProps) -> Html {
class={classes!(
"toggle-track",
"shrink-0",
- if props.web_search { "bg-vect-cyan" } else { "bg-vect-border" }
+ if props.web_search { "bg-vect-accent" } else { "bg-vect-border" }
)}
>
Html {
- {"🌐 DuckDuckGo search fetches live context for your query."} +
+ + {"DuckDuckGo search fetches live context for your query."}
} else {@@ -251,7 +252,7 @@ pub fn sidebar(props: &SidebarProps) -> Html {
{"Powered by "} - {"LMM"} + {"LMM"} {" - Large Mathematical Model. All text generation runs locally in your browser via WebAssembly."}