from datasets import load_dataset
import gradio as gr

# Load your dataset
ds = load_dataset("greghavens/fable-5-coding-and-debugging-traces")


def get_dataset_info():
  split_info = "Loaded split: train\n"
  total_rows = f"Total rows: {len(ds['train'])}\n"
  columns = f"Columns: {', '.join(ds['train'].column_names)}"
  return split_info + total_rows + columns


demo = gr.Interface(
    fn=get_dataset_info,
    inputs=[],
    outputs="text",
    title="Dataset Inspector",
    description="Inspect fable-5-coding-and-debugging-traces dataset",
)

if __name__ == "__main__":
  demo.launch()