|
import pandas as pd |
|
import pyperclip |
|
|
|
|
|
df = pd.read_csv('./result_good-workers.csv') |
|
|
|
prompt = ''' |
|
Here is a story about an event: |
|
|
|
{story} |
|
|
|
The goal of the story: {goal} |
|
|
|
Condition: {condition} |
|
|
|
Do you think this condition will lead to the goal of the story? Answer it in Yes or No. |
|
''' |
|
|
|
for index, row in df.iterrows(): |
|
story = row['Input.story'] |
|
goal = row['Answer.goal'] |
|
condition = row['Answer.condition'] |
|
not_condition = row['Answer.not_condition'] |
|
|
|
nprompt = prompt.format(story=story, goal=goal, condition=condition) |
|
print(nprompt) |
|
pyperclip.copy(nprompt) |
|
input("Press Enter to for next not condition") |
|
nprompt = prompt.format(story=story, goal=goal, condition=not_condition) |
|
print(nprompt) |
|
pyperclip.copy(nprompt) |
|
input("Press Enter to for next story") |