import pandas as pd
import json

# Read the CSV file into a DataFrame
df = pd.read_csv('train.csv')

# Check if the CSV file indeed has 3 columns
if len(df.columns) != 3:
    raise ValueError("The CSV file should have exactly 3 columns.")

# Display the first few rows of the DataFrame to see your data
print(df.head())

with open('file.json', 'w') as wrfile:
    count = 0
    for index, row in df.iterrows():
        #print(index, row['ref'], row['mr'])
        # Create a new JSON object
        try:
            new_json = {
                "instruction": row['ref'],
                "input": "",
                "output": row['mr']
            }
        except Exception as e:
            print(f'发生错误: {e}')
            #print("cttt cut ", name_parts)
            # exit()

        # Print the new JSON object
        output = json.dumps(new_json, ensure_ascii=False)
        output = output + ',\n'
        print(output)
        wrfile.write(output)
        count += 1

    print("count :", count)
# ... You can now further process or analyze your data ...