chore: fix experimental blob _repr_html_ only previews image content type (#1306)

Author: GarrettWu

bigframes/dataframe.py

@@ -19,6 +19,7 @@
 import datetime
 import inspect
 import itertools
+import json
 import re
 import sys
 import textwrap
@@ -747,10 +748,9 @@ def _repr_html_(self) -> str:
                 if df[col].dtype == bigframes.dtypes.OBJ_REF_DTYPE
             ]
             for col in blob_cols:
+                df[col] = df[col]._apply_unary_op(ops.obj_fetch_metadata_op)
+                # TODO(garrettwu): Not necessary to get access urls for all the rows. Update when having a to get URLs from local data.
                 df[col] = df[col]._apply_unary_op(ops.ObjGetAccessUrl(mode="R"))
-                df[col] = df[col]._apply_unary_op(
-                    ops.JSONValue(json_path="$.access_urls.read_url")
-                )
 
         # 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
@@ -767,10 +767,21 @@ def _repr_html_(self) -> str:
             # Allows to preview images in the DataFrame. The implementation changes the string repr as well, that it doesn't truncate strings or escape html charaters such as "<" and ">". We may need to implement a full-fledged repr module to better support types not in pandas.
             if bigframes.options.experiments.blob:
 
-                def url_to_image_html(url: str) -> str:
-                    return f'<img src="{url}">'
+                def obj_ref_rt_to_html(obj_ref_rt) -> str:
+                    obj_ref_rt_json = json.loads(obj_ref_rt)
+                    content_type = typing.cast(
+                        str,
+                        obj_ref_rt_json["objectref"]["details"]["gcs_metadata"][
+                            "content_type"
+                        ],
+                    )
+                    if content_type.startswith("image"):
+                        url = obj_ref_rt_json["access_urls"]["read_url"]
+                        return f'<img src="{url}">'
+
+                    return f'uri: {obj_ref_rt_json["objectref"]["uri"]}, authorizer: {obj_ref_rt_json["objectref"]["authorizer"]}'
 
-                formatters = {blob_col: url_to_image_html for blob_col in blob_cols}
+                formatters = {blob_col: obj_ref_rt_to_html for blob_col in blob_cols}
 
                 # set max_colwidth so not to truncate the image url
                 with pandas.option_context("display.max_colwidth", None):

bigframes/operations/blob.py

@@ -182,7 +182,7 @@ def display_single_url(read_url: str, content_type: str):
                 ipy_display.display(ipy_display.Video(url=read_url))
             else:  # display as raw data
                 response = requests.get(read_url)
-                ipy_display.display(response.content, raw=True)
+                ipy_display.display(response.content)
 
         for _, row in df.iterrows():
             display_single_url(row["read_url"], row["content_type"])