Update app.py

app.py

@@ -1,28 +1,35 @@
 import gradio as gr
 from datasets import load_dataset
 
-def peek_schema_text():
-    try:
-        ds = load_dataset("MultimodalUniverse/jwst", split="train", streaming=True)
-        first = next(iter(ds))
-        keys = list(first.keys())
-        lines = [f"Keys: {keys}", "Types:"]
-        lines += [f"  {k}: {type(first[k]).__name__}" for k in keys]
-        # show short previews if any text fields exist
-        previews = [f"  {k}: {first[k][:200]}" for k in keys if isinstance(first[k], str) and first[k]]
-        if previews:
-            lines.append("Text previews:")
-            lines += previews
-        return "\n".join(lines)
-    except Exception as e:
-        return f"Error: {e}"
+def peek_image_field():
+    ds = load_dataset("MultimodalUniverse/jwst", split="train", streaming=True)
+    first = next(iter(ds))
+
+    keys = list(first.keys())
+    lines = [f"Keys: {keys}", "Types:"]
+    lines += [f"  {k}: {type(first[k]).__name__}" for k in keys]
+
+    img = first.get("image", {})
+    img_keys = list(img.keys()) if isinstance(img, dict) else []
+    lines.append(f"image subkeys: {img_keys}")
+    path = img.get("path", None) if isinstance(img, dict) else None
+    if isinstance(path, str):
+        lines.append(f"image path: {path[:200]}")
+
+    # brief text previews if present
+    previews = [f"  {k}: {first[k][:200]}" for k in keys if isinstance(first[k], str) and first[k]]
+    if previews:
+        lines.append("Text previews:")
+        lines += previews
+
+    return "\n".join(lines)
 
 demo = gr.Interface(
-    fn=peek_schema_text,
+    fn=peek_image_field,
     inputs=None,
-    outputs="text",
+    outputs=gr.Textbox(lines=14, label="Output"),
     title="JWST dataset schema peek",
-    description="Shows keys and types for the first streamed record."
+    description="Shows keys, types, and image field details for the first record."
 )
 
 demo.launch()