import duckdb
import gradio as gr

con = duckdb.connect(":memory:")

def greet(SQL_Query):
    return con.sql(SQL_Query).df()

examples = [
    "SELECT dump, avg(token_count) FROM 'hf://datasets/HuggingFaceFW/fineweb/sample/10BT/*.parquet' GROUP BY dump;",
]
demo = gr.Interface(fn=greet, inputs="text", outputs="dataframe", examples=examples)
demo.launch()