Update app.py

app.py

@@ -180,7 +180,14 @@ def get_data(rl_env, task_id, path, selected_filters: List[str] = None) -> pd.Da
 
     # Apply selected filters for difficulty or pattern if provided
     if selected_filters:
-        filtered_data = filtered_data[filtered_data['Pattern'].isin(selected_filters) | filtered_data['Difficulty'].isin(selected_filters)]
+        if 'Pattern' in filtered_data.columns and 'Difficulty' in filtered_data.columns:
+            filtered_data = filtered_data[
+                filtered_data['Pattern'].isin(selected_filters) | filtered_data['Difficulty'].isin(selected_filters)
+            ]
+        elif 'Pattern' in filtered_data.columns:
+            filtered_data = filtered_data[filtered_data['Pattern'].isin(selected_filters)]
+        elif 'Difficulty' in filtered_data.columns:
+            filtered_data = filtered_data[filtered_data['Difficulty'].isin(selected_filters)]
 
     # Drop the "Task-ID" and "Task" columns
     filtered_data = filtered_data.drop(columns=["Task-ID", "Task"])