Update app.py

app.py

@@ -79,6 +79,17 @@ 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):
+    """スプレッドシートから指定されたpaper_idを検索し、該当する行があればその内容を返す。"""
+    records = worksheet.get_all_records()
+    
+    # 各行をループしてpaper_idを検索
+    for index, record in enumerate(records, start=2):  # 行インデックスは1ではなく2から開始（ヘッダー行を除く）
+        if record['ID'] == paper_id:
+            return record['summary']
+    # 該当する行がない場合はNoneを返す
+    return None
+
 def gradio_interface():
     paper_links = fetch_paper_links("https://huggingface.co/papers")
     paper_ids = set(link.split('/')[-1] for link in paper_links)
@@ -88,13 +99,13 @@ def gradio_interface():
 
     for paper_id in paper_ids:
         summary_info = ""
-        try:
+        summary = find_paper_in_sheet(paper_id)
+        if summary == None:
             summary, tokens_used = summarize_paper_and_save_to_sheet(paper_id)
             total_tokens_used += tokens_used
-            paper_id_url = f"https://arxiv.org/pdf/{paper_id}.pdf"
-            summary_info += f'論文: {paper_id_url}\n{summary}\n'
-        except Exception as e:
-            summary_info += f"Error processing paper ID {paper_id}: {e}\n"
+            
+        paper_id_url = f"https://arxiv.org/pdf/{paper_id}.pdf"
+        summary_info += f'論文: {paper_id_url}\n{summary}\n'
         
         summaries.append(summary_info)