Update bin_public/app/Chatbot.py

bin_public/app/Chatbot.py

@@ -7,6 +7,7 @@ from openai_func import *
 from bin_public.utils.utils import *
 from bin_public.utils.utils_db import *
 from bin_public.config.presets import *
+from bin_public.utils.Pinecone import *
 
 my_api_key = ""
 
@@ -58,6 +59,9 @@ with gr.Blocks(css=customCSS) as demo:
     user_question = gr.State("")
     outputing = gr.State(False)
     topic = gr.State("未命名对话历史记录")
+    pinecone_api_key = gr.Textbox(PINECONE_API_KEY, visible=False)
+    pinecone_api_env = gr.Textbox(PINECONE_API_ENV, visible=False)
+    pinecone_index_name = gr.Textbox(PINECONE_INDEX_NAME, visible=False)
 
     # gr.HTML("""
     # <div style="text-align: center; margin-top: 20px;">
@@ -167,6 +171,7 @@ with gr.Blocks(css=customCSS) as demo:
                 with gr.Tab(label="角色"):
                     with gr.Row():
                         interviewer = gr.Button("面试官")
+                        migraine = gr.Button("问老中医-偏头痛相关")
 
                 with gr.Tab(label="Davinci-003"):
                     with gr.Column():
@@ -323,6 +328,33 @@ with gr.Blocks(css=customCSS) as demo:
                       [interviewer_prompt],
                       [systemPromptTxt])
 
+    migraine.click(context_construction,[
+        keyTxt,
+        user_input,
+        model_select_dropdown,
+        pinecone_api_key,
+        pinecone_api_env,
+        temperature,
+        pinecone_index_name],
+        [systemPromptTxt, status_display]
+                   ).then(
+        predict, [
+            keyTxt,
+            invite_code,
+            systemPromptTxt,
+            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_textbox, [], [user_input])
+
     # Davinci
     davinci_user_input.submit(predict_davinci,
                       [