Update app.py

app.py

@@ -9,18 +9,22 @@ def show_html_in_iframe(page_idx, html_files, curr_split):
     return iframe_html
 
 with gr.Blocks() as demo:
-    # Create a dropdown for selecting the instance type
-    curr_split = gr.Radio(choices=['train', 'dev'], label='split', value='dev')
+    with gr.Row():
+        instance_type_dropdown = gr.Dropdown(choices=['test', 'dev'], label='Instance Type')
+        slider = gr.Slider(minimum=0, step=1, label='Data Instance')
+
+    # Dynamic update of slider based on instance type
+    def update_slider_options(instance_type):
+        html_files = get_html_files_list(instance_type)
+        slider.update(options=list(range(len(html_files))))
+        return len(html_files) - 1  # Return maximum index as new value for slider
+
+    # Link dropdown change to updating slider options
+    instance_type_dropdown.change(fn=update_slider_options, inputs=instance_type_dropdown, outputs=slider)
 
-    # Path to the directory containing HTML files
-    html_files_path = os.path.join("html_files", curr_split)
-    html_files = [f for f in os.listdir(html_files_path) if f.endswith('.html')]
-    
-    # Create a slider component to select the HTML page
-    slider = gr.Slider(minimum=0, maximum=len(html_files)-1, step=1, label='Data Instance')
 
     # Create the Gradio interface
-    slider.release(show_html_in_iframe, inputs=[slider, html_files, curr_split], outputs=[gr.HTML()], api_name="HTML Viewer with Event Listeners")
+    slider.release(show_html_in_iframe, inputs=[slider, html_files, instance_type_dropdown], outputs=[gr.HTML()], api_name="HTML Viewer with Event Listeners")
 
 # Display the interface
 demo.launch()