22

app.py

@@ -11,6 +11,7 @@ from videox_fun.models import ZImageControlTransformer2DModel
 from transformers import AutoTokenizer, Qwen3ForCausalLM
 from diffusers import AutoencoderKL
 from utils.image_utils import get_image_latent, scale_image
+from utils.prompt_utils import polish_prompt
 # from videox_fun.utils.utils import get_image_latent
 
 
@@ -137,7 +138,7 @@ with open('static/data.json', 'r') as file:
     data = json.load(file)
 examples = data['examples']
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
         with gr.Column():
             gr.HTML(read_file("static/header.html"))
@@ -159,7 +160,10 @@ with gr.Blocks(css=css) as demo:
 
                 run_button = gr.Button("Run", variant="primary")
             with gr.Column():
-                output_image = gr.Image(label="Result", show_label=False)
+                output_image = gr.Image(label="Generated image", show_label=False)
+                polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
+                with gr.Accordion("Control image", open=False):
+                    control_image = gr.Image(label="Control image", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
                 seed = gr.Slider(
@@ -226,4 +230,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(mcp_server=True, css=css)