File size: 835 Bytes
c65837d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c695ced
c65837d
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
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="text", title=f"jsonl converter",)
demo.launch()