Update app.py

app.py

@@ -49,10 +49,10 @@ def insert_sentence(text, sentence, interval):
 # 定义Reviewer类
 class Reviewer:
     # 初始化方法，设置属性
-    def __init__(self, api, review_format, paper_pdf, language):
+    def __init__(self, api, api_base, review_format, paper_pdf, language):
         self.api = api
         self.review_format = review_format
-
+        self.api_base = api_base
         self.language = language
         self.paper_pdf = paper_pdf
         self.max_token_num = 12000
@@ -86,7 +86,7 @@ class Reviewer:
             response = openai.ChatCompletion.create(
                 model="gpt-3.5-turbo-16k",
                 messages=messages,
-                temperature=0.7
+                temperature=0.5
             )
             result = ''
             for choice in response.choices:
@@ -131,7 +131,7 @@ class Reviewer:
                     break
         return extracted_text
 
-def main(api, review_format, paper_pdf, language):  
+def main(api,api_base, review_format, paper_pdf, language):  
     start_time = time.time()
     comments = ''
     output2 = ''
@@ -141,7 +141,7 @@ def main(api, review_format, paper_pdf, language):
     # 判断PDF文件
     else:
         # 创建一个Reader对象
-        reviewer1 = Reviewer(api, review_format, paper_pdf, language)
+        reviewer1 = Reviewer(api,api_base, review_format, paper_pdf, language)
         # 开始判断是路径还是文件：   
         comments, total_token_used = reviewer1.review_by_chatgpt(paper_list=paper_pdf)
         time_used = time.time() - start_time
@@ -176,6 +176,8 @@ description = '''<div align='left'>
 inp = [gradio.inputs.Textbox(label="请输入你的API-key(sk开头的字符串)",
                           default="",
                           type='password'),
+       gradio.inputs.Textbox(label="请输入第三方中转网址",
+                          default="https://api.openai.com/v1"),
     gradio.inputs.Textbox(lines=5,
         label="请输入特定的分析要求和格式(否则为默认格式)",
         default="""* Overall Review