Update app.py

app.py

@@ -11,6 +11,9 @@ from diffusers import QwenImageEditPipeline, FlowMatchEulerDiscreteScheduler
 from huggingface_hub import InferenceClient
 import math
 
+import base64
+from io import BytesIO
+
 # --- Prompt Enhancement using Hugging Face InferenceClient ---
 def polish_prompt_hf(original_prompt, system_prompt, img):
     """
@@ -417,7 +420,7 @@ def infer(
         ).images
         
         # Return the first (and only) image
-        return images[0], seed
+        return [image,images[0]], seed, final_prompt
         
     except Exception as e:
         print(f"Error during inference: {e}")
@@ -488,17 +491,18 @@ with gr.Blocks(css=css) as demo:
                         scale=1
                     )
             # Changed from Gallery to Image
-            result = gr.Image(
+            result = gr.ImageSlider(
                 label="Result", 
                 show_label=True, 
                 type="pil"
             )
+            final_prompt = gr.Textbox(label="Processed prompt")
             
         with gr.Row():
             prompt = gr.Text(
                 label="Edit Instruction",
                 show_label=False,
-                placeholder="Describe the edit instruction (e.g., 'Replace the background with a sunset', 'Add a red hat', 'Remove the person')",
+                placeholder="describe the edit instruction",
                 container=False,
             )
             run_button = gr.Button("Edit!", variant="primary")
@@ -551,7 +555,7 @@ with gr.Blocks(css=css) as demo:
                     ["./assets/pexels-pixabay-355465.jpg", "daylight, bright sunshine", "custom", "auto" ]           
                 ],
                 inputs=[input_image, prompt, illumination_dropdown, direction_dropdown],
-                outputs=[result, seed],
+                outputs=[result, seed, final_prompt],
                 fn=infer,
                 cache_examples="lazy"
             )
@@ -576,7 +580,7 @@ with gr.Blocks(css=css) as demo:
             rewrite_prompt,
             # Removed num_images_per_prompt from inputs
         ],
-        outputs=[result, seed],
+        outputs=[result, seed, final_prompt],
     )
 
 if __name__ == "__main__":