DField commited on
Commit
94dcd05
1 Parent(s): 9db3235

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -28
app.py CHANGED
@@ -8,8 +8,6 @@ import gradio as gr
8
  import gspread
9
  from oauth2client.service_account import ServiceAccountCredentials
10
  import json
11
- from datetime import datetime
12
-
13
 
14
  def connect_gspread(spread_sheet_key):
15
  """Google スプレッドシートに接続。"""
@@ -93,22 +91,8 @@ def find_paper_in_sheet(paper_id):
93
  # 該当する行がない場合はNoneを返す
94
  return None
95
 
96
- def gradio_interface(selected_date_str, today_pressed):
97
- # 現在の日付を YYYY-MM-DD 形式で取得
98
- current_date_str = datetime.now().strftime("%Y-%m-%d")
99
-
100
- if today_pressed:
101
- paper_links = fetch_paper_links(f"https://huggingface.co/papers")
102
- else:
103
- try:
104
- # 日付文字列を検証し、正しい場合はその値を使用
105
- datetime.strptime(selected_date_str, "%Y-%m-%d")
106
- date = selected_date_str
107
- except ValueError:
108
- # 日付形式が無効な場合、エラーメッセージを返します。
109
- return "入力された日付が無効です。YYYY-MM-DD形式で入力してください。"
110
- paper_links = fetch_paper_links(f"https://huggingface.co/papers?date={date}")
111
-
112
  paper_ids = set(link.split('/')[-1] for link in paper_links)
113
 
114
  total_tokens_used = 0
@@ -117,7 +101,7 @@ def gradio_interface(selected_date_str, today_pressed):
117
  for paper_id in paper_ids:
118
  summary_info = ""
119
  summary = find_paper_in_sheet(paper_id)
120
- if summary is None:
121
  summary, tokens_used = summarize_paper_and_save_to_sheet(paper_id)
122
  total_tokens_used += tokens_used
123
 
@@ -129,19 +113,13 @@ def gradio_interface(selected_date_str, today_pressed):
129
  summaries_markdown = "\n---\n".join(summaries) # 要約を水平線で区切る
130
  return summaries_markdown
131
 
132
- # Gradio インターフェースの設定
133
- inputs = [
134
- gr.Text(label="日付をYYYY-MM-DD形式で入力", placeholder="例: 2024-03-28"),
135
- gr.Button("Today")
136
- ]
137
-
138
  iface = gr.Interface(
139
  fn=gradio_interface,
140
- inputs=inputs,
141
  outputs=gr.Markdown(),
142
- title="Daily Papers 日本語要約ツール",
143
  description="[Daily Papers](https://huggingface.co/papers)に掲載された論文を日本語で要約します。"
144
  )
145
 
146
  if __name__ == "__main__":
147
- iface.launch()
 
8
  import gspread
9
  from oauth2client.service_account import ServiceAccountCredentials
10
  import json
 
 
11
 
12
  def connect_gspread(spread_sheet_key):
13
  """Google スプレッドシートに接続。"""
 
91
  # 該当する行がない場合はNoneを返す
92
  return None
93
 
94
+ def gradio_interface():
95
+ paper_links = fetch_paper_links("https://huggingface.co/papers")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
96
  paper_ids = set(link.split('/')[-1] for link in paper_links)
97
 
98
  total_tokens_used = 0
 
101
  for paper_id in paper_ids:
102
  summary_info = ""
103
  summary = find_paper_in_sheet(paper_id)
104
+ if summary == None:
105
  summary, tokens_used = summarize_paper_and_save_to_sheet(paper_id)
106
  total_tokens_used += tokens_used
107
 
 
113
  summaries_markdown = "\n---\n".join(summaries) # 要約を水平線で区切る
114
  return summaries_markdown
115
 
 
 
 
 
 
 
116
  iface = gr.Interface(
117
  fn=gradio_interface,
118
+ inputs=[],
119
  outputs=gr.Markdown(),
120
+ title="Dairy Papers 日本語要約ツール",
121
  description="[Daily Papers](https://huggingface.co/papers)に掲載された論文を日本語で要約します。"
122
  )
123
 
124
  if __name__ == "__main__":
125
+ iface.launch()