updata

README.md

@@ -4,7 +4,7 @@ emoji: 🚀
 colorFrom: pink
 colorTo: purple
 sdk: gradio
-sdk_version: 3.20.1
+sdk_version: 3.50.2
 app_file: app.py
 pinned: false
 license: gpl-3.0

app.py

@@ -9,7 +9,7 @@ import argparse
 import configparser
 import fitz, io, os
 from PIL import Image
-import gradio
+import gradio as gr
 import markdown
 import mistune
 import json
@@ -551,7 +551,6 @@ class Reader:
                 ) or 'approach' in parse_key.lower():
                     method_key = parse_key
                     break
-            # 找不到method的部分
             if method_key != '':
                 text = ''
                 method_text = ''
@@ -563,8 +562,8 @@ class Reader:
                 chat_method_text, utoken2, ctoken2, ttoken2 = self.chat_method(
                     text=text)
             else:
+                chat_method_text = 'Some Error for method (比如是不是没有Methods这个部分)'
                 utoken2, ctoken2, ttoken2 = 0, 0, 0
-                chat_method_text = ''
             htmls.append(chat_method_text)
             htmls.append("\n")
 
@@ -585,8 +584,13 @@ class Reader:
                 text = summary_text + "\n <Conclusion>:\n" + conclusion_text
             else:
                 text = summary_text
-            chat_conclusion_text, utoken3, ctoken3, ttoken3 = self.chat_conclusion(
-                text=text)
+            try:
+                chat_conclusion_text, utoken3, ctoken3, ttoken3 = self.chat_conclusion(
+                    text=text)
+            except Exception:
+                utoken3, ctoken3, ttoken3 = 0, 0, 0
+                print(Exception)
+                chat_conclusion_text = 'Some Error for conclusion (比如是不是没有Conclusion这个部分)'
             htmls.append(chat_conclusion_text)
             htmls.append("\n")
             # token统计
@@ -791,7 +795,7 @@ def upload_pdf(api_keys, text, model_name, p, temperature, file):
                         temperature=temperature)
         sum_info, cost = reader.summary_with_chat(
             paper_list=paper_list)  # type: ignore
-        return cost, sum_info
+        return cost, sum_info, "output.md"
 
 
 api_title = "api-key可用验证(ChatPaperFree Gemini Pro)"
@@ -818,18 +822,18 @@ Use Gemini Pro to summary the papers.Star our Github [🌟ChatPaperFree](https:/
 '''
 
 api_input = [
-    gradio.inputs.Textbox(label="请输入你的API-key(必填, 多个API-key请用英文逗号隔开)",
-                          default="",
+    gr.Textbox(label="请输入你的API-key(必填, 多个API-key请用英文逗号隔开)",
+                          value="",
                           type='password')
 ]
-api_gui = gradio.Interface(fn=valid_apikey,
+api_gui = gr.Interface(fn=valid_apikey,
                            inputs=api_input,
                            outputs="text",
                            title=api_title,
                            description=api_description)
 
 # 标题
-title = "ChatPaperFree (Gemini Pro)"
+title = "ChatPaper(Gemini Pro)"
 # 描述
 description = api_description = '''<div align='left'>
 
@@ -852,39 +856,39 @@ Use Gemini Pro to summary the papers.Star our Github [🌟ChatPaperFree](https:/
 
 </div>
 '''
-# 创建Gradio界面
+# 创建gr界面
 ip = [
-    gradio.inputs.Textbox(label="请输入你的API-key(必填, 多个API-key请用英文逗号隔开),不需要空格",
-                          default="",
+    gr.Textbox(label="请输入你的API-key(必填, 多个API-key请用英文逗号隔开),不需要空格",
+                          value="",
                           type='password'),
-    gradio.inputs.Textbox(
+    gr.Textbox(
         label="请输入论文大标题索引(用英文逗号隔开,必填)",
-        default=
+        value=
         "'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'"
     ),
-    gradio.inputs.Radio(choices=["gemini-Pro", "gemini-Pro-vision(comming Soon)"],
-                        default="gemini-Pro",
+    gr.Radio(choices=["gemini-Pro", "gemini-Pro-vision(comming Soon)"],
+                        value="gemini-Pro",
                         label="Select model"),
-    gradio.inputs.Slider(minimum=-0,
+    gr.Slider(minimum=-0,
                          maximum=1.0,
-                         default=1.0,
+                         value=1.0,
                          step=0.05,
                          label="Top-p (nucleus sampling)"),
-    gradio.inputs.Slider(minimum=-0,
+    gr.Slider(minimum=-0,
                          maximum=5.0,
-                         default=0.5,
+                         value=0.5,
                          step=0.5,
                          label="Temperature"),
-    gradio.inputs.File(label="请上传论文PDF(必填)")
+    gr.File(label="请上传论文PDF(必填)", file_types = ['.pdf']),
 ]
 
-chatpaper_gui = gradio.Interface(fn=upload_pdf,
+chatpaper_gui = gr.Interface(fn=upload_pdf,
                                  inputs=ip,
-                                 outputs=["json", "html"],
+                                 outputs=[gr.Json(label = 'ChatPaperFree Tokens'), gr.Markdown(label = "ChatPaperFree 结果"), gr.File(label = "下载文章")],
                                  title=title,
                                  description=description)
 
 # Start server
-gui = gradio.TabbedInterface(interface_list=[api_gui, chatpaper_gui],
-                             tab_names=["API-key", "ChatPaper"])
-gui.launch(quiet=True, show_api=False)
+gui = gr.TabbedInterface(interface_list=[api_gui, chatpaper_gui],
+                             tab_names=["API-key", "ChatPaperFree"])
+gui.launch(quiet=True, show_api=False, debug=True)

requirements.txt

@@ -5,10 +5,9 @@ tiktoken==0.2.0
 tenacity==8.2.2
 pybase64==1.2.3
 Pillow==9.4.0
-openai==0.27.0
 markdown
 mistune
-gradio==3.20.1
+gradio==3.50.2
 google-generativeai==0.3.1
 pydantic==1.10.7
 #tiktoken