Upload 2 files

app.py

@@ -0,0 +1,69 @@
+import openai
+import os
+import gradio as gr
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+class Conversation:
+    def __init__(self, prompt):
+        self.prompt = prompt
+        self.messages = []
+        self.messages.append({"role": "system", "content": self.prompt})
+
+    def ask(self, question):
+        try:
+            self.messages.append({"role": "user", "content": question})
+            response = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo",
+                messages=self.messages,
+                temperature=0.5,
+                max_tokens=2048,
+                top_p=1,
+            )
+        except Exception as e:
+            print(e)
+            return e
+
+        message = response["choices"][0]["message"]["content"]
+        self.messages.append({"role": "assistant", "content": message})
+
+        return message
+
+
+prompt = """你是一个英语老师，请批改英语作文。用中文回答，回答限制在100个字以内，你的回答按照以下格式:
+1. 文章主要内容：
+2. 语法错误：
+3. 拼写错误：
+4. 百分制评分：
+5. 给出评语：
+如果输入不是英文作文，则提醒输入英文作文"""
+
+conv = Conversation(prompt)
+
+def moderation(text):
+    response = openai.Moderation.create(
+        input=text
+    )
+    output = response["results"][0]['flagged']
+    return output
+
+def answer(question):
+    #调用moderation函数，对输入的作文进行审核,如果有不良内容，返回提示信息
+    moderation_flag = moderation(question)
+    if moderation_flag == True:
+        response = "输入文字有不良用语，请重新输入"
+    else:
+        response = conv.ask(question)
+    return response
+
+
+
+# 使用gradio画出修改英语作文的界面，左边是输入框和提交按钮，右边是输出评语框，并调用gpt进行作文修改
+with gr.Interface(
+    answer, 
+    gr.inputs.Textbox(lines=5,placeholder="输入英语作文",label="作文"), 
+    gr.outputs.Textbox(label="评语"), 
+    title="批改英语作文",
+    allow_flagging='never'
+) as demo:
+    demo.launch(share=True)

requirements.txt

@@ -0,0 +1 @@
+openai