Update app.py

app.py

@@ -325,36 +325,35 @@ with block:
                         [str(dp_id) for dp_id in difficulty_pattern_ids], label=dp_key
                     )
 
-                    for task_id in range(0, hivex_env["task_count"]):
-                        task_title = convert_to_title_case(
-                            get_task(hivex_env["hivex_env"], task_id, path_)
-                        )
-                        with gr.TabItem(f"Task {task_id}: {task_title}"):
+                for task_id in range(0, hivex_env["task_count"]):
+                    task_title = convert_to_title_case(
+                        get_task(hivex_env["hivex_env"], task_id, path_)
+                    )
+                    with gr.TabItem(f"Task {task_id}: {task_title}"):
 
-                            # Display initial data
-                            data = get_data(hivex_env["hivex_env"], task_id, path_)
-                            row_count = len(data)
+                        # Display initial data
+                        data = get_data(hivex_env["hivex_env"], task_id, path_)
+                        row_count = len(data)
 
-                            gr_dataframe = gr.DataFrame(
-                                value=data,
-                                headers=["User", "Model"],
-                                datatype=["markdown", "markdown"],
-                                row_count=(row_count, "fixed"),
-                            )
+                        gr_dataframe = gr.DataFrame(
+                            value=data,
+                            headers=["User", "Model"],
+                            datatype=["markdown", "markdown"],
+                            row_count=(row_count, "fixed"),
+                        )
 
-                            # Use gr.State to hold environment and task information
-                            rl_env_state = gr.State(value=hivex_env["hivex_env"])
-                            task_id_state = gr.State(value=task_id)
-                            path_state = gr.State(value=path_)
+                        # Use gr.State to hold environment and task information
+                        rl_env_state = gr.State(value=hivex_env["hivex_env"])
+                        task_id_state = gr.State(value=task_id)
+                        path_state = gr.State(value=path_)
 
-                            # Add a callback to update the DataFrame when checkboxes are changed
+                        # Add a callback to update the DataFrame when checkboxes are changed
+                        if selected_checkboxes:
                             selected_checkboxes.change(
                                 fn=update_filtered_data,
                                 inputs=[selected_checkboxes, rl_env_state, task_id_state, path_state],
                                 outputs=gr_dataframe,
                             )
-                else:
-                    gr.HTML("<p>No difficulty or pattern data available for this environment.</p>")
 
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart, "interval", seconds=86400)