Update app.py

app.py

@@ -2,22 +2,25 @@ import gradio as gr
 import os
 
 # Function to show HTML content using an iframe
-def show_html_in_iframe(page_idx):
+def show_html_in_iframe(page_idx, html_files, curr_split):
     # Construct the URL using the selected index
-    url = f'https://wmtis.s3.eu-west-1.amazonaws.com/test/{html_files[page_idx]}'
+    url = f'https://wmtis.s3.eu-west-1.amazonaws.com/{curr_split}/{html_files[page_idx]}'
     iframe_html = f'<iframe src="{url}" width="100%" height="1000"></iframe>'
     return iframe_html
 
-# Path to the directory containing HTML files
-html_files_path = os.path.join("html_files", "test")
-html_files = [f for f in os.listdir(html_files_path) if f.endswith('.html')]
-
 with gr.Blocks() as demo:
-  # 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 a dropdown for selecting the instance type
+    curr_split = gr.Radio(choices=['train', 'dev'], label='split', value='dev')
+
+    # 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], outputs=[gr.HTML()], api_name="HTML Viewer with Event Listeners")
+    # 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")
 
 # Display the interface
 demo.launch()