update

app.py

@@ -62,7 +62,7 @@ Please automatically generate as many questions as possible based on this manual
 """
 
 
-def get_answer(question,session_id,language,prompt,search_engine,index,top_k,score_type_checklist):
+def get_answer(task_type,question,session_id,language,prompt,search_engine,index,top_k,score_type_checklist):
     
     question=question.replace('AWS','亚马逊云科技').replace('aws','亚马逊云科技').replace('Aws','亚马逊云科技')
     print('question:',question)
@@ -72,12 +72,11 @@ def get_answer(question,session_id,language,prompt,search_engine,index,top_k,sco
     else:
         url = api + "hello"
     
-    # task='chat'
-    # if question.find('电商')>=0 or question.find('开店')>=0 or question.find('亚马逊')>=0:
-    #     task = 'qa'
-    # url += ('&task='+task)
-
-    task = 'qa'
+    #task type: qa,chat
+    if task_type == "Knowledge base Q&A":
+        task = 'qa'
+    else:
+        task = 'chat'
     url += ('&task='+task)
 
     if language == "english":
@@ -233,6 +232,7 @@ with demo:
 
             with gr.Row():
                 with gr.Column():
+                    qa_task_radio = gr.Radio(["Knowledge base Q&A","Chat"],value="Knowledge base Q&A",label="Task")
                     query_textbox = gr.Textbox(label="Query")
                     session_id_textbox = gr.Textbox(label="Session ID")
                     qa_button = gr.Button("Summit")
@@ -266,7 +266,7 @@ with demo:
                     text_output = gr.Textbox()
             
        
-    qa_button.click(get_answer, inputs=[query_textbox,session_id_textbox,qa_language_radio,qa_prompt_textbox,qa_search_engine_radio,qa_index_textbox,qa_top_k_slider,score_type_checklist], outputs=qa_output)
+    qa_button.click(get_answer, inputs=[qa_task_radio,query_textbox,session_id_textbox,qa_language_radio,qa_prompt_textbox,qa_search_engine_radio,qa_index_textbox,qa_top_k_slider,score_type_checklist], outputs=qa_output)
     summarize_button.click(get_summarize, inputs=[text_input,sm_language_radio,sm_prompt_textbox], outputs=text_output)
 
 demo.launch()