Update app.py

app.py

@@ -28,31 +28,43 @@ app.add_middleware(
 
 @app.get("/api/results")
 async def get_results():
-    # Load the dataset
-    dataset = load_dataset("smolagents-benchmark/results")
-    
-    # Convert to list for processing
-    data = dataset["train"].to_pandas()
-    
-    # Process the data to group by model and calculate scores
-    processed_data = []
-    grouped = data.groupby('model_id')
-    
-    for model_id, group in grouped:
-        model_data = {
-            'model_id': model_id,
-            'scores': {}
-        }
+    try:
+        # Load the dataset
+        dataset = load_dataset("smolagents-benchmark/results")
         
-        # Calculate scores for each source
-        for source in group['source'].unique():
-            source_data = group[group['source'] == source]
-            avg_acc = source_data['acc'].mean()
-            model_data['scores'][source] = float(avg_acc)
+        # Convert to list for processing
+        data = dataset["train"].to_pandas()
+        
+        # Log some info to help debug
+        print("Dataset loaded, shape:", data.shape)
+        print("Columns:", data.columns)
+        print("First row:", data.iloc[0])
+        
+        # Process the data to group by model and calculate scores
+        processed_data = []
+        grouped = data.groupby('model_id')
+        
+        for model_id, group in grouped:
+            model_data = {
+                'model_id': model_id,
+                'scores': {}
+            }
             
-        processed_data.append(model_data)
-    
-    return processed_data
+            # Calculate scores for each source
+            for source in group['source'].unique():
+                source_data = group[group['source'] == source]
+                avg_acc = source_data['acc'].mean()
+                model_data['scores'][source] = float(avg_acc)
+                
+            processed_data.append(model_data)
+        
+        return processed_data
+        
+    except Exception as e:
+        # Print the full error traceback to your logs
+        print("Error occurred:", str(e))
+        print("Traceback:", traceback.format_exc())
+        raise HTTPException(status_code=500, detail=str(e))
 
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")