Update app.py

app.py

@@ -510,14 +510,15 @@ def generate_device_cards(df):
             return '<p>No valid timestamps available to display.</p>'
 
         # Calculate device stats
-        device_stats = df_clean.groupby('device_id', as_index=False).agg({
+        # Group by device_id and aggregate
+        device_stats = df_clean.groupby('device_id').agg({
             'status': 'last',  # Most recent status
             'timestamp': 'max',  # Last log timestamp
-            'device_id': 'count'  # Usage count (number of logs)
-        })
+        }).reset_index()
 
-        # Rename the columns directly
-        device_stats.columns = ['device_id', 'status', 'timestamp', 'count']
+        # Add the count of logs per device using size()
+        counts = df_clean.groupby('device_id').size().reset_index(name='count')
+        device_stats = device_stats.merge(counts, on='device_id')
 
         # Log the aggregated stats
         logging.info(f"Device stats after aggregation:\n{device_stats.to_string()}")