Update app.py

app.py

@@ -10,14 +10,26 @@ class PaperManager:
         self.current_page = 1
         self.papers = []
         self.total_pages = 0
+        self.last_fetched_page = 0
 
     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)
-            self.total_pages = (len(self.papers) + self.papers_per_page - 1) // self.papers_per_page
+            new_papers = sorted(data, key=lambda x: x.get('paper', {}).get('upvotes', 0), reverse=True)
+            
+            if page > self.last_fetched_page:
+                self.papers.extend(new_papers)
+                self.last_fetched_page = page
+            elif page < self.last_fetched_page:
+                self.papers = new_papers + self.papers
+                self.last_fetched_page = page
+            
+            self.total_pages = max(self.total_pages, page)
+            if len(new_papers) < self.papers_per_page:
+                self.total_pages = page
+            
             return True
         except requests.RequestException as e:
             print(f"Error fetching papers: {e}")
@@ -40,7 +52,7 @@ class PaperManager:
         </div>"""
 
     def render_papers(self):
-        if not self.papers:
+        if not self.papers or self.current_page > self.last_fetched_page:
             if not self.fetch_papers(self.current_page):
                 return "<div>Failed to fetch papers. Please try again later.</div>"
         
@@ -55,22 +67,21 @@ class PaperManager:
 
     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:
+        if self.current_page < self.total_pages or len(self.papers) >= self.current_page * self.papers_per_page:
             self.current_page += 1
-            self.fetch_papers(self.current_page)
-        return self.render_papers(), f"Page {self.current_page} of {self.total_pages}"
+        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
-            self.fetch_papers(self.current_page)
-        return self.render_papers(), f"Page {self.current_page} of {self.total_pages}"
+        return self.render_papers(), f"Page {self.current_page} of {self.total_pages}+"
 
 css = """
 body { 
@@ -97,8 +108,11 @@ paper_manager = PaperManager()
 
 def refresh_papers():
     paper_manager.current_page = 1
+    paper_manager.papers = []
+    paper_manager.last_fetched_page = 0
+    paper_manager.total_pages = 0
     paper_manager.fetch_papers()
-    return paper_manager.render_papers(), f"Page {paper_manager.current_page} of {paper_manager.total_pages}"
+    return paper_manager.render_papers(), f"Page {paper_manager.current_page} of {paper_manager.total_pages}+"
 
 demo = gr.Blocks(css=css)
 
@@ -112,7 +126,7 @@ with demo:
     with gr.Row():
         prev_button = gr.Button("Previous Page")
         next_button = gr.Button("Next Page")
-        page_info = gr.Markdown(f"Page {paper_manager.current_page} of {paper_manager.total_pages}")
+        page_info = gr.Markdown(f"Page {paper_manager.current_page} of {paper_manager.total_pages}+")
     
     search_input.change(paper_manager.search_papers, inputs=[search_input], outputs=[paper_list])
     refresh_button.click(refresh_papers, outputs=[paper_list, page_info])