Update app.py

app.py

@@ -7,24 +7,21 @@ 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, page=1):
         try:
             response = requests.get(f"{API_URL}?page={page}&limit={self.papers_per_page}")
             response.raise_for_status()
             data = response.json()
-            self.papers = sorted(data, key=lambda x: x.get('paper', {}).get('upvotes', 0), reverse=True)
+            papers = sorted(data, key=lambda x: x.get('paper', {}).get('upvotes', 0), reverse=True)
             
             # Update total_pages based on the 'X-Total-Pages' header if available
-            self.total_pages = int(response.headers.get('X-Total-Pages', self.total_pages))
+            total_pages = int(response.headers.get('X-Total-Pages', 1))
             
-            return True
+            return papers, total_pages
         except requests.RequestException as e:
             print(f"Error fetching papers: {e}")
-            return False
+            return None, 1
 
     def format_paper(self, paper):
         title = paper.get('title', 'No title')
@@ -32,7 +29,9 @@ class PaperManager:
         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'))
+        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 style='border-bottom: 1px solid #eee; padding: 10px 0;'>
@@ -42,33 +41,18 @@ class PaperManager:
         </div>
         </div>"""
 
-    def render_papers(self):
-        if not self.fetch_papers(self.current_page):
-            return "<div>Failed to fetch papers. Please try again later.</div>"
-        
-        if not self.papers:
+    def render_papers(self, papers):
+        if not papers:
             return "<div>No papers available for this page.</div>"
         
-        return "".join([self.format_paper(paper) for paper in self.papers])
+        return "".join([self.format_paper(paper) for paper in papers])
 
-    def search_papers(self, query):
+    def search_papers(self, papers, query):
         if not query:
-            self.current_page = 1
-            return self.render_papers()
+            return self.render_papers(papers)
         
-        self.fetch_papers(self.current_page)  # Ensure we have the latest data
-        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])
-
-    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}"
+        filtered_papers = [paper for paper in papers if query.lower() in paper.get('title', '').lower()]
+        return self.render_papers(filtered_papers)
 
 css = """
 html, body {
@@ -116,9 +100,37 @@ html, body {
 
 paper_manager = PaperManager()
 
-def refresh_papers():
-    paper_manager.current_page = 1
-    return paper_manager.render_papers(), f"Page {paper_manager.current_page} of {paper_manager.total_pages}"
+def initialize():
+    papers, total_pages = paper_manager.fetch_papers(page=1)
+    html = paper_manager.render_papers(papers)
+    return html, 1, total_pages, papers  # paper_list, current_page, total_pages, papers
+
+def refresh_papers(current_page, query):
+    papers, total_pages = paper_manager.fetch_papers(page=1)
+    if query:
+        html = paper_manager.search_papers(papers, query)
+    else:
+        html = paper_manager.render_papers(papers)
+    return html, 1, total_pages, papers
+
+def search_papers(query, papers):
+    html = paper_manager.search_papers(papers, query)
+    return html
+
+def change_page(direction, current_page, total_pages, papers, query):
+    if direction == "next" and current_page < total_pages:
+        new_page = current_page + 1
+    elif direction == "prev" and current_page > 1:
+        new_page = current_page - 1
+    else:
+        new_page = current_page  # No change if limits are reached
+
+    papers, total_pages = paper_manager.fetch_papers(page=new_page)
+    if query:
+        html = paper_manager.search_papers(papers, query)
+    else:
+        html = paper_manager.render_papers(papers)
+    return html, new_page, total_pages, papers
 
 demo = gr.Blocks(css=css)
 
@@ -128,16 +140,38 @@ with demo:
         with gr.Row(elem_classes=["search-row"]):
             search_input = gr.Textbox(label="Search papers", placeholder="Enter search term...")
             refresh_button = gr.Button("Refresh")
-        paper_list = gr.HTML(paper_manager.render_papers(), elem_classes=["paper-list"])
+        paper_list = gr.HTML(elem_classes=["paper-list"])
         
         with gr.Row(elem_classes=["footer"]):
             prev_button = gr.Button("Previous Page")
-            page_info = gr.Markdown(f"Page {paper_manager.current_page} of {paper_manager.total_pages}")
+            page_info = gr.Markdown("Page 1 of 1")
             next_button = gr.Button("Next Page")
-    
-    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.click(paper_manager.prev_page, outputs=[paper_list, page_info])
-    next_button.click(paper_manager.next_page, outputs=[paper_list, page_info])
+        
+        # Hidden states
+        current_page_state = gr.State(1)
+        total_pages_state = gr.State(1)
+        papers_state = gr.State([])
+
+        # Initialize the app
+        demo.load(initialize, outputs=[paper_list, current_page_state, total_pages_state, papers_state])
+
+        # Search functionality
+        search_input.submit(search_papers, inputs=[search_input, papers_state], outputs=[paper_list])
+        search_input.change(search_papers, inputs=[search_input, papers_state], outputs=[paper_list])
+
+        # Refresh functionality
+        refresh_button.click(refresh_papers, 
+                             inputs=[current_page_state, search_input], 
+                             outputs=[paper_list, current_page_state, total_pages_state, papers_state])
+
+        # Previous Page
+        prev_button.click(change_page, 
+                          inputs=["prev", current_page_state, total_pages_state, papers_state, search_input], 
+                          outputs=[paper_list, current_page_state, total_pages_state, papers_state])
+
+        # Next Page
+        next_button.click(change_page, 
+                          inputs=["next", current_page_state, total_pages_state, papers_state, search_input], 
+                          outputs=[paper_list, current_page_state, total_pages_state, papers_state])
 
-demo.launch()
+demo.launch()