feat: support negative_prompt in image_edit_handler

Signed-off-by: Xin Liu <sam@secondstate.io>
LlamaEdge · Oct 11, 2024 · 9335c7d · 9335c7d
1 parent 81912da
commit 9335c7d
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/src/backend/sd.rs b/src/backend/sd.rs
@@ -296,6 +296,31 @@ pub(crate) async fn image_edit_handler(req: Request<Body>) -> Response<Body> {
                             return error::internal_server_error(err_msg);
                         }
                     },
+                    "negative_prompt" => match field.is_text() {
+                        true => {
+                            let mut negative_prompt = String::new();
+
+                            if let Err(e) = field.data.read_to_string(&mut negative_prompt) {
+                                let err_msg = format!("Failed to read the prompt. {}", e);
+
+                                // log
+                                error!(target: "stdout", "{}", &err_msg);
+
+                                return error::internal_server_error(err_msg);
+                            }
+
+                            image_request.prompt = negative_prompt;
+                        }
+                        false => {
+                            let err_msg =
+                                "Failed to get the prompt. The prompt field in the request should be a text field.";
+
+                            // log
+                            error!(target: "stdout", "{}", &err_msg);
+
+                            return error::internal_server_error(err_msg);
+                        }
+                    },
                     "mask" => {
                         let filename = match field.headers.filename {
                             Some(filename) => filename,