Update app.py

app.py

@@ -31,11 +31,18 @@ prompt_template = "Instructions: Compose a comprehensive reply to the query usin
                   "with the same name, create separate answers for each. Only include information found in the results and " \
                   "don't add any additional information. Make sure the answer is correct and don't output false content. " \
                   "Ignore outlier search results which has nothing to do with the question. Only answer what is asked. " \
-                  "The answer should be short and concise.\n\nQuery: {question}\nAnswer: "
+                  "The answer should be short and concise. Reply in {reply_language}. \n\nQuery: {question}\nAnswer: "
 
 # MODELS = ["universal-sentence-encoder", "instructor-large"]
 MODELS = ["text-davinci-001", "text-davinci-002", "text-davinci-003"]
-
+LANGUAGES = 
+    "English",
+    "简体中文",
+    "日本語",
+    "Deutsch",
+    "Vietnamese"
+    "跟随问题语言（不稳定）"
+]
 def set_openai_api_key(my_api_key):
     openai.api_key = my_api_key
     return gr.update(visible = True)
@@ -127,6 +134,7 @@ def predict(
         chatbot,
         inputs,
         temperature,
+        lang = LANGUAGES[0],
         selected_model=MODELS[0],
         files=None
 ):
@@ -142,6 +150,7 @@ def predict(
         topn_chunks = [chunks[i] for i in index_top_chunks]
         prompt = ""
         prompt += 'search results:\n\n'
+        prompt_template.replace("{reply_language}", lang)
         for c in topn_chunks:
             prompt += c + '\n\n'
         prompt += prompt_template
@@ -234,6 +243,9 @@ with gr.Blocks(css=customCSS, theme=beautiful_theme) as demo:
                     model_select_dropdown = gr.Dropdown(
                         label="Select model", choices=MODELS, multiselect=False, value=MODELS[0]
                     )
+                    language_select_dropdown = gr.Dropdown(
+                        label="Select reply language", choices=LANGUAGES, multiselect=False, value=LANGUAGES[0]
+                    )
                     index_files = gr.Files(label="Files", type="file", multiple=True)
                 with gr.Tab(label="Advanced"):
                     gr.Markdown(
@@ -248,10 +260,10 @@ with gr.Blocks(css=customCSS, theme=beautiful_theme) as demo:
                             label="Temperature",
                         )
     openAI_key.submit(set_openai_api_key, [openAI_key], [input_raws])
-    user_input.submit(predict, inputs=[history, chatbot, user_input, temperature, model_select_dropdown, index_files],
+    user_input.submit(predict, inputs=[history, chatbot, user_input, temperature, language_select_dropdown, model_select_dropdown, index_files],
                       outputs=[chatbot, history])
     user_input.submit(lambda: "", None, user_input)
-    submitBtn.click(predict, inputs=[history, chatbot, user_input, temperature, model_select_dropdown, index_files],
+    submitBtn.click(predict, inputs=[history, chatbot, user_input, temperature, language_select_dropdown, model_select_dropdown, index_files],
                     outputs=[chatbot, history])
     submitBtn.click(lambda: "", None, user_input)
     demo.queue(concurrency_count=10).launch(server_name="0.0.0.0", server_port=7860, debug=True)