Add File

app.py

@@ -1,9 +1,12 @@
 #安装一下 Python 的 Gradio 的包：conda install -c conda-forge gradio
+#依赖tiktoken：pip install tiktoken，使用不同的 GPT 模型，对应着不同的 Tiktoken 的编码器模型：https://github.com/openai/openai-cookbook/blob/main/examples/How_to_count_tokens_with_tiktoken.ipynb。
 import gradio as gr
 import openai
 import os
+import tiktoken
 
 openai.api_key = os.environ.get("OPENAI_API_KEY")
+encoding = tiktoken.get_encoding("p50k_base")
 
 class Conversation:
     def __init__(self, prompt, num_of_round):
@@ -13,12 +16,15 @@ class Conversation:
         self.messages.append(self.prompt)
 
     def ask(self, question):
+        if len(encoding.encode(prompt)) > 512:
+            return "输入的文字过长，请精简后再提问"
+            
         try:
             self.messages.append(question)
             response = openai.Completion.create(
                 engine="text-davinci-003",
                 prompt=question,
-                max_tokens=4096,
+                max_tokens=3580,
                 n=1,
                 stop=None,
                 temperature=0.5,