Update app.py

app.py

@@ -73,15 +73,16 @@ def run_diva_vqa(video_path, is_finetune, train_data_name, test_data_name, netwo
 
 def toggle_finetune_visibility(train_dataset):
     """
-    when using cross experiment & train dataset is lsvq_train
+    when using train dataset is lsvq_train
     """
     return gr.update(visible=(train_dataset == "lsvq_train"))
 
-def toggle_test_dataset(is_finetune, train_dataset):
-    """
-    show test dataset dropdown when finetuning is enabled
-    """
-    return gr.update(visible=(is_finetune and train_dataset == "lsvq_train"))
+def update_test_dataset(is_finetune, train_dataset, current_test_value):
+    if train_dataset != "lsvq_train":
+        msg = f"Intra-dataset experiment — test dataset is automatically set to **{train_dataset}**."
+        return gr.update(value=train_dataset, visible=False), gr.update(value=msg, visible=True)
+    else:
+        return gr.update(visible=is_finetune, value=current_test_value), gr.update(value="", visible=False)
 
 
 with gr.Blocks() as demo:
@@ -95,9 +96,10 @@ with gr.Blocks() as demo:
         "⚙️ This demo is currently running on <strong>Hugging Face ZeroGPU Space</strong>: Dynamic resources (NVIDIA A100)."
     )
 
-    with gr.Row():  
+    with gr.Row():
         with gr.Column(scale=2):
             video_input = gr.Video(label="Upload a Video (e.g. mp4)")
+
             train_dataset = gr.Dropdown(
                 label="Train Dataset",
                 choices=["cvd_2014", "konvid_1k", "live_vqc", "lsvq_train", "youtube_ugc"],
@@ -111,6 +113,7 @@ with gr.Blocks() as demo:
                 value="konvid_1k",
                 visible=False
             )
+            mode_message = gr.Markdown("", visible=False)
             model_dropdown = gr.Dropdown(
                 label="Our Models",
                 choices=["diva-vqa", "diva-vqa_large"],
@@ -121,7 +124,6 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             output_box = gr.Textbox(label="Predicted Perceptual Quality Score (0–100)", lines=5)
 
-
     train_dataset.change(
         fn=toggle_finetune_visibility,
         inputs=train_dataset,
@@ -129,22 +131,21 @@ with gr.Blocks() as demo:
     )
 
     is_finetune_checkbox.change(
-        fn=toggle_test_dataset,
-        inputs=[is_finetune_checkbox, train_dataset],
-        outputs=test_dataset
+        fn=update_test_dataset,
+        inputs=[is_finetune_checkbox, train_dataset, test_dataset],
+        outputs=[test_dataset, mode_message]
     )
 
     train_dataset.change(
-        fn=toggle_test_dataset,
-        inputs=[is_finetune_checkbox, train_dataset],
-        outputs=test_dataset
+        fn=update_test_dataset,
+        inputs=[is_finetune_checkbox, train_dataset, test_dataset],
+        outputs=[test_dataset, mode_message]
     )
 
-
     run_button.click(
         fn=run_diva_vqa,
         inputs=[video_input, is_finetune_checkbox, train_dataset, test_dataset, model_dropdown],
         outputs=output_box
     )
 
-demo.launch()
+demo.launch()