import gradio as gr
import pandas as pd
import requests
import spaces

@spaces.GPU
def convert_parquet_to_jsonl(parquet_file_or_url):
    if parquet_file_or_url.startswith("http"):
        response = requests.get(parquet_file_or_url)
        parquet_file = response.content
    else:
        parquet_file = parquet_file_or_url.name
    df = pd.read_parquet(parquet_file)
    jsonl_data = df.to_json(orient='records', lines=True)
    return jsonl_data

theme = "Ytheme/Minecraft"

demo = gr.Interface(theme=theme,
    fn=convert_parquet_to_jsonl,
    inputs=[gr.File(label="Parquet File"), gr.Textbox(label="Parquet File URL")],
    outputs=[gr.Textbox(label="JSONL Output")],
    title="Parquet to JSONL Converter",
    description="Input a Parquet file by a downloadable link or file upload and convert it to JSONL format"
)

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