Update app.py

app.py

@@ -1,35 +1,32 @@
 import gradio as gr
 from datasets import load_dataset
+import numpy as np
+from PIL import Image
 
-def peek_image_field():
+def show_first():
     ds = load_dataset("MultimodalUniverse/jwst", split="train", streaming=True)
-    first = next(iter(ds))
+    rec = next(iter(ds))
+    img = rec["image"]
 
-    keys = list(first.keys())
-    lines = [f"Keys: {keys}", "Types:"]
-    lines += [f"  {k}: {type(first[k]).__name__}" for k in keys]
+    # convert the flux array to a viewable grayscale image
+    flux = np.array(img["flux"], dtype=np.float32)
+    lo = np.percentile(flux, 1)
+    hi = np.percentile(flux, 99)
+    if hi <= lo:  # fallback if data is odd
+        lo, hi = float(flux.min()), float(flux.max())
+    norm = np.clip((flux - lo) / (hi - lo + 1e-9), 0, 1)
+    arr = (norm * 255).astype(np.uint8)
+    pil = Image.fromarray(arr, mode="L")
 
-    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)
+    caption = f"object_id: {rec.get('object_id')}, band: {img.get('band')}"
+    return pil, caption
 
 demo = gr.Interface(
-    fn=peek_image_field,
+    fn=show_first,
     inputs=None,
-    outputs=gr.Textbox(lines=14, label="Output"),
-    title="JWST dataset schema peek",
-    description="Shows keys, types, and image field details for the first record."
+    outputs=[gr.Image(type="pil", label="Preview"), gr.Textbox(label="Info", lines=2)],
+    title="JWST sample preview",
+    description="First streamed record as image + a little metadata."
 )
 
 demo.launch()