Update app.py

app.py

@@ -21,18 +21,21 @@ class PaperManager:
             return papers, total_pages
         except requests.RequestException as e:
             print(f"Error fetching papers: {e}")
-            return None, 1
+            return [], 1  # Return empty list and default to 1 page
 
     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', [])])
+        paper_id = paper.get('paper', {}).get('id', '')
+        url = f"https://huggingface.co/papers/{paper_id}" if paper_id else "#"
+        authors = ', '.join([author.get('name', 'Unknown') for author in paper.get('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
+        published_at = paper.get('publishedAt', datetime.now(timezone.utc).isoformat())
+        try:
+            published_time = datetime.fromisoformat(published_at.replace('Z', '+00:00'))
+            time_ago = (datetime.now(timezone.utc) - published_time).days
+        except ValueError:
+            time_ago = "Unknown"
         
         return f"""<div style='border-bottom: 1px solid #eee; padding: 10px 0;'>
         <a href='{url}' target='_blank' style='color: #000; text-decoration: none; font-weight: bold;'>{title}</a>
@@ -52,6 +55,8 @@ class PaperManager:
             return self.render_papers(papers)
         
         filtered_papers = [paper for paper in papers if query.lower() in paper.get('title', '').lower()]
+        if not filtered_papers:
+            return "<div>No papers match your search query.</div>"
         return self.render_papers(filtered_papers)
 
 css = """
@@ -106,6 +111,7 @@ def initialize():
     return html, 1, total_pages, papers  # paper_list, current_page, total_pages, papers
 
 def refresh_papers(current_page, query):
+    # Refresh always starts from page 1
     papers, total_pages = paper_manager.fetch_papers(page=1)
     if query:
         html = paper_manager.search_papers(papers, query)
@@ -132,6 +138,12 @@ def change_page(direction, current_page, total_pages, papers, query):
         html = paper_manager.render_papers(papers)
     return html, new_page, total_pages, papers
 
+def go_prev(current_page, total_pages, papers, query):
+    return change_page("prev", current_page, total_pages, papers, query)
+
+def go_next(current_page, total_pages, papers, query):
+    return change_page("next", current_page, total_pages, papers, query)
+
 demo = gr.Blocks(css=css)
 
 with demo:
@@ -165,13 +177,13 @@ with demo:
                              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], 
+        prev_button.click(go_prev, 
+                          inputs=[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], 
+        next_button.click(go_next, 
+                          inputs=[current_page_state, total_pages_state, papers_state, search_input], 
                           outputs=[paper_list, current_page_state, total_pages_state, papers_state])
 
 demo.launch()