mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-19 12:10:28 +08:00
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
def generate_cache():
|
|
start_seed = int(sys.argv[1])
|
|
end_seed = int(sys.argv[2])
|
|
|
|
# Generate story sections
|
|
prompt_num = 0
|
|
action_queue = []
|
|
prompt = prompts[prompt_num]
|
|
for seed in range(start_seed, end_seed):
|
|
result = retrieve_from_cache(seed, prompt_num, [], "story")
|
|
if result is not None:
|
|
response = result
|
|
else:
|
|
prompt = prompts[prompt_num]
|
|
# print("\n Story prompt is ", prompt)
|
|
response = generate_story_block(prompt)
|
|
# print("\n Story response is ", response)
|
|
cache_file(seed, prompt_num, [], response, "story")
|
|
|
|
action_queue.append([seed, 0, [], response])
|
|
|
|
while (True):
|
|
|
|
next_gen = action_queue.pop(0)
|
|
seed = next_gen[0]
|
|
prompt_num = next_gen[1]
|
|
choices = next_gen[2]
|
|
last_action_result = next_gen[3]
|
|
|
|
action_results = retrieve_from_cache(seed, prompt_num, choices, "choices")
|
|
|
|
if action_results is not None:
|
|
response = action_results
|
|
|
|
else:
|
|
if len(choices) is 0:
|
|
prompt = prompts[prompt_num] + last_action_result
|
|
else:
|
|
prompt = continuing_prompts[prompt_num] + last_action_result
|
|
# print("\n\n Action prompt is \n ", prompt)
|
|
action_results = [generate_action_result(prompt, phrase) for phrase in phrases]
|
|
response = json.dumps(action_results)
|
|
|
|
# print("\n\n Action
|
|
cache_file(seed, prompt_num, choices, response, "choices")
|
|
|
|
un_jsoned = json.loads(response)
|
|
for j in range(4):
|
|
new_choices = choices[:]
|
|
new_choices.append(j)
|
|
action_queue.append([seed, 0, new_choices, un_jsoned[j][1]]) |