Update bin_public/app/Chatbot.py

bin_public/app/Chatbot.py

@@ -153,6 +153,31 @@ with gr.Blocks(css=customCSS) as demo:
                                 with gr.Column():
                                     downloadFile = gr.File(interactive=True)
 
+                with gr.Tab(label="学术功能"):
+                    with gr.Row():
+                        aep = gr.Button("英语学术润色")
+                        acp = gr.Button("中文学术润色")
+                        sge = gr.Button("查找英文语法错误")
+                        ac2e = gr.Button("学术中译英")
+                        c2e = gr.Button("中译英")
+                        e2c = gr.Button("英译中")
+
+                with gr.Tab(label="角色"):
+                    with gr.Row():
+                        interviewer = gr.Button("面试官")
+
+                with gr.Tab(label="Davinci-003"):
+                    with gr.Column():
+                        with gr.Row():
+                            with gr.Column():
+                                davinci_user_input = gr.Textbox(show_label=False, placeholder="在这里输入").style(
+                                    container=False)
+                                temperature_davinci = gr.Slider(minimum=-0, maximum=1.0, value=0.7,
+                                                step=0.1, interactive=True, label="Temperature", )
+                                davinci_submitBtn = gr.Button("🚀", variant="primary")
+                            davinci_output = gr.Textbox(show_label=False, placeholder="output").style(
+                                    container=False)
+
     gr.HTML("""
     <div style="text-align: center; margin-top: 20px; margin-bottom: 20px;">
     """)
@@ -266,6 +291,53 @@ with gr.Blocks(css=customCSS) as demo:
                                   [systemPromptTxt],
                                   show_progress=True)
 
+    # 功能
+    function_button_list = [aep, acp, sge, ac2e, c2e, e2c]
+    for i in function_button_list:
+        name = gr.Dropdown(choices=list(function.keys()), value=i.value, visible=False)
+        i.click(get_function_content,
+                [function_template, name],
+                [systemPromptTxt],
+                show_progress=True)
+
+    # 角色
+    interviewer.click(predict, [
+        keyTxt,
+        invite_code,
+        interviewer_prompt,
+        history,
+        user_input,
+        chatbot,
+        token_count,
+        top_p,
+        temperature,
+        use_streaming_checkbox,
+        model_select_dropdown,
+        use_websearch_checkbox,
+        index_files],
+                    [chatbot, history, status_display, token_count], show_progress=True)
+
+    interviewer.click(get_character_content,
+                      [interviewer_prompt],
+                      [systemPromptTxt])
+
+    # Davinci
+    davinci_user_input.submit(predict_davinci,
+                      [
+                          keyTxt,
+                          davinci_user_input,
+                          temperature,
+                      ],
+                      [davinci_output], show_progress=True)
+
+    davinci_submitBtn.click(predict_davinci,
+                      [
+                          keyTxt,
+                          davinci_user_input,
+                          temperature_davinci,
+                      ],
+                      [davinci_output], show_progress=True)
+
 logging.info("\n访问 http://localhost:7860 查看界面")
 # 默认开启本地服务器，默认可以直接从IP访问，默认不创建公开分享链接
 demo.title = "ChatGPT-长江商学院 🚀"