Update app.py

app.py

@@ -17,7 +17,7 @@ def get_choices(df, column):
     vals = sorted(df[column].dropna().unique().tolist())
     return ["All"] + [str(v) for v in vals]
 
-def filter_data(platform, behavior_category, confidence):
+def filter_data(platform, behavior_category, confidence, search_term):
     df = load_data()
     if platform != "All":
         df = df[df["platform"].astype(str) == platform]
@@ -25,6 +25,14 @@ def filter_data(platform, behavior_category, confidence):
         df = df[df["behavior_category"].astype(str) == behavior_category]
     if confidence != "All":
         df = df[df["interpretive_confidence"].astype(str) == confidence]
+    if search_term.strip():
+        mask = df.apply(
+            lambda row: row.astype(str).str.contains(
+                search_term.strip(), case=False, na=False
+            ).any(),
+            axis=1
+        )
+        df = df[mask]
     return df
 
 def submit_observation(
@@ -86,17 +94,30 @@ with gr.Blocks(title="Coherence Gap Explorer") as demo:
                 value="All", label="Interpretive Confidence"
             )
 
+        search_box = gr.Textbox(
+            label="Search",
+            placeholder="Search across all columns...",
+            lines=1
+        )
+
+        result_count = gr.Markdown("")
+
         table = gr.Dataframe(
             value=initial_df, label="Observations",
             interactive=False, wrap=True
         )
 
-        for dd in [platform_dd, category_dd, confidence_dd]:
-            dd.change(
-                fn=filter_data,
-                inputs=[platform_dd, category_dd, confidence_dd],
-                outputs=table
-            )
+        def update(platform, behavior_category, confidence, search_term):
+            df = filter_data(platform, behavior_category, confidence, search_term)
+            count = f"*Showing {len(df)} of {len(load_data())} observations*"
+            return df, count
+
+        inputs = [platform_dd, category_dd, confidence_dd, search_box]
+
+        for component in [platform_dd, category_dd, confidence_dd]:
+            component.change(fn=update, inputs=inputs, outputs=[table, result_count])
+
+        search_box.change(fn=update, inputs=inputs, outputs=[table, result_count])
 
     with gr.Tab("Submit Observation"):
         gr.Markdown("### Add a New Observation")