avatar_gen/print_csv.py

27 lines
723 B
Python

import pandas as pd
from avator import gen_avator
from chatbot import ChatBot
def log_info(id, title, description, initPrompt):
print('----------id----------')
print(id)
print('----------title----------')
print(title)
print('----------description----------')
print(description)
print('----------initPrompt----------')
print(initPrompt)
print('----------end----------')
# Specify the path to the CSV file
file_path = "_Prompt__character_nsfw.csv"
# Read the CSV file
df = pd.read_csv(file_path, sep=",")
# Display the DataFrame
# print(df)
row_index = 49
log_info(df.loc[row_index, 'id'], df.loc[row_index, 'title'],df.loc[row_index, 'description'], df.loc[row_index, 'initPrompt'])