abc

.idea/.name

@@ -0,0 +1 @@
+app.py

app.py

@@ -26,17 +26,25 @@ def add_details(lst):
 
 
 prompt_template = "Instructions: Compose a comprehensive reply to the query using the search results given. " \
-                  "Make sure to cite results using [number] notation after the reference. " \
                   "If the search results mention multiple subjects " \
                   "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. \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)
 
-
+    
 def add_source_numbers(lst):
     return [item[:3] + '\t' + item[3:] for item in (lst)]
 
@@ -60,11 +68,11 @@ def preprocess(text):
 def pdf_to_text(files_src, start_page=1, end_page=None):
     text_list = []
     for file in files_src:
-        if os.path.splitext(file.name)[1] == ".pdf":
+        if (os.path.splitext(file.name)[1]).lower() == ".pdf":
             doc = fitz.open(file.name)
             total_pages = doc.page_count
-            if end_page is None:
-                end_page = total_pages
+            # if end_page is None:
+            end_page = total_pages
             for i in range(start_page - 1, end_page):
                 text = doc.load_page(i).get_text("text")
                 text = preprocess(text)
@@ -124,6 +132,7 @@ def predict(
         chatbot,
         inputs,
         temperature,
+        lang = LANGUAGES[0],
         selected_model=MODELS[0],
         files=None
 ):
@@ -142,7 +151,7 @@ def predict(
         for c in topn_chunks:
             prompt += c + '\n\n'
         prompt += prompt_template
-        prompt += f"Query: {inputs}\nAnswer:"
+        prompt += f"Query: {inputs}. Reply in {lang}\nAnswer:"
         inputs = prompt
         reference_results = add_source_numbers(topn_chunks)
         display_reference = add_details(reference_results)
@@ -215,7 +224,7 @@ with gr.Blocks(css=customCSS, theme=beautiful_theme) as demo:
         with gr.Column(scale=5):
             with gr.Row():
                 chatbot = gr.Chatbot(elem_id="chatbot").style(height="100%")
-            with gr.Row():
+            with gr.Row(visible=False) as input_raws:
                 with gr.Column(scale=12):
                     user_input = gr.Textbox(
                         show_label=False, placeholder="Enter here"
@@ -227,12 +236,21 @@ with gr.Blocks(css=customCSS, theme=beautiful_theme) as demo:
             with gr.Column(min_width=50, scale=1):
                 with gr.Tab(label="ChatGPT"):
                     gr.Markdown(f'<p style="text-align:center">Get your Open AI API key <a '
+<<<<<<< HEAD
                                 f'href="https://platform.openai.com/account/api-keys">here</a></p>')
                     openAI_key=gr.Textbox(label='Enter your OpenAI API key here')
+=======
+                    f'href="https://platform.openai.com/account/api-keys">here</a></p>')
+                    openAI_key=gr.Textbox(label='Enter your OpenAI API key here and press Enter')
+>>>>>>> 2cb7b56a3054d0b5febf8ed9db8c59395b2d0327
                     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(
                         "⚠️Be careful to change ⚠️\n\nIf you can't use it, please restore the default settings")
                 with gr.Tab(label="Advanced"):
@@ -245,10 +263,18 @@ with gr.Blocks(css=customCSS, theme=beautiful_theme) as demo:
                             interactive=True,
                             label="Temperature",
                         )
+<<<<<<< HEAD
     user_input.submit(predict, inputs=[openAI_key, history, chatbot, user_input, temperature, model_select_dropdown, index_files],
                       outputs=[chatbot, history])
     user_input.submit(lambda: "", None, user_input)
     submitBtn.click(predict, inputs=[openAI_key, history, chatbot, user_input, temperature, model_select_dropdown, index_files],
+=======
+    openAI_key.submit(set_openai_api_key, [openAI_key], [input_raws])
+    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, language_select_dropdown, model_select_dropdown, index_files],
+>>>>>>> 2cb7b56a3054d0b5febf8ed9db8c59395b2d0327
                     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)
+    demo.queue(concurrency_count=10).launch(server_name="0.0.0.0",  server_port=7862)