add : display refinement prompt

app.py

@@ -12,6 +12,11 @@ from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 import subprocess
 
+from pydantic import BaseModel
+
+class RefinedPrompt(BaseModel):
+    prompt:str
+
 subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 
 
@@ -56,7 +61,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             output_type="pil",
             good_vae=good_vae,
         ):
-            yield img, seed
+            yield img, seed, refined_prompt
     
 examples = [
     "기계공학과(로켓) 포닉스",
@@ -91,6 +96,7 @@ with gr.Blocks(css=css) as demo:
             run_button = gr.Button("Run", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
+        refined_prompt_display = gr.Text(label="정제된 프롬프트", show_label=True)
         
         with gr.Accordion("Advanced Settings", open=False):
             
@@ -144,7 +150,7 @@ with gr.Blocks(css=css) as demo:
             examples = examples,
             fn = infer,
             inputs = [prompt],
-            outputs = [result, seed],
+            outputs = [result, seed, refined_prompt_display],
             cache_examples="lazy"
         )
 
@@ -152,7 +158,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn = infer,
         inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs = [result, seed]
+        outputs = [result, seed, refined_prompt_display]
     )
 
 demo.launch()