19 lines
465 B
Rust
19 lines
465 B
Rust
|
use crate::controller::ControllerType;
|
||
|
|
mod tests;
|
||
|
|
|
||
|
|
pub fn determine_controller(
|
||
|
|
text: &str,
|
||
|
|
) -> ControllerType {
|
||
|
|
let text = text.trim();
|
||
|
|
|
||
|
|
if let Some(prompt) = text.strip_prefix(&format!("create")) {
|
||
|
|
return ControllerType::ImageGeneration(prompt.trim().to_owned());
|
||
|
|
}
|
||
|
|
|
||
|
|
if let Some(prompt) = text.strip_prefix(&format!("edit")) {
|
||
|
|
return ControllerType::ImageEdit(prompt.trim().to_owned());
|
||
|
|
}
|
||
|
|
|
||
|
|
ControllerType::UsageHelp
|
||
|
|
}
|