Update app.py

app.py

@@ -230,7 +230,6 @@ def convert_to_title_case(text: str) -> str:
 
 run_update_dataset()
 
-block = gr.Blocks(css=custom_css)  # Attach the custom CSS here
 with block:
     with gr.Row(elem_id="header-row"):
         # TITLE IMAGE
@@ -253,25 +252,44 @@ with block:
 
     path_ = download_leaderboard_dataset()
     # gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
+    
+    # Add Checkboxes for Filtering
+    with gr.Row():
+        selected_patterns = gr.CheckboxGroup(
+            choices=["Pattern A", "Pattern B", "Pattern C"],
+            label="Select Patterns"
+        )
+        selected_difficulty = gr.CheckboxGroup(
+            choices=["Easy", "Medium", "Hard"],
+            label="Select Difficulty"
+        )
+    
     # ENVIRONMENT TABS
     with gr.Tabs() as tabs: # elem_classes="tab-buttons"
         for env_index in range(0, len(hivex_envs)):
             hivex_env = hivex_envs[env_index]
             with gr.Tab(f"{hivex_env['title']}") as env_tabs:
-                # TASK TABS
+                
+                # Apply filtering based on selected checkboxes
                 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}"):
-                        with gr.Row():
-                            data = get_data(hivex_env["hivex_env"], task_id, path_)
-                            row_count = len(data)  # Number of rows in the data
-                            
-                            gr_dataframe = gr.components.Dataframe(
-                                value=data,
-                                headers=["User", "Model"],
-                                datatype=["markdown", "markdown"],
-                                row_count=(row_count, 'fixed')  # Set to the exact number of rows in the data
-                            )
+                    task_info = get_task(hivex_env["hivex_env"], task_id, path_)
+                    task_title = convert_to_title_case(task_info)
+                    
+                    # Example filtering logic based on selected patterns and difficulty
+                    if (any(pattern in task_info for pattern in selected_patterns.value) and 
+                        any(difficulty in task_info for difficulty in selected_difficulty.value)):
+                        
+                        with gr.TabItem(f"Task {task_id}: {task_title}"):
+                            with gr.Row():
+                                data = get_data(hivex_env["hivex_env"], task_id, path_)
+                                row_count = len(data)  # Number of rows in the data
+                                
+                                gr_dataframe = gr.components.Dataframe(
+                                    value=data,
+                                    headers=["User", "Model"],
+                                    datatype=["markdown", "markdown"],
+                                    row_count=(row_count, 'fixed')  # Set to the exact number of rows in the data
+                                )
 
 
 scheduler = BackgroundScheduler()