Update app.py

app.py

@@ -1,3 +1,227 @@
+import gradio as gr
+import requests
+from datetime import datetime, timezone
+
+API_URL = "https://huggingface.co/api/daily_papers"
+
+class PaperManager:
+    def __init__(self, papers_per_page=10):
+        self.papers_per_page = papers_per_page
+        self.current_page = 1
+        self.papers = []
+        self.total_pages = 1
+
+    def fetch_papers(self):
+        try:
+            response = requests.get(f"{API_URL}?limit=50")
+            response.raise_for_status()
+            data = response.json()
+            self.papers = sorted(data, key=lambda x: x.get('paper', {}).get('upvotes', 0), reverse=True)
+            self.total_pages = (len(self.papers) + self.papers_per_page - 1) // self.papers_per_page
+            self.current_page = 1
+            return True
+        except requests.RequestException as e:
+            print(f"Error fetching papers: {e}")
+            return False
+
+    def format_paper(self, paper):
+        title = paper.get('title', 'No title')
+        url = f"https://huggingface.co/papers/{paper['paper'].get('id', '')}"
+        authors = ', '.join([author.get('name', '') for author in paper['paper'].get('authors', [])])
+        upvotes = paper.get('paper', {}).get('upvotes', 0)
+        comments = paper.get('numComments', 0)
+        published_time = datetime.fromisoformat(paper.get('publishedAt', datetime.now(timezone.utc).isoformat()).replace('Z', '+00:00'))
+        time_ago = (datetime.now(timezone.utc) - published_time).days
+        
+        return f"""
+        <div class="paper-item">
+            <h3><a href="{url}" target="_blank">{title}</a></h3>
+            <p class="paper-meta">
+                <span class="upvotes">{upvotes} ▲</span> 
+                <span class="authors">{authors}</span>
+                <span class="time-ago">{time_ago}d</span>
+                <span class="comments">{comments}💬</span>
+            </p>
+        </div>
+        """
+
+    def render_papers(self):
+        start = (self.current_page - 1) * self.papers_per_page
+        end = start + self.papers_per_page
+        current_papers = self.papers[start:end]
+        
+        if not current_papers:
+            return "<div class='no-papers'>No papers available for this page.</div>"
+        
+        return "".join([self.format_paper(paper) for paper in current_papers])
+
+    def search_papers(self, query):
+        if not query:
+            self.current_page = 1
+            return self.render_papers()
+        
+        filtered_papers = [paper for paper in self.papers if query.lower() in paper.get('title', '').lower()]
+        return "".join([self.format_paper(paper) for paper in filtered_papers[:self.papers_per_page]])
+
+    def next_page(self):
+        if self.current_page < self.total_pages:
+            self.current_page += 1
+        return self.render_papers(), f"Page {self.current_page} of {self.total_pages}"
+
+    def prev_page(self):
+        if self.current_page > 1:
+            self.current_page -= 1
+        return self.render_papers(), f"Page {self.current_page} of {self.total_pages}"
+
+css = """
+body {
+    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Oxygen', 'Ubuntu', 'Cantarell', 'Fira Sans', 'Droid Sans', 'Helvetica Neue', sans-serif;
+    background-color: #f0f2f5;
+    margin: 0;
+    padding: 0;
+}
+.container {
+    max-width: 100%;
+    margin: 0 auto;
+    background-color: white;
+    padding: 1rem;
+}
+.title {
+    text-align: center;
+    color: #1a202c;
+    font-size: 1.5rem;
+    margin-bottom: 1rem;
+}
+.search-row {
+    display: flex;
+    gap: 0.5rem;
+    margin-bottom: 1rem;
+    align-items: flex-end;
+}
+.search-row > div:first-child {
+    flex-grow: 1;
+}
+.paper-list {
+    max-height: 70vh;
+    overflow-y: auto;
+    border: 1px solid #e2e8f0;
+    border-radius: 8px;
+    padding: 0.5rem;
+}
+.paper-item {
+    border-bottom: 1px solid #e2e8f0;
+    padding: 0.5rem 0;
+}
+.paper-item:last-child {
+    border-bottom: none;
+}
+.paper-item h3 {
+    margin: 0 0 0.25rem 0;
+    font-size: 1rem;
+}
+.paper-item a {
+    color: #2b6cb0;
+    text-decoration: none;
+    font-weight: 600;
+}
+.paper-item a:hover {
+    text-decoration: underline;
+}
+.paper-meta {
+    font-size: 0.75rem;
+    color: #4a5568;
+    display: flex;
+    flex-wrap: wrap;
+    gap: 0.5rem;
+}
+.footer {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    margin-top: 0.5rem;
+}
+button, .button {
+    background-color: #4299e1;
+    color: white;
+    border: none;
+    padding: 0.25rem 0.5rem;
+    border-radius: 4px;
+    cursor: pointer;
+    transition: background-color 0.3s;
+    font-size: 0.875rem;
+    line-height: 1.25rem;
+}
+button:hover, .button:hover {
+    background-color: #3182ce;
+}
+.no-papers {
+    text-align: center;
+    color: #718096;
+    padding: 1rem;
+}
+#component-0 > div:first-child {
+    border-top-right-radius: 0;
+    border-bottom-right-radius: 0;
+}
+#component-1 {
+    height: 45px;
+    border-top-left-radius: 0;
+    border-bottom-left-radius: 0;
+}
+
+@media (min-width: 640px) {
+    .container {
+        max-width: 640px;
+        padding: 1.5rem;
+    }
+    .title {
+        font-size: 1.75rem;
+    }
+    .paper-item h3 {
+        font-size: 1.125rem;
+    }
+    .paper-meta {
+        font-size: 0.875rem;
+    }
+}
+
+@media (min-width: 768px) {
+    .container {
+        max-width: 768px;
+    }
+}
+"""
+
+paper_manager = PaperManager()
+
+def initialize_app():
+    if paper_manager.fetch_papers():
+        prev_disabled = paper_manager.current_page == 1
+        next_disabled = paper_manager.current_page >= paper_manager.total_pages
+
+        return (
+            paper_manager.render_papers(),
+            f"Page {paper_manager.current_page} of {paper_manager.total_pages}",
+            prev_disabled,
+            next_disabled
+        )
+    else:
+        return "<div class='no-papers'>Failed to fetch papers. Please try again later.</div>", "Error", True, True
+
+def refresh_papers():
+    if paper_manager.fetch_papers():
+        prev_disabled = paper_manager.current_page == 1
+        next_disabled = paper_manager.current_page >= paper_manager.total_pages
+        
+        return (
+            paper_manager.render_papers(),
+            f"Page {paper_manager.current_page} of {paper_manager.total_pages}",
+            prev_disabled,
+            next_disabled
+        )
+    else:
+        return "<div class='no-papers'>Failed to refresh papers. Please try again later.</div>", "Error", True, True
+
 demo = gr.Blocks(css=css)
 
 with demo:
@@ -13,7 +237,6 @@ with demo:
             page_info = gr.Markdown()
             next_button = gr.Button("Next →", interactive=True)
     
-    # Initialize app and set button states
     demo.load(initialize_app, outputs=[paper_list, page_info, prev_button, next_button])
     search_input.change(paper_manager.search_papers, inputs=[search_input], outputs=[paper_list])
     refresh_button.click(refresh_papers, outputs=[paper_list, page_info, prev_button, next_button])