Update app.py

app.py

@@ -41,10 +41,12 @@ class PaperManager:
         Returns a dictionary with counts for each type.
         """
         try:
+            print(f"Fetching repos for arxiv_id: {arxiv_id}")  # Debug
             response = requests.get(REPOS_API_URL_TEMPLATE.format(arxiv_id=arxiv_id))
             response.raise_for_status()
             data = response.json()
-            
+            print(f"Repos data for arxiv_id {arxiv_id}: {data}")  # Debug
+
             counts = {'models': 0, 'datasets': 0, 'spaces': 0}
             for repo in data:
                 repo_type = repo.get('type', '').lower()
@@ -54,6 +56,7 @@ class PaperManager:
                     counts['datasets'] += 1
                 elif repo_type == 'space':
                     counts['spaces'] += 1
+            print(f"Counts for arxiv_id {arxiv_id}: {counts}")  # Debug
             return counts
         except requests.RequestException as e:
             print(f"Error fetching repos for arxiv_id {arxiv_id}: {e}")
@@ -82,6 +85,7 @@ class PaperManager:
                 future_to_paper = {}
                 for paper in self.raw_papers:
                     arxiv_id = paper.get('paper', {}).get('arxiv_id', '')
+                    print(f"Processing paper ID: {paper.get('paper', {}).get('id', '')}, arxiv_id: {arxiv_id}")  # Debug
                     if arxiv_id:
                         future = executor.submit(self.fetch_repos_counts, arxiv_id)
                         future_to_paper[future] = paper
@@ -94,6 +98,7 @@ class PaperManager:
                 for future in as_completed(future_to_paper):
                     paper = future_to_paper[future]
                     counts = future.result()
+                    print(f"Paper ID: {paper.get('paper', {}).get('id', '')}, Counts: {counts}")  # Debug
                     paper['models'] = counts['models']
                     paper['datasets'] = counts['datasets']
                     paper['spaces'] = counts['spaces']
@@ -147,12 +152,13 @@ class PaperManager:
                 key=lambda x: self.calculate_score(x),
                 reverse=True
             )
+        print(f"Papers sorted by {self.sort_method}")  # Debug
 
     def set_sort_method(self, method):
         valid_methods = ["hot", "new", "most_models", "most_datasets", "most_spaces"]
         if method not in valid_methods:
             method = "hot"
-        print(f"Setting sort method to: {method}")
+        print(f"Setting sort method to: {method}")  # Debug
         self.sort_method = method
         self.sort_papers()
         self.current_page = 1
@@ -178,6 +184,8 @@ class PaperManager:
         datasets = paper.get('datasets', 0)
         spaces = paper.get('spaces', 0)
 
+        print(f"Formatting paper {rank}: Models={models}, Datasets={datasets}, Spaces={spaces}")  # Debug
+
         return f"""
         <tr class="athing">
             <td align="right" valign="top" class="title"><span class="rank">{rank}.</span></td>
@@ -237,7 +245,7 @@ def refresh_papers():
 
 def change_sort_method(method):
     method_lower = method.lower().replace(" ", "_")
-    print(f"Changing sort method to: {method_lower}")
+    print(f"Changing sort method to: {method_lower}")  # Debug
     if paper_manager.set_sort_method(method_lower):
         print("Sort method set successfully.")
         return paper_manager.render_papers()
@@ -430,4 +438,4 @@ with demo:
         outputs=[paper_list]
     )
 
-demo.launch()
+demo.launch()