Update UI and new properties

app.py

@@ -351,6 +351,7 @@ with gr.Blocks() as demo:
             gr.Markdown("## CV Agent Arena  🎨🤖")
             with gr.Row():
                 with gr.Column():
+                    task_dropdown = gr.Dropdown(choices=TASK_CHOICES, label="Task Category", type="index")
                     original = gr.Image(type="pil", label="Upload Original Image")
                     prompt   = gr.Textbox(lines=2, label="Prompt",
                                          placeholder="e.g. Make it look like a sunny day")
@@ -364,7 +365,7 @@ with gr.Blocks() as demo:
 
             run_btn.click(
                 fn=generate_and_store,
-                inputs=[original, prompt, agent1, agent2],
+                inputs=[task_dropdown, original, prompt, agent1, agent2],
                 outputs=[out1, out2],
                 show_api=False
             )
@@ -394,7 +395,9 @@ with gr.Blocks() as demo:
             )
 
             with gr.Row():
-                gr.Markdown("## Please Evaluate the Original Image from 3 Aspects").style(text_align="center")
+                gr.Markdown(
+                    "<h2 style='text-align: center;'>Please Evaluate the Original Image from 3 Aspects</h2>"
+                )
                 with gr.Column():
                     a1_follow     = gr.Radio([0,1,2,3,4,5], label="Follow Prompt")
                     a1_creativity = gr.Radio([0,1,2,3,4,5], label="Creativity")