Update app.py

app.py

@@ -9,26 +9,17 @@ class PaperManager:
         self.papers_per_page = papers_per_page
         self.current_page = 1
         self.papers = []
-        self.total_pages = 0
-        self.last_fetched_page = 0
+        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()
-            new_papers = sorted(data, key=lambda x: x.get('paper', {}).get('upvotes', 0), reverse=True)
+            self.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
+            # 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))
             
             return True
         except requests.RequestException as e:
@@ -52,81 +43,98 @@ class PaperManager:
         </div>"""
 
     def render_papers(self):
-        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>"
-        
-        start_index = (self.current_page - 1) * self.papers_per_page
-        end_index = start_index + self.papers_per_page
-        current_papers = self.papers[start_index:end_index]
+        if not self.fetch_papers(self.current_page):
+            return "<div>Failed to fetch papers. Please try again later.</div>"
         
-        if not current_papers:
-            return "<div>No more papers available.</div>"
+        if not self.papers:
+            return "<div>No papers available for this page.</div>"
         
-        return "".join([self.format_paper(paper) for paper in current_papers])
+        return "".join([self.format_paper(paper) for paper in self.papers])
 
     def search_papers(self, query):
         if not query:
             self.current_page = 1
             return self.render_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[:self.papers_per_page]])
+        return "".join([self.format_paper(paper) for paper in filtered_papers])
 
     def next_page(self):
-        if self.current_page < self.total_pages or len(self.papers) >= self.current_page * self.papers_per_page:
+        if self.current_page < self.total_pages:
             self.current_page += 1
-        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
-        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 { 
-    font-family: Arial, sans-serif; 
+html, body {
+    height: 100%;
+    margin: 0;
+    padding: 0;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    background-color: #f0f0f0;
+}
+.container {
+    font-family: Arial, sans-serif;
     max-width: 800px;
-    margin: 0 auto;
+    width: 100%;
+    background-color: white;
     padding: 20px;
+    border-radius: 10px;
+    box-shadow: 0 0 10px rgba(0,0,0,0.1);
 }
 .paper-list { 
-    max-height: 600px; 
+    max-height: 400px; 
     overflow-y: auto;
     border: 1px solid #eee;
     border-radius: 5px;
     padding: 10px;
+    margin-bottom: 10px;
 }
 .search-row {
     display: flex;
     gap: 10px;
     margin-bottom: 20px;
 }
+.title {
+    text-align: center;
+    color: #333;
+}
+.footer {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    margin-top: 10px;
+}
 """
 
 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)
 
 with demo:
-    gr.Markdown("# Daily Papers - HackerNews Style")
-    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"])
-    
-    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}+")
+    with gr.Column(elem_classes=["container"]):
+        gr.Markdown("# Daily Papers - HackerNews Style", elem_classes=["title"])
+        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"])
+        
+        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}")
+            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])