重构代码

app.py

@@ -28,7 +28,7 @@ with gr.Blocks() as demo:
             ).style(container=False)
 
             image_button = gr.Button("点击生图")
-        image_button.click(None, inputs=txt_img, outputs=image_out)  # 生图函数待填充 todo
+        image_button.click(None, inputs=txt_img, outputs=image_out)  # todo 生图函数待填充
 
 
 demo.queue()

chat_poets/chat.py

@@ -12,9 +12,19 @@ class ChatPoet:
     @classmethod
     def allow_chat(cls, user_message: str) -> bool:
         """
-        对话开始时判断用户输入是否合规：提及诗人或古诗
+        对话开始时判断用户输入是否合规：提及诗人或古诗 todo 需要进一步优化提示词
         """
-        return user_message == "y"
+        prompt_allow = load_prompt("chat_poets/prompts/allow_chat.json")
+        chain_allow = LLMChain(llm=cls.llm, prompt=prompt_allow)
+        response = chain_allow.run(user_message)
+        print(f"response: {response}")
+
+        while True:
+            try:
+                allow = int(response)
+                return allow == 1
+            except ValueError:
+                response = chain_allow.run(user_message)
 
     @classmethod
     def gen_response(cls, pattern: str, history: list[list]) -> str:
@@ -24,4 +34,3 @@ class ChatPoet:
         注：对话历史的最后一项是需要填充的内容，即history[-1] = [Question, None]；Question为用户刚提出的问题，None即尚未回答
         """
         return "哇哇哇!"
-

chat_poets/prompts/allow_chat.json

@@ -0,0 +1,5 @@
+{
+  "_type": "prompt",
+  "input_variables": ["user_message"],
+  "template": "请分析问题中是否提到了诗人或古诗。如果是则输出1，否则输出0；不输出除结果外的其他内容。问题如下->{user_message}"
+}

chat_poets/prompts/example1.json

@@ -1,5 +0,0 @@
-{
-  "_type": "prompt",
-  "input_variables": ["text"],
-  "template": "阿巴{text}"
-}

chat_poets/prompts/example2.json

@@ -1,5 +0,0 @@
-{
-  "_type": "prompt",
-  "input_variables": ["text"],
-  "template": "阿巴阿巴{text}"
-}

gradio_ui/gr_chat.py

@@ -15,7 +15,7 @@ def chat_poetry(tab: gr.Tab):
     pattern = gr.Markdown(f"{tab.id}")
     action = gr.State(value=False)  # 指示有效对话是否开始（开始时用户未提及古诗诗人则未进入）
     with gr.Row():
-        # 历史记录 todo
+        # todo 历史记录
         with gr.Column(scale=2):
             gr.Markdown("## 历史记录")
             chat_history = gr.State([[]])  # 存储单次聊天记录的组件
@@ -54,7 +54,7 @@ def newchat(chat_history: list[list]):
     print("#########")
     for chat_list in chat_history:
         print(f"##用户：{chat_list[0]} ## Bot：{chat_list[1]}")
-    # 将已有聊天记录存入历史记录 todo
+    # todo 将已有聊天记录存入历史记录
 
     # 刷新聊天记录
     return [None, None, None]