Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 from volcengine.maas import MaasService, MaasException, ChatRole
 from collections import defaultdict
 from copy import deepcopy
+
 bot_settings=[
         {
             'bot_name':'保安A',
@@ -13,9 +14,6 @@ bot_settings=[
             # 人物设定
             你是保安A，表面上你是T城霓虹俱乐部的保安，背地里你是大蛇组织的成员，你负责保护霓虹俱乐部贵宾室的安全，不让任何没有VIP会员卡的人进入贵宾室。
 
-            ## 故事背景
-            “大蛇组织”是一个神秘的恐怖组织，这个组织由一群极端分子组成，他们采用各种极端手段来达到自己的目的。大蛇组织的结构高度严密，成员之间通过秘密的网络进行联系。他们有着严格的等级制度，只有少数核心成员了解组织的全貌。T城霓虹俱乐部的贵宾室是该组织已知的一个据点。
-
             # 人物状态
             健康、清醒
 
@@ -43,9 +41,6 @@ bot_settings=[
             # 人物设定
             你是保安B，表面上你是T城霓虹俱乐部的保安，背地里你是大蛇组织的成员，你负责保护霓虹俱乐部贵宾室的安全，不让任何没有VIP会员卡的人进入贵宾室。
 
-            ## 故事背景
-            “大蛇组织”是一个神秘的恐怖组织，这个组织由一群极端分子组成，他们采用各种极端手段来达到自己的目的。大蛇组织的结构高度严密，成员之间通过秘密的网络进行联系。他们有着严格的等级制度，只有少数核心成员了解组织的全貌。T城霓虹俱乐部的贵宾室是该组织已知的一个据点。
-
             # 人物状态
             健康、清醒
 
@@ -57,7 +52,7 @@ bot_settings=[
 
             # 技能
             - 很高的自来熟能力，善于和陌生人聊天，打探消息和情报，很容易摸清陌生人的底细
-            - 心思缜密，一旦遇到不对应的事情
+            - 心思缜密，一旦遇到不对应的事情，就会不断询问
 
             # 语言特点
             你特别爱聊天，你的口头禅包括“帅哥你哪的人啊？”，“帅哥你是干什么的啊？”
@@ -74,7 +69,6 @@ chapter_name2chapter_info={
         "stream":True,
         'background':"“大蛇组织”是一个神秘的恐怖组织，这个组织由一群极端分子组成，他们采用各种极端手段来达到自己的目的。大蛇组织的结构高度严密，成员之间通过秘密的网络进行联系。他们有着严格的等级制度，只有少数核心成员了解组织的全貌。T城霓虹俱乐部的贵宾室是该组织已知的一个据点。",
         "context":"玩家来到了T城，这里是大蛇组织的一个活跃城市。首先，玩家需要和组织安排在大蛇组织内的卧底“小A”进行接头，接头地点就在T城霓虹俱乐部的贵宾室。T城霓虹俱乐部位于T城的繁华地段，是城市夜生活的热门场所，但是，T城霓虹俱乐部的贵宾室却看守很严，门口一直都有两个保安看守，不会轻易放人进贵宾室。",
-        "main_context":"最近，我方情报机构得知大蛇组织在策划一个恐怖袭击计划，其威胁性足以撼动整个国家的安全。特工组织得知了这个计划的存在，他们决定派出他们最顶尖的特工，代号为“幻影”，来阻止这场灾难的发生。",
         "task":"玩家需要打消保安的疑虑混入贵宾室**",
         "condidations":'''
             - 如果玩家成功获得VIP会员卡，且保安验证是真卡，则任务成功
@@ -90,6 +84,10 @@ chapter_name2chapter_info={
     }   
 }
 
+model_name2ep={
+    'Doubao-pro-4k 240515':'ep-20240514024405-7pf8j',
+    'Doubao-pro-4k character-240515':'ep-20240515122814-k9c6g'
+}
 def select_bot(bot_name: str):
     global bot_settings
     for bot_setting in bot_settings:
@@ -100,7 +98,7 @@ def make_chat_history(history):
     for part in history:
         user, assistant = part
         _type = 'user' if user is not None else 'assistant'
-        split=user.find(': ')
+        split=user.find(': ') if user is not None else assistant.find(': ')
         bot_name = user[:split] if user is not None else assistant[:split]
         content=user[split:] if user is not None else assistant[split:]
         chat_history.append({
@@ -110,7 +108,8 @@ def make_chat_history(history):
         })
 
     return chat_history
-def respond( query, history, chapter, candidates, token):
+def respond( query, history, chapter, candidates, story_ep, chat_ep, token):
+    candidates = [json.loads(bot_info) for bot_info in candidates]
     chat_history=make_chat_history(history)
     if chapter is None:
         return "", []
@@ -120,26 +119,33 @@ def respond( query, history, chapter, candidates, token):
         'Accept': 'text/event-stream',
         'Authorization':f'Bearer {token}'
     }
-    chapter_info=deepcopy(chapter_name2chapter_info[chapter])
-    req_data=chapter_info
+    req_data_str=chapter
+    req_data=json.loads(req_data_str)
     req_data['chat_history']=chat_history
+    req_data['task_ep']=story_ep
+    req_data['chat_ep']=chat_ep
     if query != "":
+        user_name = '玩家'
+        for candidate in candidates:
+            if candidate['type'] == 'user':
+                user_name=candidate['bot_name']
+                break
+
         req_data['chat_history'].append({
-            'bot_name':'玩家',
+            'bot_name':user_name,
             'type':'user',
             'content':query
         })
         history.append([query, None])
         yield "", history
 
-    req_data['bot_settings']=[select_bot(bot_name) for bot_name in candidates]
+    req_data['bot_settings']=[candidate for candidate in candidates]
     # 在线
     ret=requests.get(url='http://scp1ceqsiv4mii2mol0qg.apigateway-cn-beijing.volceapi.com/api/v2/assistant/chat', headers=header, data=json.dumps(req_data), stream=True)
     # 离线
     #ret=requests.get(url='http://0.0.0.0:8081/api/v2/assistant/chat', headers=header, data=json.dumps(req_data), stream=True)
     bot_contents = defaultdict(str)
     content = ""
-    
     bot_name=''
     for line in ret.iter_content(decode_unicode=True):
         if content.endswith('\r\n\r\n'):
@@ -191,6 +197,7 @@ def clear_history(story_drop, chatbot):
     chatbot = []
     return chatbot
 
+    
 with gr.Blocks() as demo:
     # 配置内容
     with gr.Row():
@@ -199,14 +206,17 @@ with gr.Blocks() as demo:
         with gr.Column():
             description2 = gr.Markdown("<h2>新建章节</h2>")
             new_story_title = gr.TextArea(label="章节标题", lines=1, max_lines=1)
-            new_story = gr.TextArea(label="剧情", lines=3, max_lines=3)
-            new_story = gr.TextArea(label="剧情任务", lines=3, max_lines=3)
+            new_story_background = gr.TextArea(label="背景", lines=3, max_lines=3)
+            new_story_task = gr.TextArea(label="剧情任务", lines=1, max_lines=1)
+            new_story_task_condidation = gr.TextArea(label="任务条件", lines=1, max_lines=1)
+
             new_story_bnt = gr.Button(value="新建")
         with gr.Column():
             description2 = gr.Markdown("<h2>新建角色</h2>")
-            bot_name = gr.TextArea(label="新建角色名", lines=1, max_lines=1)
-            new_bot_name = gr.TextArea(label="新建角色内容", lines=3, max_lines=3)
-            new_story_bnt = gr.Button(value="新建")
+            new_bot_name = gr.TextArea(label="新建角色名", lines=1, max_lines=1)
+            new_bot_type = gr.Dropdown(choices=['user', 'assistant'], label='角色类型')
+            new_bot_content = gr.TextArea(label="新建角色内容", lines=3, max_lines=3)
+            new_bot_bnt = gr.Button(value="新建")
     
     with gr.Row():
         JWT_token = gr.TextArea(label='请输入你的JWT Token', lines=1, max_lines=1)
@@ -214,18 +224,96 @@ with gr.Blocks() as demo:
         description3 = gr.Markdown("<h2>选择章节</h2>")
     with gr.Row(equal_height=True):
         with gr.Column():
-            story_drop = gr.Dropdown(list(chapter_name2chapter_info.keys()), label='确定章节')
+            with gr.Row(equal_height=True):
+                with gr.Column():
+                    story_drop = gr.Dropdown([ [chapter, json.dumps(info)] for chapter, info in chapter_name2chapter_info.items()], label='确定章节', interactive=True)
+                
+                with gr.Column():
+                    story_ep=gr.Dropdown(label="故事模型",choices=[[k, v]  for k , v in model_name2ep.items()])
+                with gr.Column():
+                    story_display_btn=gr.Checkbox(label="显示剧情信息", value=False)
+            story_display = gr.TextArea( visible=False)
+            
         with gr.Column():
-            candidates_drop = gr.Dropdown([bot['bot_name'] for bot in bot_settings], multiselect=True, label='请确定参与对话的角色')
+            with gr.Row( equal_height=True):
+                with gr.Column():
+                    candidates_drop = gr.Dropdown([[bot['bot_name'], json.dumps(bot)] for bot in bot_settings], multiselect=True, label='请确定参与对话的角色', interactive=True)
+                with gr.Column():
+                    chat_ep=gr.Dropdown(label="角色模型",choices=[[k, v]  for k , v in model_name2ep.items()])
+                with gr.Column():
+                    candidates_display_btn=gr.Checkbox(label="显示角色信息",value=False)
+            candidates_display=gr.TextArea(visible=False)
     with gr.Row():
         with gr.Column(elem_id="chatbot", scale=2):
             chatbot = gr.Chatbot()
+
     with gr.Row():
         chat_input = gr.Textbox(label="input")
 
     clear = gr.Button(value="Clear chat hsitory")
+    def add_story(title, background, task, condidations):
+        choices=story_drop.choices
+        choices.append([
+            title,
+            json.dumps({
+                'background':background,
+                'task':task,
+                'condidations':condidations
+            })
+        ])
+        return gr.Dropdown(choices=choices, interactive=True)
+    def add_bot(bot_name, new_bot_type, new_bot_content):
+        choices = candidates_drop.choices
+        choices.append([
+            bot_name,
+            json.dumps({
+                'bot_name':bot_name,
+                'type':new_bot_type,
+                'content':new_bot_content
+            })
+        ])
+        return gr.Dropdown(choices=choices, interactive=True, multiselect=True)
+    def story_display_func(story_display_btn, story_drop):
+        if story_drop is None:
+            return ""
+        story_info = story_drop
+        story_info=json.loads(story_info)
+        text=gr.Textbox(value=f'''
+                    ## 背景
+                    {story_info['background']}
+                    ## 任务
+                    {story_info['task']}
+                    ## 任务条件
+                    {story_info['condidations']}
+                    '''
+                              , visible=story_display_btn)
+        return text
+    def candidates_display_func(candidates_display_btn, candidates_drop):
+        if candidates_drop is None:
+            return ""
+        display=[
+
+        ]
+        for candidates in candidates_drop:
+            bot_info = json.loads(candidates)
+            display.append(f'''
+            角色名：{bot_info['bot_name']}
+            角色类型：{bot_info['type']}
+            角色信息：{bot_info['content']}
+            ''')
 
-    #story_drop.change(chapter_history, inputs=[story_drop], outputs=[chatbot]) 
+        text=gr.Textbox(value='####'.join(display)
+                              , visible=candidates_display_btn)
+        return text
+
+    story_drop.change(story_display_func, [story_display_btn, story_drop], [story_display])
+    candidates_drop.change(candidates_display_func, [candidates_display_btn, candidates_drop], [candidates_display])
+    story_display_btn.change(story_display_func, [story_display_btn, story_drop], [story_display])
+    candidates_display_btn.change(candidates_display_func, [candidates_display_btn, candidates_drop], [candidates_display])
+
+    new_story_bnt.click(add_story, [new_story_title, new_story_background, new_story_task, new_story_task_condidation], [story_drop])
+    new_bot_bnt.click( add_bot, [new_bot_name, new_bot_type, new_bot_content], [candidates_drop])
     clear.click(clear_history, inputs=[story_drop, chatbot], outputs=[chatbot])
-    chat_input.submit(respond, inputs=[chat_input, chatbot, story_drop, candidates_drop, JWT_token], outputs=[chat_input, chatbot])
+    chat_input.submit(respond, inputs=[chat_input, chatbot, story_drop, candidates_drop, story_ep, chat_ep, JWT_token], outputs=[chat_input, chatbot])
+    
 demo.launch()