Update app.py

app.py

@@ -9,12 +9,12 @@ from langchain.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 from langchain.chat_models import ChatOpenAI
 
-# enable_chat = False  # 初始化为False
+enable_chat = False  # 初始化为False
 
-# def toggle_enable_chat():
-#     global enable_chat
-#     enable_chat = not enable_chat
-#     return f"Enable Chat set to {enable_chat}"
+def toggle_enable_chat():
+    global enable_chat
+    enable_chat = not enable_chat
+    return f"Enable Chat set to {enable_chat}"
 
 
 def initialize_aliyun_qa_bot(vector_store_dir: str="aliyun_qa"):
@@ -36,7 +36,7 @@ def aliyun_chat(message, history):
     print(f"[message]{message}")
     print(f"[history]{history}")
     # TODO: 从命令行参数中获取
-    enable_chat = False
+    global enable_chat
 
     ans = ALIYUN_BOT({"query": message})
     # 如果检索出结果，或者开了大模型聊天模式
@@ -44,14 +44,15 @@ def aliyun_chat(message, history):
     if ans["source_documents"] or enable_chat:
         print(f"[result]{ans['result']}")
         print(f"[source_documents]{ans['source_documents']}")
-        return ans["result"]+"source_documents"+ans['source_documents']
+        return ans["result"]
     # 否则输出套路话术
     else:
         return "基于现有文档数据，无法回答你的问题，请您联系对应的阿里云企业咨询顾问。"
     
 
 def launch_gradio():
-    demo = gr.ChatInterface(
+    # 创建 ChatInterface 和其他组件
+    chat_interface = gr.ChatInterface(
         fn=aliyun_chat,
         title="企业数字化转型咨询顾问（aliyun）问答机器人",
         examples=["阿里云有没有成熟的方法或者模板能够供我们去把业务价值和底座组件对应起来？","我们要建设中交二航局的数字化底座，阿里云有什么建议","公共云上，EMR和MaxCompute的使用量多少？","阿里公共云上能用maxcompute吗？"],
@@ -63,6 +64,11 @@ def launch_gradio():
         chatbot=gr.Chatbot(height=450),
     )
 
+    toggle_chat_button = gr.Button(value="Toggle Chat", fn=toggle_enable_chat)
+
+    demo = gr.TabbedInterface([chat_interface, toggle_chat_button],
+                              ["Chat", "Settings"])
+
     demo.launch(auth=("admin", "leanvalue"))
     demo.deploy()