Create app.py

app.py

@@ -0,0 +1,213 @@
+#!/usr/bin/env python
+
+import datetime
+
+import gradio as gr
+import pandas as pd
+from apscheduler.schedulers.background import BackgroundScheduler
+from gradio_calendar import Calendar
+
+from papers import PaperList, get_df
+
+DESCRIPTION = "# [Papers Leaderboard](https://huggingface.co/papers)\nAn interactive interface to explore and filter the latest papers."
+FOOT_NOTE = """\
+Community: https://discord.gg/openfreeai
+"""
+
+paper_list = PaperList(get_df())
+
+def update_paper_list() -> None:
+    global paper_list  # noqa: PLW0603
+    paper_list = PaperList(get_df())
+
+scheduler = BackgroundScheduler()
+scheduler.add_job(func=update_paper_list, trigger="cron", hour="*", timezone="UTC", misfire_grace_time=60)
+scheduler.start()
+
+def update_df() -> gr.Dataframe:
+    return gr.Dataframe(value=paper_list.df_prettified)
+
+def update_num_papers(df: pd.DataFrame) -> str:
+    return f"{len(df)} / {len(paper_list.df_raw)}"
+
+def search(
+    start_date: datetime.datetime,
+    end_date: datetime.datetime,
+    search_title: str,
+    search_abstract: str,
+    max_num_to_retrieve: int,
+) -> pd.DataFrame:
+    return paper_list.search(start_date, end_date, search_title, search_abstract, max_num_to_retrieve)
+
+
+# 추가할 CSS (Gradio Blocks의 css 파라미터)
+css = """
+/* 전체 배경이나 컨테이너 스타일 */
+body {
+  margin: 0;
+  padding: 0;
+  background: linear-gradient(135deg, #eef2ff 0%, #fdfdfd 100%);
+  font-family: "Helvetica Neue", Arial, sans-serif;
+}
+
+/* Hero / Header 섹션 스타일 */
+#hero-section {
+  background: linear-gradient(135deg, #3b82f6 0%, #9333ea 100%);
+  padding: 2rem;
+  border-radius: 0.5rem;
+  margin-bottom: 1rem;
+  color: white;
+}
+#hero-section h1 {
+  font-size: 2.2rem;
+  font-weight: bold;
+  margin-bottom: 0.5rem;
+}
+#hero-section p {
+  font-size: 1.1rem;
+  margin: 0;
+}
+
+/* 검색 섹션 스타일 */
+.search-container {
+  background-color: #ffffffdd;
+  backdrop-filter: blur(6px);
+  border-radius: 0.75rem;
+  padding: 1rem 1.5rem;
+  box-shadow: 0 3px 5px rgba(0,0,0,0.1);
+  margin-bottom: 1rem;
+}
+.search-container h2 {
+  margin-top: 0;
+  font-size: 1.4rem;
+  color: #374151;
+}
+
+/* 그리드나 카드 스타일 */
+.card {
+  background-color: #fff;
+  border-radius: 0.75rem;
+  padding: 1.5rem;
+  box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
+  margin-bottom: 1rem;
+}
+.card h3 {
+  margin-top: 0;
+  font-size: 1.2rem;
+  color: #111827;
+}
+
+/* 푸터/하단부 스타일 */
+.footer {
+  color: #6b7280;
+  font-size: 0.9rem;
+  text-align: center;
+  margin-top: 2rem;
+}
+
+/* DataFrame 테이블 추가 스타일 */
+#table table {
+  border-collapse: collapse;
+  width: 100%;
+  background: #fafafa;
+}
+#table th, #table td {
+  border: 1px solid #e5e7eb;
+  padding: 0.5rem;
+  text-align: left;
+}
+#table thead {
+  background-color: #f3f4f6;
+  font-weight: 600;
+}
+"""
+
+with gr.Blocks(css=css) as demo:
+    # 상단 Hero 섹션
+    with gr.Box(elem_id="hero-section"):
+        gr.Markdown(
+            """
+            # Papers Leaderboard
+            Explore the latest papers and filter by date, title, or abstract keywords.
+            """
+        )
+        gr.Markdown("#### [View on HuggingFace](https://huggingface.co/papers)")
+
+    gr.Markdown("### Search & Filter", elem_classes="search-container")
+
+    # 검색 영역
+    with gr.Group():
+        search_title = gr.Textbox(label="Search title")
+        with gr.Row():
+            with gr.Column(scale=4):
+                search_abstract = gr.Textbox(
+                    label="Search abstract",
+                    info="Search within abstracts (may not be fully accurate).",
+                )
+            with gr.Column(scale=1):
+                max_num_to_retrieve = gr.Slider(
+                    label="Max number to retrieve",
+                    info="Applies only to abstract-based searching",
+                    minimum=1,
+                    maximum=len(paper_list.df_raw),
+                    step=1,
+                    value=100,
+                )
+        with gr.Row():
+            start_date = Calendar(label="Start date", type="datetime", value="2023-05-05")
+            end_date = Calendar(label="End date", type="datetime")
+
+    # 결과 표시
+    with gr.Group(elem_id="results-section"):
+        with gr.Box(elem_id="stats-container", _class="card"):
+            num_papers = gr.Textbox(
+                label="Number of papers",
+                value=update_num_papers(paper_list.df_raw),
+                interactive=False
+            )
+        # 테이블
+        df = gr.Dataframe(
+            value=paper_list.df_prettified,
+            datatype=paper_list.column_datatype,
+            type="pandas",
+            interactive=False,
+            max_height=1000,
+            elem_id="table",
+            column_widths=["10%", "10%", "60%", "10%", "5%", "5%"],
+            wrap=True,
+        )
+
+    gr.Markdown(FOOT_NOTE, elem_classes="footer")
+
+    # 이벤트 연결
+    gr.on(
+        triggers=[start_date.change, end_date.change, search_title.submit, search_abstract.submit],
+        fn=search,
+        inputs=[start_date, end_date, search_title, search_abstract, max_num_to_retrieve],
+        outputs=df,
+        api_name=False,
+    ).then(
+        fn=update_num_papers,
+        inputs=df,
+        outputs=num_papers,
+        queue=False,
+        api_name=False,
+    )
+
+    # 첫 로드 시 df 갱신 -> 논문 수 갱신
+    demo.load(
+        fn=update_df,
+        outputs=df,
+        queue=False,
+        api_name=False,
+    ).then(
+        fn=update_num_papers,
+        inputs=df,
+        outputs=num_papers,
+        queue=False,
+        api_name=False,
+    )
+
+if __name__ == "__main__":
+    # queue()로 비동기 처리, API 비공개
+    demo.queue(api_open=False).launch(show_api=False)