Improve the UI and training

services/forecasting/app/core/database.py

@@ -70,6 +70,47 @@ async def get_db_health() -> bool:
         logger.error("Database health check failed", error=str(e))
         return False
 
+
+async def get_connection_pool_stats() -> dict:
+    """
+    Get current connection pool statistics for monitoring.
+
+    Returns:
+        Dictionary with pool statistics including usage and capacity
+    """
+    try:
+        pool = async_engine.pool
+
+        # Get pool stats
+        stats = {
+            "pool_size": pool.size(),
+            "checked_in_connections": pool.checkedin(),
+            "checked_out_connections": pool.checkedout(),
+            "overflow_connections": pool.overflow(),
+            "total_connections": pool.size() + pool.overflow(),
+            "max_capacity": 10 + 20,  # pool_size + max_overflow
+            "usage_percentage": round(((pool.size() + pool.overflow()) / 30) * 100, 2)
+        }
+
+        # Add health status
+        if stats["usage_percentage"] > 90:
+            stats["status"] = "critical"
+            stats["message"] = "Connection pool near capacity"
+        elif stats["usage_percentage"] > 80:
+            stats["status"] = "warning"
+            stats["message"] = "Connection pool usage high"
+        else:
+            stats["status"] = "healthy"
+            stats["message"] = "Connection pool healthy"
+
+        return stats
+    except Exception as e:
+        logger.error("Failed to get connection pool stats", error=str(e))
+        return {
+            "status": "error",
+            "message": f"Failed to get pool stats: {str(e)}"
+        }
+
 # Database manager instance for service_base compatibility
 database_manager = DatabaseManager(
     database_url=settings.DATABASE_URL,