Update app.py

app.py

@@ -79,9 +79,8 @@ def summarize_paper_and_save_to_sheet(paper_id):
     worksheet.append_row([paper_id, paper_url, summary, token])
     return summary, token
 
-def find_paper_in_sheet(paper_id):
+def find_paper_in_sheet(records, paper_id):
     """スプレッドシートから指定されたpaper_idを検索し、該当する行があればその内容を返す。"""
-    records = worksheet.get_all_records()
     paper_id_url = f"https://arxiv.org/pdf/{paper_id}.pdf"
     
     # 各行をループしてpaper_idを検索
@@ -97,10 +96,11 @@ def gradio_interface():
 
     total_tokens_used = 0
     summaries = []
+    records = worksheet.get_all_records()
 
     for paper_id in paper_ids:
         summary_info = ""
-        summary = find_paper_in_sheet(paper_id)
+        summary = find_paper_in_sheet(records, paper_id)
         if summary == None:
             summary, tokens_used = summarize_paper_and_save_to_sheet(paper_id)
             total_tokens_used += tokens_used