Update app.py

app.py

@@ -12,15 +12,21 @@ def show_html_in_iframe(page_idx, curr_split):
 # Example lists of HTML files for 'test' and 'dev' instances
 train_html_files = [f for f in os.listdir(os.path.join("html_files", "train")) if f.endswith('.html')]
 dev_html_files = [f for f in os.listdir(os.path.join("html_files", "dev")) if f.endswith('.html')]
+test_html_files = [f for f in os.listdir(os.path.join("html_files", "test")) if f.endswith('.html')]
 
 # Function to return the appropriate list based on instance type
 def get_html_files_list(instance_type):
-    return train_html_files if instance_type == 'train' else dev_html_files
+    if instance_type == 'train':
+        return train_html_files
+    elif instance_type == 'dev':
+        return dev_html_files
+    elif instance_type == 'test':
+        return test_html_files
 
 
 with gr.Blocks() as demo:
     with gr.Row():
-        instance_type_dropdown = gr.Dropdown(choices=['train', 'dev'], label='Instance Type')
+        instance_type_dropdown = gr.Dropdown(choices=['train', 'dev', 'test'], label='Instance Type')
         slider = gr.Slider(minimum=0, step=1, label='Data Instance')
 
     # Dynamic update of slider based on instance type