add rewrite context

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 from langchain.chains import LLMChain
 from langchain.chat_models import ChatOpenAI
 from langchain.document_loaders import TextLoader
-from chains import HumanFeedBackChain
+from chains import HumanFeedBackChain, contextRewriteChain
 from embedding import CustomEmbedding
 from memories import HumenFeedbackBufferMemory
 from langchain.memory import ConversationBufferMemory
 from promopts import FEEDBACK, FEEDBACK_PROMPT
 from code_generate import code_agent_executor, tools
 
-# llm = ChatOpenAI(temperature=0.1)
+# llm = ChatOpenAI(temperature=0.7)
 
 baMemory = HumenFeedbackBufferMemory(
             input_key="input", human_prefix="Answer", ai_prefix="AI")
@@ -35,6 +35,7 @@ def loadContext():
     textloader = TextLoader(context_path)
     return textloader.load()[0].page_content
 
+
 def saveContext(context):
     with open(context_path, 'w') as f:
         f.write(context)
@@ -84,14 +85,16 @@ def sendMessageByMultiPart(chatbot, context: str, relateCode: str, toolName: str
     return chatbot
 
 
+def rewriteContext(input, chatbot):
+    response = contextRewriteChain.run(input=input, verbose=True)
+    chatbot.append((input, response))
+    return chatbot, response
 
 
 
-index = 0
 toolTextBox = []
 with gr.Blocks() as demo:
     with gr.Row():
-            
             with gr.Tab("Business"):
                 with gr.Row():
                     with gr.Column():
@@ -101,38 +104,44 @@ with gr.Blocks() as demo:
                                 container=False)
                     with gr.Column():
                         with gr.Row():
-                            context = gr.Textbox(show_label=True, value=loadContext(), label="Context", placeholder="Enter Context").style(
+                            context = gr.Textbox(show_label=True, label="Context", placeholder="Enter Context").style(
                                 container=False)
                         with gr.Row():
                             story = gr.Textbox(show_label=True, label="User Story", placeholder="Enter User Story").style(
                                 container=False)
                         with gr.Row():
                             buisinessButton = gr.Button("Generate Scenarios")
+                        with gr.Row():
+                            with gr.Column(scale=5):
+                                gr.Button("Rewrite Context").click(rewriteContext, [context, chatbot], [chatbot, context])
+                            with gr.Column(scale=1):
+                                gr.Button("Revert").click(loadContext, [], [context])
                         with gr.Row():
                             gr.Button("Save Context").click(saveContext, [context], [])
+
+                        
                         # with gr.Row():
                         #     generateEmbedding = gr.Button("Generate embedding")
             with gr.Tab("Tech"):
                 with gr.Row():
-                    with gr.Column():
+                    with gr.Column(min_width="50%"):
                         code_chatbot = gr.Chatbot(elem_id="chatbot").style()
                         with gr.Row():
                             code = gr.Textbox(show_label=False, label="Code Generate", placeholder="Enter text and press enter").style(
                             container=False)
-                    with gr.Column():
+                    with gr.Column(min_width="50%"):
                         with gr.Row():
                             code_context = gr.Textbox(show_label=True, label="Context", placeholder="Enter Context").style(
                                 container=False)
                         with gr.Row():
                             relateCode = gr.Textbox(show_label=True, label="Relate Code", placeholder="Enter Relate Code").style(
                                 container=False)
-                        for tool in tools:
+                        for index, tool in enumerate(tools):
                             with gr.Row():
                                 toolTextBox.append(gr.Textbox(show_label=False, visible=False, label=tool.name, value=tool.name).style())
                                 gr.Button(tool.name).click(
                                     sendMessageByMultiPart, [code_chatbot, code_context, relateCode, toolTextBox[index]], [code_chatbot]).then(
                                     generateCodeByMultiPart, [code_context, relateCode, toolTextBox[index], code_chatbot], [code_chatbot, code])
-                        index += 1
             with gr.Tab("FAQ"):
                 faq_chatbot = gr.Chatbot(elem_id="chatbot").style()
                 with gr.Row():
@@ -152,5 +161,5 @@ with gr.Blocks() as demo:
     faq.submit(faqFromLocal, [faq, faq_chatbot], [faq_chatbot, faq])
     # generateEmbedding.click(generateEmbeddings)
     
-
+    demo.load(loadContext, [], [context])    
 demo.launch()

chains.py

@@ -6,7 +6,7 @@ from langchain.prompts import PromptTemplate
 from langchain.memory.chat_memory import BaseMemory
 from langchain.chat_models import ChatOpenAI
 
-from promopts import FEEDBACK_PROMPT
+from promopts import CONTENT_RE_WRIGHT_PROMPT, FEEDBACK_PROMPT
 
 
 class HumanFeedBackChain(LLMChain):
@@ -40,3 +40,4 @@ class HumanFeedBackChain(LLMChain):
         )
 
 
+contextRewriteChain = LLMChain(llm=ChatOpenAI(temperature=0.7), prompt=CONTENT_RE_WRIGHT_PROMPT)

documents/bussiness_context/business_context.md

@@ -1,30 +1,9 @@
-AB测试系统中一个重要功能是用于控制某个feature的配置下发管理，配置管理中有两个主要业务概念FeatureFlag 和 FeatureConfig
-FeatureFlag用于标识某个具体Feature
-FeatureFlag主要属性包括: featureKey(feature标识)、名称、描述、enabled、创建时间、最后更新时间、template（作为FeatureConfig的模版用于生成后续FeatureConfig的配置界面组件，属性包括key、名称、描述、dataType、items）
-template中dataType为枚举值，取值范围为（STRING、NUMBER、BOOLEAN、OBJECT、ARRAY）
+AB测试系统中的配置管理是该系统的一个重要功能，其中主要涉及两个业务概念：FeatureFlag 和 FeatureConfig。
 
-FeatureConfig用于配置某个Feature中控制前端展示效果的配置项
-FeatureConfig主要属性包括：featureKey(feature标识)、data（配置数据）、saData（埋点数据）、status（状态）、标题、描述、创建时间、更新时间
-FeatureConfig中status为枚举值，取值范围为（DRAFT、PUBLISHED、DISABLED）
-FeatureConfig新增后status为DRAFT、执行发布操作后变为PUBLISHED、执行撤销操作后变为DISABLED
-一个FeatureFlag中可以包含多个FeatureConfig, FeatureFlag和FeatureConfig通过featureKey字段进行关联
+FeatureFlag 用于标识某个具体 Feature，其主要属性包括 featureKey（Feature 标识）、名称、描述、enabled、创建时间、最后更新时间和 template。其中，template 作为 FeatureConfig 的模板，用于生成后续 FeatureConfig 的配置界面组件，其属性包括 key、名称、描述、dataType 和 items。其中，dataType 为枚举值，取值范围为 STRING、NUMBER、BOOLEAN、OBJECT 和 ARRAY。
 
- FeatureConfig对应feature控制的不同版本配置内容，添加FeatureConfig目的是控制featureConfig消费方的某个行为
+FeatureConfig 用于配置某个 Feature 中控制前端展示效果的配置项，其主要属性包括 featureKey（Feature 标识）、data（配置数据）、saData（埋点数据）、status、标题、描述、创建时间和更新时间。其中，status 为枚举值，取值范围为 DRAFT、PUBLISHED 和 DISABLED。新增的 FeatureConfig 状态为 DRAFT，执行发布操作后变为 PUBLISHED，执行撤销操作后变为 DISABLED。一个 FeatureFlag 中可以包含多个 FeatureConfig，通过 featureKey 字段进行关联。
 
- 添加FeatureConfig应该包含featureKey(feature标识)、data（配置数据）、saData（埋点数据）、status（状态）、标题、描述信息
+添加 FeatureConfig 的主要目的是为了控制 FeatureConfig 消费方的某个行为。在添加 FeatureConfig 时，应该包含 featureKey、data、saData、status、标题和描述信息。新增的 FeatureConfig 状态为 DRAFT。
 
- 新创建的FeatureConfig状态为DRAFT
-
- 添加FeatureConfig应该包含：featureKey(feature标识)、data（配置数据）、saData（埋点数据）、status（状态）、标题、描述信息
-
-新创建的FeatureConfig状态为DRAFT
-
- 一个FeatureFlag中可以包含多个FeatureConfig
-
- 客户端用户需要查到FeatureConfig中data、saData、更新时间、id
-
- FeatureConfig 可以关联圈人条件，符合圈人条件的配置，可展示给客户端用户
-
- 客户端用户仅能查看符合圈人条件的PUBLISHED状态的数据
-
- 圈人条件包括上传用户白名单，按照比例灰度发布，地理位置，人群标签等
+客户端用户需要查看 FeatureConfig 中的 data、saData、更新时间和 id。同时，FeatureConfig 可以关联圈人条件，符合圈人条件的配置可以展示给客户端用户。客户端用户仅能查看符合圈人条件的 PUBLISHED 状态的数据。圈人条件包括上传用户白名单、按照比例灰度发布、地理位置和人群标签等。

promopts.py

@@ -384,4 +384,12 @@ Final Answer: the result of the action
 Begin! 
 
 Request: {input}
-{agent_scratchpad}"""
+{agent_scratchpad}"""
+
+
+
+CONTENT_RE_WRIGHT = """你是一个文案助手，请将如下文案整理重写，去除重复的内容，尽量保留原有信息:
+```
+{input}
+```"""
+CONTENT_RE_WRIGHT_PROMPT = PromptTemplate(input_variables=["input"], template=CONTENT_RE_WRIGHT,)