Update app.py

app.py

@@ -18,10 +18,12 @@ generate_prompt = 'please generate {generate_number} difficult multiple-choice q
                   'question according the paragraph below. Each question should have four choices and only one of them ' \
                   'is the right answer.' \
 
-correct_prompt = 'the following paragraph is a composition of a student. ' \
-                 'please first judge the writing level of the composition, ' \
-                 'then point out the all the problems with examples, ' \
-                 'and finally output the complete revised version.'
+# correct_prompt = 'the following paragraph is a composition of a student. ' \
+#                  'please first judge the writing level of the composition, ' \
+#                  'then point out the all the problems with examples, ' \
+#                  'and finally output the complete revised version.'
+
+correct_prompt = '你是一个英语老师，现在正在批改学生的英语作文。你会用中文说明并逐一标记出每个英语语法错误，并且说明如何改正；最后，给出完整的正确作文。'
 
 llm_name_dict = {'chatgpt-3.5': 'OpenAI3', 'chatgpt-4': 'OpenAI4', 'chatglm—6b': 'ChatGLM'}
 
@@ -50,7 +52,7 @@ def llm_change(name):
 
 
 def text_generate(chatbot, text, generate_number, llm):
-    prompt = eval('f"' + generate_prompt + '"') + '\n\n' + text
+    prompt = eval('f"' + generate_prompt + '"') + '\n\nstop\n\n' + text
     answer = llm(prompt)
     chatbot = chatbot + [[text, answer]]
     return chatbot
@@ -60,7 +62,7 @@ def files_generate(chatbot, files_list, generate_number, llm):
     try:
         texts = files_list_to_texts(files_list)
         for text in texts:
-            prompt = eval('f"' + generate_prompt + '"') + '\n\n' + text
+            prompt = eval('f"' + generate_prompt + '"') + '\n\nstop\n\n' + text
             answer = llm(prompt)
             chatbot = chatbot + [[text, answer]]
     except:
@@ -68,20 +70,23 @@ def files_generate(chatbot, files_list, generate_number, llm):
     return chatbot
 
 
-def text_correct(chatbot, text, llm):
-    prompt = correct_prompt + '\n\n' + text
-    answer = llm(prompt)
-    chatbot = chatbot + [[text, answer]]
+def text_correct(chatbot, require_texts, feedback_texts, llm):
+    req = '写作要求为：\n' + require_texts
+    answer = '学生的作文内容为：\n' + feedback_texts
+    prompt = correct_prompt + req + '\n\nstop\n\n' + answer
+    response = llm(prompt)
+    chatbot = chatbot + [[answer, response]]
     return chatbot
 
 
-def files_correct(chatbot, files_list, llm):
+def files_correct(chatbot, require_texts, files_list, llm):
+    req = '写作要求为：\n' + require_texts
     try:
-        texts = files_list_to_texts(files_list)
-        for text in texts:
-            prompt = correct_prompt + '\n\n' + text
-            answer = llm(prompt)
-            chatbot = chatbot + [[text, answer]]
+        answers = files_list_to_texts(files_list)
+        for answer in answers:
+            prompt = correct_prompt + req + '\n\nstop\n\n' + '学生的作文内容为：\n' + answer
+            response = llm(prompt)
+            chatbot = chatbot + [['学生的作文内容为：\n' + answer, response]]
     except:
         chatbot = chatbot + [[None, f"所选文件夹中的文件添加失败，请确保文件夹中含有txt类型文件"]]
     return chatbot
@@ -154,10 +159,16 @@ with gr.Blocks(css=block_css) as demo:
 
             with gr.Tab('批改配置'):
 
+                gr.Markdown(f'写作要求')
+                require_texts = gr.Textbox(show_label=False, placeholder="文本内容", lines=4).style(container=False)
+                # require_texts_button = gr.Button(value="批改")
+                # require_texts_button.click(fn=text_correct, inputs=[chatbot, texts, llm], outputs=[chatbot])
+
                 gr.Markdown(f'单篇批改')
-                texts = gr.Textbox(show_label=False, placeholder="文本内容", lines=16).style(container=False)
-                text_button = gr.Button(value="批改")
-                text_button.click(fn=text_correct, inputs=[chatbot, texts, llm], outputs=[chatbot])
+                feedback_texts = gr.Textbox(show_label=False, placeholder="文本内容", lines=12).style(container=False)
+                feedback_button = gr.Button(value="批改")
+                feedback_button.click(fn=text_correct, inputs=[chatbot, require_texts, feedback_texts, llm],
+                                      outputs=[chatbot])
 
                 gr.Markdown(f'批量批改')
                 folder_address = gr.File(label="添加文件",
@@ -166,7 +177,8 @@ with gr.Blocks(css=block_css) as demo:
                                          show_label=False
                                          )
                 file_button = gr.Button(value="批改")
-                file_button.click(fn=files_correct, inputs=[chatbot, folder_address, llm], outputs=[chatbot])
+                file_button.click(fn=files_correct, inputs=[chatbot, require_texts, folder_address, llm],
+                                  outputs=[chatbot])
 
 
-demo.queue(concurrency_count=5).launch()
+demo.queue(concurrency_count=5).launch(share=True)