From b06ebc45c2c21454448439ca12701ca4995e9474 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 24 Apr 2024 09:12:14 +0000
Subject: refactor: adjust order of execute actions

---
 src/main.rs | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index b396917..f3d3846 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -196,7 +196,7 @@ async fn execute(config: &GlobalConfig, mut input: Input) -> Result<()> {
         loop {
             let answer = Select::new(
                 markdown_render.render(&eval_str).trim(),
-                vec!["✅ Execute", "📙 Explain", "🤔 Revise", "❌ Cancel"],
+                vec!["✅ Execute", "🤔 Revise", "📙 Explain", "❌ Cancel"],
             )
             .prompt()?;
 
@@ -207,6 +207,14 @@ async fn execute(config: &GlobalConfig, mut input: Input) -> Result<()> {
                         process::exit(code);
                     }
                 }
+                "🤔 Revise" => {
+                    let revision = Text::new("Enter your revision:").prompt()?;
+                    let text = input.text();
+                    let text =
+                        format!("[INST] {text} [/INST]\n{eval_str}\n[INST] {revision} [/INST]\n");
+                    input.set_text(text);
+                    return execute(config, input).await;
+                }
                 "📙 Explain" => {
                     if !explain {
                         config.write().set_role(EXPLAIN_ROLE)?;
@@ -217,15 +225,6 @@ async fn execute(config: &GlobalConfig, mut input: Input) -> Result<()> {
                     explain = true;
                     continue;
                 }
-                "🤔 Revise" => {
-                    let revision = Text::new("Enter your revision:").prompt()?;
-                    let text = format!(
-                        "[INST] {} [/INST]\n{eval_str}\n[INST] {revision} [/INST]\n",
-                        input.text()
-                    );
-                    input.set_text(text);
-                    return execute(config, input).await;
-                }
                 _ => {}
             }
             break;
-- 
cgit v1.2.3