Update app.py

app.py

@@ -13,11 +13,11 @@ import time
 os.environ["OPENAI_API_KEY"] = "sb-f6c206ba7ea4df761e227eba65e65bedad1e24e73b702967"
 os.environ['OPENAI_BASE_URL'] = "https://api.openai-sb.com/v1/"
 os.environ["SERPAPI_API_KEY"] = "dcc98b22d5f7d413979a175ff7d75b721c5992a3ee1e2363020b2bbdf4f82404"
-os.environ['TAVILY_API_KEY'] = "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"
+# os.environ['TAVILY_API_KEY'] = "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" #you
+os.environ['TAVILY_API_KEY'] = "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" #zeng
 
 llm = ChatOpenAI(model="gpt-4o", temperature=0.6)
 
-
 def search(query) -> str:
     params = {
         "engine": "baidu_news",
@@ -131,35 +131,45 @@ Thought: {agent_scratchpad}
 origin_prompt = ChatPromptTemplate.from_template(template)
 chain = origin_prompt | llm | StrOutputParser()
 
+selected_language = "ch"
+
+def SelectLanguage(option):
+    global selected_language
+    if option == "英文":
+        selected_language = "en"
+    else:
+        selected_language = "ch"
+language_option = ["中文","英文"]
 
-def react(input, language):
+def react(input):
     ori_result = chain.invoke(input)
     tools = [TavilySearchResults(max_result=1),search_tool]
-    if language == "en":
+    if selected_language == "en":
         prompt = ChatPromptTemplate.from_template(en_prompt)
-    elif language == "ch":
+    elif selected_language == "ch":
         prompt = ChatPromptTemplate.from_template(ch_prompt)
     agent = create_react_agent(llm, tools, prompt)
     agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, handle_parsing_errors=True)
     result = agent_executor.invoke({"input": input})
     return result['output'], ori_result
+
+
 title = "# 虚假信息检测"
 
 with gr.Blocks(css=css1) as demo:
-    # gr.Markdown(
-    #     "<h1 style='font-size: 36px; text-align: center; color: #333333; margin-bottom: 20px;'>虚假信息检测</h1>")
     gr.Markdown(title, elem_id="title")
     with gr.Row():
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(elem_classes="gradio-output")
-            input_box = gr.Textbox(label="新闻", elem_classes="gradio-input", placeholder="请输入新闻", lines=3)
-            language_box = gr.Textbox(label="语言", elem_classes="gradio-input", placeholder="请输入要使用的语言")
+            language_select = gr.Dropdown(choices= language_option, elem_classes="gradio-input", label= "请选择要使用的语言")
+            input_box = gr.Textbox(label="输入", elem_classes="gradio-input", placeholder="请输入要判断的新闻", lines=3)
+
         with gr.Column(scale=1):
             ans_box = gr.Textbox(label="gpt-4o", lines=5, elem_classes="gradio-output")
-            react_box = gr.Textbox(label="React", lines=5, elem_classes="gradio-output")
+            react_box = gr.Textbox(label="React", lines=5, elem_classes="gradio-output", visible=False)
             clear = gr.Button("清空页面", elem_classes="gradio-button")
             submit_btn = gr.Button("提交", elem_classes="gradio-button")
-
+        language_select.change(SelectLanguage, language_select)
 
     def user(user_input, history):
         if history is None:
@@ -177,9 +187,7 @@ with gr.Blocks(css=css1) as demo:
             time.sleep(0.01)
             yield history
 
-
-
-    submit_btn.click(react, [input_box, language_box], [react_box, ans_box]
+    submit_btn.click(react, [input_box], [react_box, ans_box]
     ).then(
         user, [input_box, chatbot], [input_box, chatbot]
     ).then(
@@ -189,4 +197,3 @@ with gr.Blocks(css=css1) as demo:
 
 if __name__ == "__main__":
     demo.launch()
-