|
prompts_hs = """ 请以“{headstart}”为开头,编写一个小说的第一幕。 |
|
|
|
- 尽量短,不要包含太多情节,因为你接下来将会与用户互动续写下面的情节,要留出足够的互动空间。 |
|
- 出现人物时,给出人物的名字。 |
|
- 积极地运用环境描写、人物描写等手法,让读者能够感受到你的故事世界。 |
|
- 积极地运用修辞手法,比如比喻、拟人、排比、对偶、夸张等等。 |
|
- 字数要求:第一幕的字数少于300字,且少于2个段落。 |
|
""" |
|
|
|
prompts_interact = """ 小说的前文回顾: |
|
「 |
|
{previously_on_story} |
|
」 |
|
|
|
你是一个作家,根据以上的情节,给出4种不同的后续剧情发展方向,每个发展方向都精明扼要地用一句话说明。稍后,我将在这4个选择中,挑选一种剧情发展。 |
|
|
|
输出格式例如: |
|
1. 后续剧情发展1 |
|
2. 后续剧情发展2 |
|
3. 后续剧情发展3 |
|
4. 后续剧情发展4 |
|
""" |
|
|
|
|
|
prompts_resume = """小说的前文回顾: |
|
「 |
|
{previously_on_story} |
|
」 |
|
|
|
你是一个作家,我们正在互相讨论,确定后续剧情的发展。 |
|
在以下的剧情发展中, |
|
「 |
|
{choice} |
|
」 |
|
我认为更合理的是:{user_choice}。 |
|
请在前文的基础上(不要重复前文),围绕我选定的剧情情节,编写小说的下一幕。 |
|
|
|
- 禁止杜撰不符合我选择的剧情。 |
|
- 尽量短,不要包含太多情节,因为你接下来将会与用户互动续写下面的情节,要留出足够的互动空间。 |
|
- 不要重复前文。 |
|
- 出现人物时,给出人物的名字。 |
|
- 积极地运用环境描写、人物描写等手法,让读者能够感受到你的故事世界。 |
|
- 积极地运用修辞手法,比如比喻、拟人、排比、对偶、夸张等等。 |
|
- 小说的下一幕字数少于300字,且少于2个段落。 |
|
""" |
|
|
|
|
|
prompts_terminate = """小说的前文回顾: |
|
「 |
|
{previously_on_story} |
|
」 |
|
|
|
你是一个作家,我们正在互相讨论,确定后续剧情的发展。 |
|
现在,故事该结束了,我认为最合理的故事结局是:{user_choice}。 |
|
|
|
请在前文的基础上(不要重复前文),编写小说的最后一幕。 |
|
|
|
- 不要重复前文。 |
|
- 出现人物时,给出人物的名字。 |
|
- 积极地运用环境描写、人物描写等手法,让读者能够感受到你的故事世界。 |
|
- 积极地运用修辞手法,比如比喻、拟人、排比、对偶、夸张等等。 |
|
- 字数要求:最后一幕的字数少于1000字。 |
|
""" |
|
|
|
|
|
from toolbox import CatchException, update_ui, update_ui_lastest_msg |
|
from crazy_functions.multi_stage.multi_stage_utils import GptAcademicGameBaseState |
|
from crazy_functions.crazy_utils import request_gpt_model_in_new_thread_with_ui_alive |
|
from request_llms.bridge_all import predict_no_ui_long_connection |
|
from crazy_functions.game_fns.game_utils import get_code_block, is_same_thing |
|
import random |
|
|
|
|
|
class MiniGame_ResumeStory(GptAcademicGameBaseState): |
|
story_headstart = [ |
|
'先行者知道,他现在是全宇宙中唯一的一个人了。', |
|
'深夜,一个年轻人穿过天安门广场向纪念堂走去。在二十二世纪编年史中,计算机把他的代号定为M102。', |
|
'他知道,这最后一课要提前讲了。又一阵剧痛从肝部袭来,几乎使他晕厥过去。', |
|
'在距地球五万光年的远方,在银河系的中心,一场延续了两万年的星际战争已接近尾声。那里的太空中渐渐隐现出一个方形区域,仿佛灿烂的群星的背景被剪出一个方口。', |
|
'伊依一行三人乘坐一艘游艇在南太平洋上做吟诗航行,他们的目的地是南极,如果几天后能顺利到达那里,他们将钻出地壳去看诗云。', |
|
'很多人生来就会莫名其妙地迷上一样东西,仿佛他的出生就是要和这东西约会似的,正是这样,圆圆迷上了肥皂泡。' |
|
] |
|
|
|
|
|
def begin_game_step_0(self, prompt, chatbot, history): |
|
|
|
self.headstart = random.choice(self.story_headstart) |
|
self.story = [] |
|
chatbot.append(["互动写故事", f"这次的故事开头是:{self.headstart}"]) |
|
self.sys_prompt_ = '你是一个想象力丰富的杰出作家。正在与你的朋友互动,一起写故事,因此你每次写的故事段落应少于300字(结局除外)。' |
|
|
|
|
|
def generate_story_image(self, story_paragraph): |
|
try: |
|
from crazy_functions.图片生成 import gen_image |
|
prompt_ = predict_no_ui_long_connection(inputs=story_paragraph, llm_kwargs=self.llm_kwargs, history=[], sys_prompt='你需要根据用户给出的小说段落,进行简短的环境描写。要求:80字以内。') |
|
image_url, image_path = gen_image(self.llm_kwargs, prompt_, '512x512', model="dall-e-2", quality='standard', style='natural') |
|
return f'<br/><div align="center"><img src="file={image_path}"></div>' |
|
except: |
|
return '' |
|
|
|
def step(self, prompt, chatbot, history): |
|
|
|
""" |
|
首先,处理游戏初始化等特殊情况 |
|
""" |
|
if self.step_cnt == 0: |
|
self.begin_game_step_0(prompt, chatbot, history) |
|
self.lock_plugin(chatbot) |
|
self.cur_task = 'head_start' |
|
else: |
|
if prompt.strip() == 'exit' or prompt.strip() == '结束剧情': |
|
|
|
self.delete_game = True |
|
yield from update_ui_lastest_msg(lastmsg=f"游戏结束。", chatbot=chatbot, history=history, delay=0.) |
|
return |
|
if '剧情收尾' in prompt: |
|
self.cur_task = 'story_terminate' |
|
|
|
|
|
|
|
yield from update_ui(chatbot=chatbot, history=history) |
|
|
|
|
|
""" |
|
处理游戏的主体逻辑 |
|
""" |
|
if self.cur_task == 'head_start': |
|
""" |
|
这是游戏的第一步 |
|
""" |
|
inputs_ = prompts_hs.format(headstart=self.headstart) |
|
history_ = [] |
|
story_paragraph = yield from request_gpt_model_in_new_thread_with_ui_alive( |
|
inputs_, '故事开头', self.llm_kwargs, |
|
chatbot, history_, self.sys_prompt_ |
|
) |
|
self.story.append(story_paragraph) |
|
|
|
yield from update_ui_lastest_msg(lastmsg=story_paragraph + '<br/>正在生成插图中 ...', chatbot=chatbot, history=history, delay=0.) |
|
yield from update_ui_lastest_msg(lastmsg=story_paragraph + '<br/>'+ self.generate_story_image(story_paragraph), chatbot=chatbot, history=history, delay=0.) |
|
|
|
|
|
previously_on_story = "" |
|
for s in self.story: |
|
previously_on_story += s + '\n' |
|
inputs_ = prompts_interact.format(previously_on_story=previously_on_story) |
|
history_ = [] |
|
self.next_choices = yield from request_gpt_model_in_new_thread_with_ui_alive( |
|
inputs_, '请在以下几种故事走向中,选择一种(当然,您也可以选择给出其他故事走向):', self.llm_kwargs, |
|
chatbot, |
|
history_, |
|
self.sys_prompt_ |
|
) |
|
self.cur_task = 'user_choice' |
|
|
|
|
|
elif self.cur_task == 'user_choice': |
|
""" |
|
根据用户的提示,确定故事的下一步 |
|
""" |
|
if '请在以下几种故事走向中,选择一种' in chatbot[-1][0]: chatbot.pop(-1) |
|
previously_on_story = "" |
|
for s in self.story: |
|
previously_on_story += s + '\n' |
|
inputs_ = prompts_resume.format(previously_on_story=previously_on_story, choice=self.next_choices, user_choice=prompt) |
|
history_ = [] |
|
story_paragraph = yield from request_gpt_model_in_new_thread_with_ui_alive( |
|
inputs_, f'下一段故事(您的选择是:{prompt})。', self.llm_kwargs, |
|
chatbot, history_, self.sys_prompt_ |
|
) |
|
self.story.append(story_paragraph) |
|
|
|
yield from update_ui_lastest_msg(lastmsg=story_paragraph + '<br/>正在生成插图中 ...', chatbot=chatbot, history=history, delay=0.) |
|
yield from update_ui_lastest_msg(lastmsg=story_paragraph + '<br/>'+ self.generate_story_image(story_paragraph), chatbot=chatbot, history=history, delay=0.) |
|
|
|
|
|
previously_on_story = "" |
|
for s in self.story: |
|
previously_on_story += s + '\n' |
|
inputs_ = prompts_interact.format(previously_on_story=previously_on_story) |
|
history_ = [] |
|
self.next_choices = yield from request_gpt_model_in_new_thread_with_ui_alive( |
|
inputs_, |
|
'请在以下几种故事走向中,选择一种。当然,您也可以给出您心中的其他故事走向。另外,如果您希望剧情立即收尾,请输入剧情走向,并以“剧情收尾”四个字提示程序。', self.llm_kwargs, |
|
chatbot, |
|
history_, |
|
self.sys_prompt_ |
|
) |
|
self.cur_task = 'user_choice' |
|
|
|
|
|
elif self.cur_task == 'story_terminate': |
|
""" |
|
根据用户的提示,确定故事的结局 |
|
""" |
|
previously_on_story = "" |
|
for s in self.story: |
|
previously_on_story += s + '\n' |
|
inputs_ = prompts_terminate.format(previously_on_story=previously_on_story, user_choice=prompt) |
|
history_ = [] |
|
story_paragraph = yield from request_gpt_model_in_new_thread_with_ui_alive( |
|
inputs_, f'故事收尾(您的选择是:{prompt})。', self.llm_kwargs, |
|
chatbot, history_, self.sys_prompt_ |
|
) |
|
|
|
yield from update_ui_lastest_msg(lastmsg=story_paragraph + '<br/>正在生成插图中 ...', chatbot=chatbot, history=history, delay=0.) |
|
yield from update_ui_lastest_msg(lastmsg=story_paragraph + '<br/>'+ self.generate_story_image(story_paragraph), chatbot=chatbot, history=history, delay=0.) |
|
|
|
|
|
self.delete_game = True |
|
return |
|
|