Update app.py

app.py

@@ -15,9 +15,13 @@ def filter_dataset(dataset_name, split_name, keywords_text):
         if not keywords:
             return pd.DataFrame(), "Error: No keywords provided."
 
-        filtered_dataset = dataset.filter(
-            lambda example: any(keyword in example["prompt"].lower() for keyword in keywords)
-        )
+        # Define a filtering function that handles potential KeyErrors
+        def filter_func(example):
+            prompt_value = example.get("prompt", "")  # Get prompt, default to empty string
+            return any(keyword in prompt_value.lower() for keyword in keywords)
+
+        filtered_dataset = dataset.filter(filter_func)
+
 
         # --- 3.  Select Indices (Efficiently) ---
         matching_indices = []