Update bin_public/app/Chatbot.py

bin_public/app/Chatbot.py

@@ -90,7 +90,8 @@ with gr.Blocks(css=customCSS) as demo:
                 with gr.Column(min_width=50, scale=1):
                     submitBtn = gr.Button("🚀", variant="primary")
                     cancelBtn = gr.Button("取消", variant="secondary", visible=False)
-            finalBtn = gr.Button("结果提交", variant="primary")
+            finalBtn = gr.Button("诊断结果提交", variant="primary")
+            kaiyaoBtn = gr.Button("开药结果提交", variant="primary")
             with gr.Row(scale=1):
                 emptyBtn = gr.Button("🧹 新的对话", visible=False)
                 retryBtn = gr.Button("🔄 重新生成", visible=False)
@@ -310,7 +311,7 @@ with gr.Blocks(css=customCSS) as demo:
     ''' =============================================================================================='''
     # 角色
 
-    finalBtn.click(predict, [
+    finalBtn.click(construct_zhenduan, [user_input], [user_input]).then(predict, [
         keyTxt,
         invite_code,
         gr.Textbox('不管我说了什么，你只需要回复：“好的谢谢医生”', visible=False),
@@ -324,10 +325,34 @@ with gr.Blocks(css=customCSS) as demo:
         model_select_dropdown,
         use_websearch_checkbox,
         index_files],
-                    [chatbot, history, status_display, token_count], show_progress=True).then(reset_state, outputs=[chatbot, history, token_count, status_display], show_progress=True)
+                    [chatbot, history, status_display, token_count], show_progress=True)
 
     finalBtn.click(reset_textbox, [], [user_input])
 
+    kaiyaoBtn.click(construct_kaiyao, [user_input], [user_input]).then(predict, [
+        keyTxt,
+        invite_code,
+        gr.Textbox('不管我说了什么，你只需要回复：“好的谢谢医生”', visible=False),
+        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).then(reset_state,
+                                                                                                 outputs=[chatbot,
+                                                                                                          history,
+                                                                                                          token_count,
+                                                                                                          status_display],
+                                                                                                 show_progress=True)
+
+    kaiyaoBtn.click(reset_textbox, [], [user_input])
+
     interviewer.click(info_context, [file_list_m],
                                     [patient_prompt], show_progress=True).then(predict, [
         keyTxt,