From d9ac789235499840b8965cbe5efe1d26406f5d9a Mon Sep 17 00:00:00 2001
From: Arash Kadkhodaei <arash77.kad@gmail.com>
Date: Mon, 2 Dec 2024 21:18:30 +0100
Subject: [PATCH] Improve FLUX tool for output label and prompt display (#1551)

* Improve output label based on input type and print the prompt itself

* Update version suffix in flux from 3 to 4
---
 tools/flux/flux.py  | 1 +
 tools/flux/flux.xml | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/tools/flux/flux.py b/tools/flux/flux.py
index 4de9f03bd2..0b3e70bb18 100644
--- a/tools/flux/flux.py
+++ b/tools/flux/flux.py
@@ -12,6 +12,7 @@
         prompt = f.read().strip()
 elif prompt_type == "text":
     prompt = sys.argv[3]
+print(f"Creating image from prompt: {prompt}")
 
 if "dev" in model_path:
     num_inference_steps = 20
diff --git a/tools/flux/flux.xml b/tools/flux/flux.xml
index 7c0dc6006b..2361506de3 100644
--- a/tools/flux/flux.xml
+++ b/tools/flux/flux.xml
@@ -2,7 +2,7 @@
     <description>text-to-image model</description>
     <macros>
         <token name="@TOOL_VERSION@">2024</token>
-        <token name="@VERSION_SUFFIX@">3</token>
+        <token name="@VERSION_SUFFIX@">4</token>
     </macros>
     <requirements>
         <container type="docker">quay.io/galaxy/black_forest_labs_flux:2024-0</container>
@@ -39,7 +39,7 @@ python3 '$__tool_directory__/flux.py'
         </conditional>
     </inputs>
     <outputs>
-        <data name="output" format="png" label="${tool.name} on ${on_string}" from_work_dir="./output.png"/>
+        <data name="output" format="png" label="${tool.name} on ${on_string if $input_type_selector=='file' else 'text prompt'}" from_work_dir="./output.png"/>
     </outputs>
     <tests>
         <test expect_exit_code="1" expect_failure="true">