Update app.py

app.py

@@ -13,7 +13,6 @@ from model import *
 from gradio import ChatMessage, on
 from tools import *
 from prompt import *
-from config import *
 import sys
 from gradio.components.chatbot import MessageDict
 from langchain_community.llms import Tongyi
@@ -176,7 +175,7 @@ class FactChecker:
             if len(action_input) >= 3:
                 # agent_thought.append({"title": f"工具调用:{action}", "content": action_input})
                 web_search_tools = {"BoCha Webs Search","Image Search","Baidu News Search","tavily_search_results_json"}
-                if action in web_search_tools:
+                if action in web_search_tools or action == "":
                     action = "网页搜索"
                 agent_thought.append(
                     ChatMessage(
@@ -468,7 +467,7 @@ class History:
 def build_interface() -> gr.Blocks:
     with gr.Blocks(css=css, theme='soft') as demo:
         gr.HTML(
-            "<h1 style='font-size: 36px; text-align: center; color: #333333; margin-bottom: 20px;'>虚假信息检测系统</h1>")
+            "<h1 style='font-size: 36px; text-align: center; color: #333333; margin-bottom: 20px;'>惊堂木-虚假信息检测系统</h1>")
         with gr.Tab(label='Chat'):
             with gr.Row():
                 with gr.Sidebar():
@@ -491,9 +490,9 @@ def build_interface() -> gr.Blocks:
                         )
 
                         language_select = gr.Dropdown(["中文", "英文"], label="请选择要使用的语言", scale=1, value="中文")
-                        model_select = gr.Dropdown(["自研", "llama3-70b", "llama3-8b", "mistral"],
-                                                   label="请选择要使用的大模型",
-                                                   scale=1, value="自研")
+                        # model_select = gr.Dropdown(["自研", "llama3-70b", "llama3-8b", "mistral"],
+                        #                            label="请选择要使用的大模型",
+                        #                            scale=1, value="自研")
                         # mode_select = gr.Dropdown(["事实核查","谣言检测"],label= "请选择使用的模式",scale= 1, value= "谣言检测")
 
                 with gr.Column(scale=3):
@@ -505,7 +504,7 @@ def build_interface() -> gr.Blocks:
                             {
                                 "text": "据最新天文研究，火星的轨道将逐渐接近地球，最终成为地球的“第二月亮”。天文学家预测这一变化将在2025年发生，届时火星将在夜空中与月亮一样明亮，影响全球潮汐和生态平衡。"}
                         ],
-                        chatbot=gr.Chatbot(label='自研系统',
+                        chatbot=gr.Chatbot(label='惊堂木',
                                            avatar_images=("./image/user.png", "./image/logo.png"),
                                            type="messages",
                                            height=600,
@@ -522,8 +521,8 @@ def build_interface() -> gr.Blocks:
                     )
 
                 with gr.Column(scale=1):
-                    with gr.Accordion("图片检测", open=False):
-                        img_info = gr.Textbox(label="检测结果", lines=5)
+                    with gr.Accordion("图片分析", open=False):
+                        img_info = gr.Textbox(label="分析结果", lines=5)
                     # contrast_model = gr.Textbox(label="对比模型", lines=5)
                     # with gr.Accordion("展示对比模型prompt", open=False):
                     #     model_prompt = gr.Textbox(label="对比模型prompt", lines=5, placeholder=constract_model_prompt, interactive=True)
@@ -531,10 +530,10 @@ def build_interface() -> gr.Blocks:
                     #                                label="请选择使用的对比模型", scale=1, value="qwen")
                     # contrast_bn = gr.Button("展示对比模型")
                     # prompt_bn = gr.Button("确定更改提示词")
-                    img_bn = gr.Button("显示检测结果")
+                    img_bn = gr.Button("显示分析结果")
             img_bn.click(factChecker.update_image_summaries, [], img_info)
             language_select.change(factChecker.SelectLanguage, language_select, [])
-            model_select.change(factChecker.SelectModel, model_select, [])
+            # model_select.change(factChecker.SelectModel, model_select, [])
             # mode_select.change(factChecker.SelectMode, mode_select, [])
             # contrast_select.change(factChecker.SelectContrast, contrast_select, [])
             # contrast_bn.click(factChecker.contract, [], contrast_model)