import pandas as pd
import json
import gradio as gr

def create_jsonl(csv_file):
    df = pd.read_csv(csv_file)

    column_list = df.columns
    output = 'metadata.jsonl'

    total_data = []

    for index, row in df.iterrows():

        data = {}

        for column in column_list:

            if column == 'file_name':
                file_name = row[column]

            data[column] = row[column]

        row_data = {"file_name": file_name, "ground_truth": json.dumps(data)}
        
        total_data.append(row_data)

    with open(output, 'a', encoding='utf-8') as f:
        
        for row_data in total_data:
            f.write(json.dumps(row_data) + '\n')

    return output

demo = gr.Interface(fn=create_jsonl,inputs=gr.File(),outputs=gr.File(), title=f"jsonl converter",)
demo.launch()