Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 import re
 import tiktoken
 import pandas as pd
+import functools
 
 api_key = os.getenv('API_KEY')
 base_url = os.getenv("BASE_URL")
@@ -77,7 +78,8 @@ def extract_pdf_pypdf(pdf_dir):
 
     return file_content
 
-
+# 设置缓存大小，可以根据需要调整
+@functools.lru_cache(maxsize=128)
 def extract_pdf_md(pdf_dir):
     print(f"start convert pdf 2 md: {pdf_dir}")
     try:
@@ -453,7 +455,6 @@ with gr.Blocks(title="Automated Enzyme Kinetics Extractor") as demo:
                 search_output.value = initial_output.to_html(classes='data', index=False, header=True)
 
 
-
     extract_button_md.click(extract_pdf_md, inputs=file_input, outputs=text_output)
     extract_button_text.click(extract_pdf_pypdf, inputs=file_input, outputs=text_output)
     exp.click(update_input, outputs=model_input)