add batch test

ape/ape.py

@@ -53,9 +53,10 @@ def single_test(test_prompt, instruction, input, openai_key):
     global LLM
     if LLM is None:
         LLM = LLMGPT(openai_key)
-
-    output = LLM.generate_output(test_prompt, instruction, input)
-    return output['text']
+    output = []
+    for i in input.split('\n'):
+        output.append(LLM.generate_output(test_prompt, instruction, i)['text'])
+    return '\n'.join(output)
 
 
 def score_single(eval_prompt, instance, instruction, openai_key):

ape/data/test_loader2.txt

@@ -0,0 +1,5 @@
+横结肠造口还纳术 横结肠造口闭合术
+右肾上腺巨大肿瘤切除术 肾上腺病损切除术
+左侧单侧乳房根治性切除术 单侧根治性乳房切除术
+经皮三叉神经半月节射频热凝术 三叉神经半月节射频热凝术
+经内镜支撑喉镜下双侧声带小结摘除术 内镜下声带病损切除术

app.py

@@ -71,7 +71,7 @@ with gr.Blocks(title="Automatic Prompt Engineer", theme=gr.themes.Glass()) as de
                 with gr.Row():
                     with gr.Column(scale=1):
                         test_instruction = gr.Textbox(lines=4, value="", label="Prompt to test")
-                        test_input = gr.Textbox(lines=1, value="", label="Input used to test prompt")
+                        test_input = gr.Textbox(lines=1, value="", label="Inputs used to test prompt[多个输入以换行分割]")
                         test_button = gr.Button("Test")
                     with gr.Column(scale=1):
                         test_output = gr.Textbox(lines=9, value="", label="Model Output")
@@ -111,4 +111,4 @@ with gr.Blocks(title="Automatic Prompt Engineer", theme=gr.themes.Glass()) as de
     # 7. 输入指令打分
     score_button.click(score_single, [eval_prompt, instance, score_instruction, openai_key], [test_score])
 
-    demo.launch(show_error=True)
+    demo.launch(show_error=True)