updates

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import requests
-from datetime import datetime, timezone
+from datetime import datetime, timezone, timedelta
 
 API_URL = "https://huggingface.co/api/daily_papers"
 
@@ -10,6 +10,7 @@ class PaperManager:
         self.current_page = 1
         self.papers = []
         self.total_pages = 1
+        self.time_filter = 'All Time'  # Default filter
 
     def calculate_score(self, paper):
         """
@@ -31,21 +32,28 @@ class PaperManager:
         score = upvotes / ((time_diff_hours + 2) ** 1.5)
         return score
 
-    def fetch_papers(self):
+    def fetch_papers(self, time_filter='All Time'):
+        """
+        Fetch papers from the API and apply time filtering.
+        """
         try:
             response = requests.get(f"{API_URL}?limit=100")
             response.raise_for_status()
             data = response.json()
 
+            # Apply time filter
+            filtered_data = self.apply_time_filter(data, time_filter)
+
             # Sort papers by calculated score descending
             self.papers = sorted(
-                data,
+                filtered_data,
                 key=lambda x: self.calculate_score(x),
                 reverse=True
             )
 
             self.total_pages = max((len(self.papers) + self.papers_per_page - 1) // self.papers_per_page, 1)
             self.current_page = 1
+            self.time_filter = time_filter
             return True
         except requests.RequestException as e:
             print(f"Error fetching papers: {e}")
@@ -54,6 +62,36 @@ class PaperManager:
             print(f"Unexpected error: {e}")
             return False
 
+    def apply_time_filter(self, data, time_filter):
+        """
+        Filter papers based on the selected timeframe.
+        """
+        if time_filter == 'All Time':
+            return data
+        
+        now = datetime.now(timezone.utc)
+        if time_filter == 'Last Week':
+            threshold = now - timedelta(weeks=1)
+        elif time_filter == 'Last Month':
+            threshold = now - timedelta(days=30)
+        elif time_filter == 'Last Year':
+            threshold = now - timedelta(days=365)
+        else:
+            # If an unknown filter is provided, default to all time
+            return data
+
+        filtered = []
+        for paper in data:
+            published_at_str = paper.get('publishedAt', '')
+            try:
+                published_time = datetime.fromisoformat(published_at_str.replace('Z', '+00:00'))
+                if published_time >= threshold:
+                    filtered.append(paper)
+            except ValueError:
+                # Skip papers with invalid date formats
+                continue
+        return filtered
+
     def format_paper(self, paper, rank):
         title = paper.get('title', 'No title')
         paper_id = paper.get('paper', {}).get('id', '')
@@ -110,6 +148,15 @@ class PaperManager:
             self.current_page -= 1
         return self.render_papers()
 
+    def set_time_filter(self, time_filter):
+        """
+        Set the time filter and fetch papers accordingly.
+        """
+        if self.fetch_papers(time_filter):
+            return self.render_papers()
+        else:
+            return "<div class='no-papers'>Failed to fetch papers. Please try again later.</div>"
+
 paper_manager = PaperManager()
 
 def initialize_app():
@@ -119,7 +166,7 @@ def initialize_app():
         return "<div class='no-papers'>Failed to fetch papers. Please try again later.</div>"
 
 def refresh_papers():
-    if paper_manager.fetch_papers():
+    if paper_manager.fetch_papers(paper_manager.time_filter):
         return paper_manager.render_papers()
     else:
         return "<div class='no-papers'>Failed to refresh papers. Please try again later.</div>"
@@ -284,7 +331,8 @@ with demo:
 
             Once your paper is submitted, it will automatically appear in this demo.
             """)
-        # Header with Refresh Button
+        
+        # Header with Refresh Button and Time Filter
         with gr.Row():
             gr.HTML("""
             <table border="0" cellpadding="0" cellspacing="0" class="header-table">
@@ -300,14 +348,26 @@ with demo:
                 </tr>
             </table>
             """)
+        
+        # Time Filter Dropdown
+        with gr.Row(elem_classes=["time-filter-row"], elem_id="time-filter-row"):
+            gr.HTML("<label for='time-filter'>Filter by Timeframe: </label>")
+            time_filter_dropdown = gr.Dropdown(
+                choices=["All Time", "Last Week", "Last Month", "Last Year"],
+                value="All Time",
+                label="Timeframe",
+                interactive=True,
+                elem_id="time-filter-dropdown"
+            )
+        
         # Paper list
         paper_list = gr.HTML()
+        
         # Navigation Buttons
         with gr.Row():
             prev_button = gr.Button("Prev")
             next_button = gr.Button("Next")
 
-
     # Load papers on app start
     demo.load(initialize_app, outputs=[paper_list])
 
@@ -317,6 +377,13 @@ with demo:
     refresh_button = gr.Button("Refresh", visible=False, elem_id="refresh-hidden")
     refresh_button.click(refresh_papers, outputs=[paper_list])
 
+    # Time Filter change
+    time_filter_dropdown.change(
+        paper_manager.set_time_filter,
+        inputs=[time_filter_dropdown],
+        outputs=[paper_list]
+    )
+
     # Bind the visible Refresh button to the hidden one using JavaScript
     gr.HTML("""
     <script>