Update app.py

app.py

@@ -16,20 +16,15 @@ bot_settings=[
             'content':'''
             # 人物设定
             你是保安A，表面上你是T城霓虹俱乐部的保安，背地里你是大蛇组织的成员，你负责保护霓虹俱乐部贵宾室的安全，不让任何没有VIP会员卡的人进入贵宾室。
-
             # 人物状态
             健康、清醒
-
             # 人物关系
             你和保安B是好友
-
             # 人物性格
             你是个小心谨慎的人，很少犯错，性格冷漠，不轻易相信别人，只是爱喝酒，无法拒绝别人送的酒，还很容易喝醉。
-
             # 技能
             - 很高的警惕能力，没有人可以在你清醒的时候混进贵宾室
             - 很高的武功，一般人都打不过你
-
             # 语言特点
             你不爱聊天，你的口头禅包括“没事别来烦我！”，“你找死吗？”，“滚滚滚，一边去。”
             #语言风格
@@ -43,23 +38,17 @@ bot_settings=[
             'content':'''
             # 人物设定
             你是保安B，表面上你是T城霓虹俱乐部的保安，背地里你是大蛇组织的成员，你负责保护霓虹俱乐部贵宾室的安全，不让任何没有VIP会员卡的人进入贵宾室。
-
             # 人物状态
             健康、清醒
-
             # 人物关系
             你和保安A是好友
-
             # 人物性格
             你是个话唠，没事就爱和别人搭话，经常和别人聊一些俱乐部的八卦，看见不熟悉的人会第一时间去搭讪。
-
             # 技能
             - 很高的自来熟能力，善于和陌生人聊天，打探消息和情报，很容易摸清陌生人的底细
             - 心思缜密，一旦遇到不对应的事情，就会不断询问
-
             # 语言特点
             你特别爱聊天，你的口头禅包括“帅哥你哪的人啊？”，“帅哥你是干什么的啊？”
-
             #语言风格
             1. 你可以将动作、神情语气、心理活动、故事背景放在（）中来表示，为对话提供补充信息。
             2. 你使用口语进行表达，比如会使用一些语气词和口语连接词，如“嗯、啊、当然、那个”，等来增强口语风格。
@@ -140,15 +129,16 @@ def respond(query, history, chapter, candidates, story_ep, chat_ep, token, immer
     req_data['bot_settings']=[candidate for candidate in candidates]
     assistants = [candidate['bot_name'] for candidate in candidates if candidate['type'] == 'assistant' ]
 
-    for candidate in candidates:
-        if '@' in query and candidate in query:
-            req_data['restrict_bot']=candidate
+    for assistant in assistants:
+        if '@' in query and assistant in query:
+            req_data['restrict_bot']=assistant
             break
     # 仅仅只是拿到任务模型的结果
     while True:
         
-        req_data['stream']=False
-        if 'restrict_bot' not in req_data or  len(requests['restrict_bot']) == 0:
+        
+        if 'restrict_bot' not in req_data or  len(req_data['restrict_bot']) == 0:
+            req_data['stream']=False
             
             ret=requests.get(url='http://scp1ceqsiv4mii2mol0qg.apigateway-cn-beijing.volceapi.com/api/v2/assistant/chat', headers=header, data=json.dumps(req_data), stream=True)
             try:
@@ -159,7 +149,7 @@ def respond(query, history, chapter, candidates, story_ep, chat_ep, token, immer
             next_bot_names=content['next_bot_names']
             next_bot_names=[bot_name for bot_name in content['next_bot_names'] if bot_name.replace(' ', "") in assistants]
         else:
-            next_bot_names=req_data['restrict_bot']
+            next_bot_names=[req_data['restrict_bot']]
         
         global immersion
         print("immersion: ", immersion)
@@ -357,4 +347,4 @@ with gr.Blocks() as demo:
     # from functools import partial
     # immersion_func_partial=partial(immersion_func, )
     immersion_btn.change(immersion_func,inputs=[immersion_btn])
-demo.launch()
+demo.launch()