app.py

# Import necessary libraries
import gradio as gr
from data_loader import read_dask_data, read_polars_data, read_another_dask_data

# Function to handle the data loading logic
def load_and_display_data():
    # Load datasets
    dask_data = read_dask_data()
    polars_data = read_polars_data()
    another_dask_data = read_another_dask_data()

    # Optionally process data (e.g., filter, summarize)
    # Display the first few rows of the datasets
    return {
        "Dask Data from Codeforces Python Submissions": dask_data.head(),
        "Polars Data from Python Reasoning Dataset": polars_data.head(),
        "Another Dask Data from Python Github Code": another_dask_data.head()
    }

# Create a Gradio interface to display the data
demo = gr.Interface(
    load_and_display_data,
    inputs=[],  # No inputs for this simple demo
    outputs="json",  # Display output as a JSON to show data head
    title="Dataset Loader Demo",
)

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