Update app.py

app.py

@@ -150,9 +150,10 @@ examples = [
 ]
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Qwen-Image-Edit [FAST] with HF Inference Prompt Enhancement")
+    gr.Markdown("# Qwen-Image-Edit [FAST] with HF Prompt Enhancement")
     gr.Markdown("✨ **8-step lightning inferencing with lightx2v's LoRA.**")
     gr.Markdown("⚠️ **Prompt rewriting requires your own [Hugging Face token](https://huggingface.co/settings/tokens)**")
+    gr.Markdown("🚧 **Work in progress, further improvements coming soon.**")
 
     with gr.Row():
         with gr.Column():
@@ -190,29 +191,31 @@ with gr.Blocks() as demo:
                         step=1,
                         value=1
                     )
-                with gr.Group():
-                    rewrite_toggle = gr.Checkbox(label="Use Prompt Rewriter (Requires HF Token)", value=False, interactive=True)
-                    hf_token_input = gr.Textbox(
-                        label="Your Hugging Face Token",
-                        type="password",
-                        placeholder="hf_xxxxxxxxxxxxxxxx",
-                        visible=False,
-                        info="Required for prompt rewriting - get yours from [Hugging Face settings](https://huggingface.co/settings/tokens)"
-                    )
-                    
-                    def toggle_token_visibility(checked):
-                        return gr.update(visible=checked)
-                    
-                    rewrite_toggle.change(
-                        toggle_token_visibility,
-                        inputs=[rewrite_toggle],
-                        outputs=[hf_token_input]
-                    )
+
 
             run_button = gr.Button("Edit", variant="primary")
             
         with gr.Column():
-            result = gr.Gallery(label="Output Images", show_label=False)
+            result = gr.Gallery(label="Output Images", show_label=False, columns=1)
+            
+            with gr.Group():
+                rewrite_toggle = gr.Checkbox(label="Use Prompt Rewriter (Requires HF Token)", value=False, interactive=True)
+                hf_token_input = gr.Textbox(
+                    label="Your Hugging Face Token",
+                    type="password",
+                    placeholder="hf_xxxxxxxxxxxxxxxx",
+                    visible=False,
+                    info="Required for prompt rewriting - get yours from [Hugging Face settings](https://huggingface.co/settings/tokens). API tokens are kept safe locally, but as good practice please make sure to double check the source code. Invalid or missing keys will revert to the original prompt entered."
+                )
+                
+                def toggle_token_visibility(checked):
+                    return gr.update(visible=checked)
+                
+                rewrite_toggle.change(
+                    toggle_token_visibility,
+                    inputs=[rewrite_toggle],
+                    outputs=[hf_token_input]
+                )