jsonl-converter / app.py
uartimcs's picture
Update app.py
86fdbb7 verified
raw
history blame contribute delete
838 Bytes
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()