Update app.py

app.py

@@ -14,8 +14,9 @@ import gradio
 # 定义Response类
 class Response:
     # 初始化方法，设置属性
-    def __init__(self, api, comment, language):
+    def __init__(self, api, api_base, comment, language):
         self.api = api
+        self.api_base = api_base
         self.comment = comment
         self.language = language     
         self.max_token_num = 14096
@@ -27,6 +28,7 @@ class Response:
                     reraise=True)
     def chat_response(self, comment):
         openai.api_key = self.api
+        openai.api_base = self.api_base
         response_prompt_token = 1000        
         text_token = len(self.encoding.encode(comment))
         input_text_index = int(len(comment)*(self.max_token_num-response_prompt_token)/text_token)
@@ -82,12 +84,12 @@ class Response:
                         
                            
 
-def main(api, comment, language):  
+def main(api, api_base, comment, language):  
     start_time = time.time()
     if not api or not comment:
         return "请输入API-key以及审稿意见！"
     else:
-        Response1 = Response(api, comment, language)
+        Response1 = Response(api,api_base, comment, language)
         # 开始判断是路径还是文件：   
         response, total_token_used = Response1.chat_response(comment)
     time_used = time.time() - start_time
@@ -118,6 +120,8 @@ description = '''<div align='left'>
 inp = [gradio.inputs.Textbox(label="请输入你的API-key(sk开头的字符串)",
                           default="",
                           type='password'),
+       gradio.inputs.Textbox(label="请输入第三方中转网址(以/v1结尾，使用原始OpenAI的API请跳过这里)",
+                          default="https://api.openai.com/v1"),
     gradio.inputs.Textbox(lines=5,
         label="请输入要回复的审稿意见",
         default=""