perf: Default to interactive display for SQL in anywidget mode

Previously, SQL queries in anywidget mode would fall back to deferred execution, showing a dry run instead of an interactive table.

This change modifies the display logic to directly use the anywidget interactive display for SQL queries, providing a more consistent and responsive user experience. A test case has been added to verify this behavior.

Author: shuoweil

bigframes/dataframe.py

@@ -783,11 +783,26 @@ def __repr__(self) -> str:
 
         opts = bigframes.options.display
         max_results = opts.max_rows
-        # anywdiget mode uses the same display logic as the "deferred" mode
-        # for faster execution
-        if opts.repr_mode in ("deferred", "anywidget"):
+
+        # Only deferred mode shows dry run
+        if opts.repr_mode in ("deferred"):
             return formatter.repr_query_job(self._compute_dry_run())
 
+        # Anywidget mode uses interative display
+        if opts.repr_mode == "anywidget":
+            # Try to display with anywidget, fall back to deferred if not in IPython
+            try:
+                from IPython.display import display as ipython_display
+
+                from bigframes import display
+
+                widget = display.TableWidget(self.copy())
+                ipython_display(widget)
+                return ""  # Return empty string since we used display()
+            except (AttributeError, ValueError, ImportError):
+                # Not in IPython environment, fall back to deferred mode
+                return formatter.repr_query_job(self._compute_dry_run())
+
         # TODO(swast): pass max_columns and get the true column count back. Maybe
         # get 1 more column than we have requested so that pandas can add the
         # ... for us?

tests/system/small/test_anywidget.py

@@ -436,6 +436,21 @@ def test_widget_creation_should_load_css_for_rendering(table_widget):
     assert ".bigframes-widget .footer" in css_content
 
 
+def test_sql_anywidget_mode(session: bf.Session):
+    """
+    Test that a SQL query runs in anywidget mode.
+    """
+    sql = "SELECT * FROM `bigquery-public-data.usa_names.usa_1910_current` LIMIT 5"
+
+    with bf.option_context("display.repr_mode", "anywidget"):
+        df = session.read_gbq(sql)
+        # In a real environment, this would display a widget.
+        # For testing, we just want to make sure we're in the anywidget code path.
+        # The `_repr_html_` method in anywidget mode will return an empty string
+        # and display the widget via IPython's display mechanism.
+        assert df._repr_html_() == ""
+
+
 # TODO(shuowei): Add tests for custom index and multiindex
 # This may not be necessary for the SQL Cell use case but should be
 # considered for completeness.