output prompt

app.py

@@ -138,7 +138,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
 
         torch.cuda.empty_cache()  # Clean up GPU memory after generation
         progress(1.0, desc="Done!")
-        return image, seed
+        return prompt, image, seed
     except Exception as e:
         return None, f"Error generating image: {str(e)}"
  
@@ -332,7 +332,7 @@ def create_interface():
             triggers=[run_button.click],
             fn = infer,
             inputs = [refined_prompt, seed, randomize_seed, width, height, num_inference_steps],
-            outputs = [generated_image, seed]
+            outputs = [prompt, generated_image, seed]
         )
 
     return demo