Update app.py

app.py

@@ -8,9 +8,18 @@ def show_html_in_iframe(page_idx, html_files, curr_split):
     iframe_html = f'<iframe src="{url}" width="100%" height="1000"></iframe>'
     return iframe_html
 
+# 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')]
+
+# 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
+
+
 with gr.Blocks() as demo:
     with gr.Row():
-        instance_type_dropdown = gr.Dropdown(choices=['test', 'dev'], label='Instance Type')
+        instance_type_dropdown = gr.Dropdown(choices=['train', 'dev'], label='Instance Type')
         slider = gr.Slider(minimum=0, step=1, label='Data Instance')
 
     # Dynamic update of slider based on instance type