import gradio as gr
import pandas as pd

def txt_to_parquet(file):
    with open(file.name, "r", encoding="utf-8") as f:
        lines = f.readlines()
    df = pd.DataFrame({"text": [line.strip() for line in lines]})
    output_file = "output.parquet"
    df.to_parquet(output_file, engine="pyarrow", index=False)
    return output_file 
    
with gr.Blocks() as app:
    gr.Markdown("## 📝 Convert TXT File to Parquet")
    txt_file = gr.File(label="Upload .txt file", file_types=[".txt"])
    convert_button = gr.Button("Convert to Parquet")
    parquet_file = gr.File(label="Download .parquet file", interactive=False)
    convert_button.click(txt_to_parquet, inputs=txt_file, outputs=parquet_file)

app.launch()