update

app.py

@@ -767,13 +767,13 @@ class Reader:
         print(f"Sort: {self.sort}")
 
 
-def upload_pdf(api_keys, text, model_name, p, temperature, file):
+def upload_pdf( text, model_name, p, temperature, file):
     # 检查两个输入都不为空
-    api_key_list = None
-    if api_keys:
-        api_key_list = api_keys.split(',')
-    elif not api_keys and valid_api_keys != []:
-        api_key_list = valid_api_keys
+    api_key_list = [os.environ.get('GOOGLE_API_KEY')]
+    # if api_keys:
+    #     api_key_list = api_keys.split(',')
+    # elif not api_keys and valid_api_keys != []:
+    #     api_key_list = valid_api_keys
     if not text or not file or not api_key_list:
         return "两个输入都不能为空，请输入字符并上传 PDF 文件！"
 
@@ -829,7 +829,7 @@ api_gui = gr.Interface(fn=valid_apikey,
                            description=api_description)
 
 # 标题
-title = "ChatPaper(Gemini Pro)"
+title = "ChatPaperFree (Gemini Pro)"
 # 描述
 description = api_description = '''<div align='left'>
 
@@ -854,9 +854,9 @@ Use Gemini Pro to summary the papers.Star our Github [🌟ChatPaperFree](https:/
 '''
 # 创建gr界面
 ip = [
-    gr.Textbox(label="请输入你的API-key(必填, 多个API-key请用英文逗号隔开),不需要空格",
-                          value="",
-                          type='password'),
+    # gr.Textbox(label="请输入你的API-key(必填, 多个API-key请用英文逗号隔开),不需要空格",
+    #                       value="",
+    #                       type='password'),
     gr.Textbox(
         label="请输入论文大标题索引(用英文逗号隔开,必填)",
         lines=3,
@@ -883,9 +883,17 @@ chatpaper_gui = gr.Interface(fn=upload_pdf,
                                  inputs=ip,
                                  outputs=[gr.Json(label = 'ChatPaperFree Tokens'), gr.Markdown(label = "ChatPaperFree 结果"), gr.File(label = "下载文章")],
                                  title=title,
-                                 description=description)
+                                 description=description,
+                                 examples = [[
+                                     "'Abstract,Introduction,Related Work,Background,Preliminary,Problem Formulation,Methods,Methodology,Method,Approach,Approaches,Materials and Methods,Experiment Settings,Experiment,Experimental Results,Evaluation,Experiments,Results,Findings,Data Analysis,Discussion,Results and Discussion,Conclusion,References'",
+                                     'gemini-Pro','1.0', '0.5', 'https://arxiv.org/pdf/2301.13430.pdf'
+                                 ],[
+                                     "'Abstract,Introduction,Related Work,Background,Preliminary,Problem Formulation,Methods,Methodology,Method,Approach,Approaches,Materials and Methods,Experiment Settings,Experiment,Experimental Results,Evaluation,Experiments,Results,Findings,Data Analysis,Discussion,Results and Discussion,Conclusion,References'",
+                                     'gemini-Pro','1.0', '0.5', 'https://arxiv.org/pdf/2311.06772.pdf']
+                                 ]
+                                             )
 
 # Start server
 gui = gr.TabbedInterface(interface_list=[api_gui, chatpaper_gui],
-                             tab_names=["API-key", "ChatPaperFree"])
+                             tab_names=["ChatPaperFree"])
 gui.launch(quiet=True, show_api=False, debug=True)