Update app.py

app.py

@@ -148,9 +148,10 @@ def infer(input_image, prompt, illumination_dropdown, direction_dropdown, seed=4
         progress (gr.Progress): A Gradio progress tracker for the UI.
     
     Returns:
-        tuple[PIL.Image.Image, int]: A tuple containing:
+        tuple[PIL.Image.Image, int, str]: A tuple containing:
             - The relighted output image.
             - The seed used for the generation.
+            - The final constructed prompt string used by the model.
     
     ------------------------------------------------------------------------------------
     Dropdown Options
@@ -294,7 +295,7 @@ def infer(input_image, prompt, illumination_dropdown, direction_dropdown, seed=4
         height=input_image.size[1],
         generator=torch.Generator().manual_seed(seed),
     ).images[0]
-    return image, seed
+    return image, seed, prompt_with_template
 
 def update_prompt_from_dropdown(illumination_option):
     """Update the prompt textbox based on dropdown selection"""
@@ -372,7 +373,7 @@ with gr.Blocks(css=css) as demo:
                     
             with gr.Column():
                 result = gr.Image(label="Result", show_label=False, interactive=False)
-        
+                final_prompt = gr.Textbox(label="Processed prompt")
         # update prompt when dropdown changes
         illumination_dropdown.change(
             fn=update_prompt_from_dropdown,
@@ -402,7 +403,7 @@ with gr.Blocks(css=css) as demo:
             triggers=[run_button.click, prompt.submit],
             fn = infer,
             inputs = [input_image, prompt, illumination_dropdown, direction_dropdown, seed, randomize_seed, guidance_scale],
-            outputs = [result, seed]
+            outputs = [result, seed, final_prompt]
         )