# Demo: (Dataframe) -> (Dataframe)

import gradio as gr
import numpy as np

def transpose(matrix):
    return matrix.T


iface = gr.Interface(
    transpose,
    gr.inputs.Dataframe(type="numpy", datatype="number", row_count=5, col_count=3),
    "numpy",
    examples=[
        [np.zeros((3,3)).tolist()],
        [np.ones((2,2)).tolist()],
        [np.random.randint(0, 10, (3,10)).tolist()],
        [np.random.randint(0, 10, (10,3)).tolist()],
        [np.random.randint(0, 10, (10,10)).tolist()],
    ]
)

iface.test_launch()

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