import gradio as gr
import pandas as pd

def topn_tokens(sequence, n):
    chars = list(sequence)
    n_column = [n] * len(chars)  # Create a column with the chosen integer repeated
    df = pd.DataFrame({'Character': chars, 'N': n_column})  # Create the DataFrame
    return df

demo = gr.Interface(
    topn_tokens,
    [
        "text",
        gr.Dropdown([int(i) for i in range(1, 21)]),  # Dropdown with numbers from 1 to 20
    ],
    "dataframe",
    description="Choose a number between 1-20 to display n tokens",
)

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